Function runWatchChatGitWorkspaceLookupTest

coderd/workspaceagents_internal_test.go:110–163

runWatchChatGitWorkspaceLookupTest exercises the GetWorkspaceByID error branches in authorizeChatWorkspaceExec. The chat middleware always succeeds; the workspace lookup returns workspaceErr, and the handler is expected to respond with wantStatus.

(t *testing.T, workspaceErr error, wantStatus int)

// always succeeds; the workspace lookup returns workspaceErr, and the
// handler is expected to respond with wantStatus.
func runWatchChatGitWorkspaceLookupTest(t *testing.T, workspaceErr error, wantStatus int) {
	t.Helper()

	var (
		ctx    = testutil.Context(t, testutil.WaitShort)
		logger = slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug).Named("coderd")

		mCtrl = gomock.NewController(t)
		mDB   = dbmock.NewMockStore(mCtrl)

		chatID      = uuid.New()
		workspaceID = uuid.New()

		r = chi.NewMux()

		api = API{
			ctx: ctx,
			Options: &Options{
				AgentInactiveDisconnectTimeout: testutil.WaitShort,
				Database:                       mDB,
				Logger:                         logger,
				DeploymentValues:               &codersdk.DeploymentValues{},
			},
			HTTPAuth: &HTTPAuthorizer{
				Authorizer: &mockAuthorizer{},
				Logger:     logger,
			},
		}
	)

	mDB.EXPECT().GetChatByID(gomock.Any(), chatID).Return(database.Chat{
		ID:          chatID,
		OwnerID:     uuid.New(),
		WorkspaceID: uuid.NullUUID{UUID: workspaceID, Valid: true},
	}, nil)

	mDB.EXPECT().GetWorkspaceByID(gomock.Any(), workspaceID).Return(database.Workspace{}, workspaceErr)

	r.With(injectSystemActor, httpmw.ExtractChatParam(mDB)).
		Get("/chats/{chat}/stream/git", api.watchChatGit)

	srv := httptest.NewServer(r)
	defer srv.Close()

	req, err := http.NewRequestWithContext(ctx, http.MethodGet,
		fmt.Sprintf("%s/chats/%s/stream/git", srv.URL, chatID), nil)
	require.NoError(t, err)

	resp, err := http.DefaultClient.Do(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	require.Equal(t, wantStatus, resp.StatusCode)
}

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