Function TestWorkspaceAgentAppStatus

coderd/agentapi/apps_test.go:253–429

(t *testing.T)

}

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

	t.Run("Success", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitShort)
		ctrl := gomock.NewController(t)
		mDB := dbmock.NewMockStore(ctrl)
		fEnq := &notificationstest.FakeEnqueuer{}
		mClock := quartz.NewMock(t)
		agent := database.WorkspaceAgent{
			ID:             uuid.UUID{2},
			LifecycleState: database.WorkspaceAgentLifecycleStateReady,
		}
		workspaceUpdates := make(chan wspubsub.WorkspaceEventKind, 100)

		workspace := database.Workspace{
			ID: uuid.UUID{9},
			TaskID: uuid.NullUUID{
				Valid: true,
				UUID:  uuid.UUID{7},
			},
		}
		cachedWs := &agentapi.CachedWorkspaceFields{}
		cachedWs.UpdateValues(workspace)

		api := &agentapi.AppsAPI{
			AgentID: agent.ID,
			AgentFn: func(context.Context) (database.WorkspaceAgent, error) {
				return agent, nil
			},
			Database:  mDB,
			Log:       testutil.Logger(t),
			Workspace: cachedWs,
			PublishWorkspaceUpdateFn: func(_ context.Context, agnt uuid.UUID, kind wspubsub.WorkspaceEventKind) error {
				assert.Equal(t, agnt, agent.ID)
				testutil.AssertSend(ctx, t, workspaceUpdates, kind)
				return nil
			},
			NotificationsEnqueuer: fEnq,
			Clock:                 mClock,
		}

		app := database.WorkspaceApp{
			ID: uuid.UUID{8},
		}
		mDB.EXPECT().GetWorkspaceAppByAgentIDAndSlug(gomock.Any(), database.GetWorkspaceAppByAgentIDAndSlugParams{
			AgentID: agent.ID,
			Slug:    "vscode",
		}).Times(1).Return(app, nil)
		task := database.Task{
			ID: uuid.UUID{7},
			WorkspaceAppID: uuid.NullUUID{
				Valid: true,
				UUID:  app.ID,
			},
		}