test: add WorkspaceApps functionality tests

blink-so[bot] · blink-so[bot] · commit 0c281d93692d · 2025-07-19T19:49:14.000Z
Add comprehensive tests for the workspace_apps feature:
- Test that workspace_apps values are correctly stored and retrieved
- Test default behavior when workspace_apps is not specified
- Follow existing test patterns for DisplayApps functionality
diff --git a/coderd/provisionerdserver/provisionerdserver_test.go b/coderd/provisionerdserver/provisionerdserver_test.go
@@ -3167,6 +3167,63 @@ func TestInsertWorkspaceResource(t *testing.T) {
 		require.Equal(t, []database.DisplayApp{}, agent.DisplayApps)
 	})
 
+	t.Run("WorkspaceApps", func(t *testing.T) {
+		t.Parallel()
+		db, _ := dbtestutil.NewDB(t)
+		dbtestutil.DisableForeignKeysAndTriggers(t, db)
+		job := uuid.New()
+		err := insert(db, job, &sdkproto.Resource{
+			Name: "something",
+			Type: "aws_instance",
+			Agents: []*sdkproto.Agent{{
+				Name: "dev",
+				DisplayApps: &sdkproto.DisplayApps{
+					Vscode:      true,
+					WebTerminal: true,
+					SshHelper:   true,
+				},
+				WorkspaceApps: []string{"vscode", "custom-app"},
+			}},
+		})
+		require.NoError(t, err)
+		resources, err := db.GetWorkspaceResourcesByJobID(ctx, job)
+		require.NoError(t, err)
+		require.Len(t, resources, 1)
+		agents, err := db.GetWorkspaceAgentsByResourceIDs(ctx, []uuid.UUID{resources[0].ID})
+		require.NoError(t, err)
+		require.Len(t, agents, 1)
+		agent := agents[0]
+		require.Equal(t, []string{"vscode", "custom-app"}, agent.WorkspaceApps)
+	})
+
+	t.Run("WorkspaceAppsEmpty", func(t *testing.T) {
+		t.Parallel()
+		db, _ := dbtestutil.NewDB(t)
+		dbtestutil.DisableForeignKeysAndTriggers(t, db)
+		job := uuid.New()
+		err := insert(db, job, &sdkproto.Resource{
+			Name: "something",
+			Type: "aws_instance",
+			Agents: []*sdkproto.Agent{{
+				Name: "dev",
+				DisplayApps: &sdkproto.DisplayApps{
+					Vscode: true,
+				},
+				// WorkspaceApps not specified, should be nil
+			}},
+		})
+		require.NoError(t, err)
+		resources, err := db.GetWorkspaceResourcesByJobID(ctx, job)
+		require.NoError(t, err)
+		require.Len(t, resources, 1)
+		agents, err := db.GetWorkspaceAgentsByResourceIDs(ctx, []uuid.UUID{resources[0].ID})
+		require.NoError(t, err)
+		require.Len(t, agents, 1)
+		agent := agents[0]
+		// When WorkspaceApps is not specified, it should be nil (default behavior)
+		require.Nil(t, agent.WorkspaceApps)
+	})
+
 	t.Run("ResourcesMonitoring", func(t *testing.T) {
 		t.Parallel()
 		db, _ := dbtestutil.NewDB(t)
