Function TestReinitializeAgent

enterprise/coderd/workspaceagents_test.go:88–266

(t *testing.T)

}

func TestReinitializeAgent(t *testing.T) {
	t.Parallel()

	// Ensure that workspace agents can reinitialize against claimed prebuilds in non-default organizations:
	for _, useDefaultOrg := range []bool{true, false} {
		t.Run(fmt.Sprintf("useDefaultOrg=%t", useDefaultOrg), func(t *testing.T) {
			t.Parallel()

			// Create the temp file in os.TempDir() rather than t.TempDir().
			// On Windows, t.TempDir() includes the test name which
			// contains "=" (e.g. useDefaultOrg=true). The "=" in the
			// path breaks both cmd.exe and powershell scripts, causing
			// the startup script to exit 1 and the agent to never
			// reach the ready lifecycle state.
			tempAgentLog := testutil.CreateTemp(t, os.TempDir(), "testReinitializeAgent")

			// Dump environment variables to a temp file so we can verify
			// CODER_AGENT_TOKEN appears twice (once per init). On Windows
			// the agent runs scripts via powershell.exe /c, so we must
			// use PowerShell-native commands.
			var startupScript string
			if runtime.GOOS == "windows" {
				startupScript = fmt.Sprintf(
					`[System.Environment]::GetEnvironmentVariables().GetEnumerator() | ForEach-Object { "$($_.Key)=$($_.Value)" } | Add-Content -Path '%s'; '---' | Add-Content -Path '%s'`,
					tempAgentLog.Name(), tempAgentLog.Name(),
				)
			} else {
				startupScript = fmt.Sprintf("printenv >> %s; echo '---\n' >> %s", tempAgentLog.Name(), tempAgentLog.Name())
			}

			db, ps := dbtestutil.NewDB(t)
			// GIVEN a live enterprise API with the prebuilds feature enabled
			client, user := coderdenttest.New(t, &coderdenttest.Options{
				Options: &coderdtest.Options{
					Database: db,
					Pubsub:   ps,
					DeploymentValues: coderdtest.DeploymentValues(t, func(dv *codersdk.DeploymentValues) {
						dv.Prebuilds.ReconciliationInterval = serpent.Duration(time.Second)
					}),
				},
				LicenseOptions: &coderdenttest.LicenseOptions{
					Features: license.Features{
						codersdk.FeatureWorkspacePrebuilds:         1,
						codersdk.FeatureExternalProvisionerDaemons: 1,
					},
				},
			})

			orgID := user.OrganizationID
			if !useDefaultOrg {
				secondOrg := dbgen.Organization(t, db, database.Organization{})
				orgID = secondOrg.ID
			}
			provisionerCloser := coderdenttest.NewExternalProvisionerDaemon(t, client, orgID, map[string]string{
				provisionersdk.TagScope: provisionersdk.ScopeOrganization,
			})
			defer provisionerCloser.Close()