Function TestReinit

coderd/workspaceagents_test.go:3368–3617

(t *testing.T)

}

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

	// Helper to create the prebuilds system user's workspace (an
	// unclaimed prebuild) and return the build result. The first
	// build's InitiatorID defaults to PrebuildsSystemUserID via
	// dbfake.
	setupPrebuildWorkspace := func(t *testing.T, db database.Store, orgID uuid.UUID) dbfake.WorkspaceResponse {
		t.Helper()
		return dbfake.WorkspaceBuild(t, db, database.WorkspaceTable{
			OrganizationID: orgID,
			OwnerID:        database.PrebuildsSystemUserID,
		}).WithAgent().Do()
	}

	// Helper to simulate claiming a prebuild: change the workspace
	// owner to the real user and create a second (claim) build.
	claimPrebuild := func(t *testing.T, db database.Store, sqlDB *sql.DB, ws database.WorkspaceTable, claimerID uuid.UUID, templateVersionID uuid.UUID, complete bool) dbfake.WorkspaceResponse {
		t.Helper()
		// Change the workspace owner to the claiming user.
		_, err := sqlDB.Exec("UPDATE workspaces SET owner_id = $1 WHERE id = $2", claimerID, ws.ID)
		require.NoError(t, err)

		// Update the in-memory workspace to reflect the new owner
		// so that dbfake uses it for the second build.
		ws.OwnerID = claimerID

		builder := dbfake.WorkspaceBuild(t, db, ws).
			Seed(database.WorkspaceBuild{
				TemplateVersionID: templateVersionID,
				BuildNumber:       2,
				InitiatorID:       claimerID,
				Transition:        database.WorkspaceTransitionStart,
			}).
			MarkPrebuiltWorkspaceClaim().
			WithAgent()
		if !complete {
			builder = builder.Starting()
		}
		return builder.Do()
	}

	t.Run("unclaimed prebuild receives reinit via pubsub", func(t *testing.T) {
		t.Parallel()

		db, ps := dbtestutil.NewDB(t)
		pubsubSpy := pubsubReinitSpy{
			Pubsub:           ps,
			triedToSubscribe: make(chan string),
		}
		client := coderdtest.New(t, &coderdtest.Options{
			Database: db,
			Pubsub:   &pubsubSpy,
		})
		user := coderdtest.CreateFirstUser(t, client)

		r := setupPrebuildWorkspace(t, db, user.OrganizationID)