chore: fix TestWorkspaceAgentAppHealth test after agent ID field now required

dannykopping · dannykopping · commit 4816ed5b60b5 · 2025-06-23T10:52:30.000+02:00
Signed-off-by: Danny Kopping &lt;dannykopping@gmail.com&gt;
diff --git a/coderd/workspaceagents_test.go b/coderd/workspaceagents_test.go
@@ -2,6 +2,7 @@ package coderd_test
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"fmt"
 	"maps"
@@ -1506,39 +1507,52 @@ func TestWorkspaceAgentAppHealth(t *testing.T) {
 	t.Parallel()
 	client, db := coderdtest.NewWithDatabase(t, nil)
 	user := coderdtest.CreateFirstUser(t, client)
-	apps := []*proto.App{
-		{
-			Slug:    "code-server",
-			Command: "some-command",
-			Url:     "http://localhost:3000",
-			Icon:    "/code.svg",
-		},
-		{
-			Slug:        "code-server-2",
-			DisplayName: "code-server-2",
-			Command:     "some-command",
-			Url:         "http://localhost:3000",
-			Icon:        "/code.svg",
-			Healthcheck: &proto.Healthcheck{
-				Url:       "http://localhost:3000",
-				Interval:  5,
-				Threshold: 6,
-			},
-		},
-	}
+
+	// When using WithAgent() here and setting some *proto.App instances on the agent, Do() ends up trying to insert duplicate records.
 	r := dbfake.WorkspaceBuild(t, db, database.WorkspaceTable{
 		OrganizationID: user.OrganizationID,
 		OwnerID:        user.UserID,
-	}).WithAgent(func(agents []*proto.Agent) []*proto.Agent {
-		agents[0].Apps = apps
-		return agents
 	}).Do()
 
+	res := dbgen.WorkspaceResource(t, db, database.WorkspaceResource{JobID: r.Build.JobID})
+	agent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{ResourceID: res.ID})
+
+	// It's simpler to call db.InsertWorkspaceApp directly than dbgen.WorkspaceApp because it's more terse and direct;
+	// the latter sets a bunch of defaults which make this test hard.
+	_, err := db.InsertWorkspaceApp(dbauthz.AsSystemRestricted(t.Context()), database.InsertWorkspaceAppParams{
+		ID: uuid.New(),
+		Slug:         "code-server",
+		AgentID:      agent.ID,
+		Icon:         "/code.svg",
+		Command:      sql.NullString{String: "some-command", Valid: true},
+		Url:          sql.NullString{String: "http://localhost:3000", Valid: true},
+		SharingLevel: database.AppSharingLevelOwner,
+		Health:       database.WorkspaceAppHealthDisabled,
+		OpenIn:       database.WorkspaceAppOpenInWindow,
+	})
+	require.NoError(t, err)
+	_, err = db.InsertWorkspaceApp(dbauthz.AsSystemRestricted(t.Context()), database.InsertWorkspaceAppParams{
+		ID: uuid.New(),
+		Slug:                 "code-server-2",
+		DisplayName:          "code-server-2",
+		AgentID:              agent.ID,
+		Icon:                 "/code.svg",
+		Command:              sql.NullString{String: "some-command", Valid: true},
+		Url:                  sql.NullString{String: "http://localhost:3000", Valid: true},
+		HealthcheckInterval:  5,
+		HealthcheckUrl:       "http://localhost:3000",
+		HealthcheckThreshold: 6,
+		Health:               database.WorkspaceAppHealthInitializing,
+		SharingLevel:         database.AppSharingLevelOwner,
+		OpenIn:               database.WorkspaceAppOpenInWindow,
+	})
+	require.NoError(t, err)
+
 	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
 	defer cancel()
 
 	agentClient := agentsdk.New(client.URL)
-	agentClient.SetSessionToken(r.AgentToken)
+	agentClient.SetSessionToken(agent.AuthToken.String())
 	conn, err := agentClient.ConnectRPC(ctx)
 	require.NoError(t, err)
 	defer func() {
diff --git a/go.sum b/go.sum
@@ -928,6 +928,7 @@ github.com/coder/tailscale v1.1.1-0.20250611020837-f14d20d23d8c h1:d/qBIi3Ez7Kko
 github.com/coder/tailscale v1.1.1-0.20250611020837-f14d20d23d8c/go.mod h1:l7ml5uu7lFh5hY28lGYM4b/oFSmuPHYX6uk4RAu23Lc=
 github.com/coder/terraform-config-inspect v0.0.0-20250107175719-6d06d90c630e h1:JNLPDi2P73laR1oAclY6jWzAbucf70ASAvf5mh2cME0=
 github.com/coder/terraform-config-inspect v0.0.0-20250107175719-6d06d90c630e/go.mod h1:Gz/z9Hbn+4KSp8A2FBtNszfLSdT2Tn/uAKGuVqqWmDI=
+github.com/coder/terraform-provider-coder/v2 v2.5.4-0.20250619093524-d9b0f892f9c2/go.mod h1:WrdLSbihuzH1RZhwrU+qmkqEhUbdZT/sjHHdarm5b5g=
 github.com/coder/terraform-provider-coder/v2 v2.6.0 h1:ybSVxkblpFdanNX7hibex41yvwjswUlA3RPh4BAHjBI=
 github.com/coder/terraform-provider-coder/v2 v2.6.0/go.mod h1:WrdLSbihuzH1RZhwrU+qmkqEhUbdZT/sjHHdarm5b5g=
 github.com/coder/trivy v0.0.0-20250527170238-9416a59d7019 h1:MHkv/W7l9eRAN9gOG0qZ1TLRGWIIfNi92273vPAQ8Fs=
