mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
Kyle CarberryandBryan
35ae532f7c
* ci: Update DataDog GitHub branch to fallback to GITHUB_REF This was detecting branches, but not our "main" branch before. Hopefully this fixes it! * Add basic Terraform Provider * Rename post files to upload * Add tests for resources * Skip instance identity test * Add tests for ensuring agent get's passed through properly * Fix linting errors * Add echo path * Fix agent authentication * Update codersdk/files.go Co-authored-by: Bryan <bryan@coder.com> Co-authored-by: Bryan <bryan@coder.com>
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
// There isn't a portable Windows binary equivalent to "echo".
|
|
// This can be tested, but it's a bit harder.
|
|
|
|
package provisionersdk_test
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/go-chi/render"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/provisionersdk"
|
|
)
|
|
|
|
func TestAgentScript(t *testing.T) {
|
|
t.Parallel()
|
|
t.Run("Run", func(t *testing.T) {
|
|
t.Parallel()
|
|
srv := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
echoPath, err := exec.LookPath("echo")
|
|
require.NoError(t, err)
|
|
content, err := os.ReadFile(echoPath)
|
|
require.NoError(t, err)
|
|
render.Status(r, http.StatusOK)
|
|
render.Data(rw, r, content)
|
|
}))
|
|
t.Cleanup(srv.Close)
|
|
srvURL, err := url.Parse(srv.URL)
|
|
require.NoError(t, err)
|
|
script, err := provisionersdk.AgentScript(srvURL, runtime.GOOS, runtime.GOARCH)
|
|
require.NoError(t, err)
|
|
|
|
output, err := exec.Command("sh", "-c", script).CombinedOutput()
|
|
t.Log(string(output))
|
|
require.NoError(t, err)
|
|
// Because we use the "echo" binary, we should expect the arguments provided
|
|
// as the response to executing our script.
|
|
require.Equal(t, "agent", strings.TrimSpace(string(output)))
|
|
})
|
|
|
|
t.Run("UnsupportedOS", func(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := provisionersdk.AgentScript(nil, "unsupported", "")
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("UnsupportedArch", func(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := provisionersdk.AgentScript(nil, runtime.GOOS, "unsupported")
|
|
require.Error(t, err)
|
|
})
|
|
}
|