add coderd test

sreya · sreya · commit 8888c2e550f4 · 2023-05-17T04:07:06.000Z
diff --git a/coderd/workspaceagents_test.go b/coderd/workspaceagents_test.go
@@ -1326,3 +1326,85 @@ func TestWorkspaceAgent_Metadata(t *testing.T) {
 	err = agentClient.PostMetadata(ctx, "unknown", unknownKeyMetadata)
 	require.NoError(t, err)
 }
+
+func TestWorkspaceAgent_Startup(t *testing.T) {
+	t.Parallel()
+
+	t.Run("OK", func(t *testing.T) {
+		t.Parallel()
+
+		client := coderdtest.New(t, &coderdtest.Options{
+			IncludeProvisionerDaemon: true,
+		})
+		user := coderdtest.CreateFirstUser(t, client)
+		authToken := uuid.NewString()
+		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
+			Parse:          echo.ParseComplete,
+			ProvisionPlan:  echo.ProvisionComplete,
+			ProvisionApply: echo.ProvisionApplyWithAgent(authToken),
+		})
+		template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
+		coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
+		workspace := coderdtest.CreateWorkspace(t, client, user.OrganizationID, template.ID)
+		coderdtest.AwaitWorkspaceBuildJob(t, client, workspace.LatestBuild.ID)
+
+		agentClient := agentsdk.New(client.URL)
+		agentClient.SetSessionToken(authToken)
+
+		ctx := testutil.Context(t, testutil.WaitMedium)
+
+		const (
+			expectedVersion   = "v1.2.3"
+			expectedDir       = "/home/coder"
+			expectedSubsystem = codersdk.AgentSubsystemEnvbox
+		)
+
+		err := agentClient.PostStartup(ctx, agentsdk.PostStartupRequest{
+			Version:           expectedVersion,
+			ExpandedDirectory: expectedDir,
+			Subsystem:         expectedSubsystem,
+		})
+		require.NoError(t, err)
+
+		workspace, err = client.Workspace(ctx, workspace.ID)
+		require.NoError(t, err)
+
+		wsagent, err := client.WorkspaceAgent(ctx, workspace.LatestBuild.Resources[0].Agents[0].ID)
+		require.NoError(t, err)
+		require.Equal(t, expectedVersion, wsagent.Version)
+		require.Equal(t, expectedDir, wsagent.ExpandedDirectory)
+		require.Equal(t, expectedSubsystem, wsagent.Subsystem)
+	})
+
+	t.Run("InvalidSemver", func(t *testing.T) {
+		t.Parallel()
+
+		client := coderdtest.New(t, &coderdtest.Options{
+			IncludeProvisionerDaemon: true,
+		})
+		user := coderdtest.CreateFirstUser(t, client)
+		authToken := uuid.NewString()
+		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
+			Parse:          echo.ParseComplete,
+			ProvisionPlan:  echo.ProvisionComplete,
+			ProvisionApply: echo.ProvisionApplyWithAgent(authToken),
+		})
+		template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
+		coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
+		workspace := coderdtest.CreateWorkspace(t, client, user.OrganizationID, template.ID)
+		coderdtest.AwaitWorkspaceBuildJob(t, client, workspace.LatestBuild.ID)
+
+		agentClient := agentsdk.New(client.URL)
+		agentClient.SetSessionToken(authToken)
+
+		ctx := testutil.Context(t, testutil.WaitMedium)
+
+		err := agentClient.PostStartup(ctx, agentsdk.PostStartupRequest{
+			Version: "1.2.3",
+		})
+		require.Error(t, err)
+		cerr, ok := codersdk.AsError(err)
+		require.True(t, ok)
+		require.Equal(t, http.StatusBadRequest, cerr.StatusCode())
+	})
+}
