Function TestWatchChatGit

coderd/workspaceagents_internal_test.go:165–730

(t *testing.T)

}

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

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

		// This test ensures that a chat with no workspace ID
		// returns a 400 error.

		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()

			r = chi.NewMux()

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

		// Setup: Return a chat with no workspace ID.
		mDB.EXPECT().GetChatByID(gomock.Any(), chatID).Return(database.Chat{
			ID:          chatID,
			OwnerID:     uuid.New(),
			WorkspaceID: uuid.NullUUID{Valid: false},
		}, nil)

		// And: We mount the HTTP handler.
		r.With(httpmw.ExtractChatParam(mDB)).
			Get("/chats/{chat}/stream/git", api.watchChatGit)

		// Given: We create the HTTP server.
		srv := httptest.NewServer(r)
		defer srv.Close()

		// When: We make a request.
		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()

		// Then: We expect a 400 response.
		require.Equal(t, http.StatusBadRequest, resp.StatusCode)
	})