Function TestWatcher_PublishChanges

coderd/workspaceconnwatcher/watcher_test.go:290–414

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitShort)
	logger := testutil.Logger(t)
	h := newHarness(ctx, t, logger)

	// Initial build update, job is running.
	build0 := h.db.EXPECT().GetLatestWorkspaceBuildWithStatusByWorkspaceID(gomock.Any(), h.workspace.ID).
		Times(1).
		Return(database.GetLatestWorkspaceBuildWithStatusByWorkspaceIDRow{
			Transition:  database.WorkspaceTransitionStart,
			BuildNumber: 1,
			JobStatus:   database.ProvisionerJobStatusRunning,
			WorkspaceTable: database.WorkspaceTable{
				ID:             h.workspace.ID,
				OwnerID:        userID,
				OrganizationID: orgID,
			},
		}, nil)

	dec, err := h.Dial(ctx, "wss://local.test/")
	require.NoError(t, err)
	defer func() {
		_ = dec.Close()
	}()
	events := dec.Chan()

	e0 := testutil.RequireReceive(ctx, t, events)
	require.Equal(t, workspacesdk.ConnectionWatchEvent{
		BuildUpdate: &workspacesdk.BuildUpdate{
			Transition: codersdk.WorkspaceTransitionStart,
			JobStatus:  codersdk.ProvisionerJobRunning,
		},
	}, e0)

	// Since job is still running, we don't immediately query for agents. Next we set up the db queries and send in an
	// update over the pubsub to kick  a new query.
	build1 := h.db.EXPECT().GetLatestWorkspaceBuildWithStatusByWorkspaceID(gomock.Any(), h.workspace.ID).
		After(build0).
		Times(1).
		Return(database.GetLatestWorkspaceBuildWithStatusByWorkspaceIDRow{
			Transition:  database.WorkspaceTransitionStart,
			BuildNumber: 1,
			JobStatus:   database.ProvisionerJobStatusSucceeded,
			WorkspaceTable: database.WorkspaceTable{
				ID:             h.workspace.ID,
				OwnerID:        userID,
				OrganizationID: orgID,
			},
		}, nil)
	// RBAC check for agent query
	h.db.EXPECT().GetWorkspaceByID(gomock.Any(), h.workspace.ID).
		After(build1).
		Times(2). // these queries are identical between the initial and the update below
		Return(h.workspace, nil)
	agent0 := h.db.EXPECT().GetWorkspaceAgentsByWorkspaceAndBuildNumber(
		gomock.Any(),