Function TestWorker_MarkStale_NilChatID_Broadcasts

coderd/x/gitsync/worker_test.go:872–931

(t *testing.T)

}

func TestWorker_MarkStale_NilChatID_Broadcasts(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)

	workspaceID := uuid.New()
	ownerID := uuid.New()
	chat1 := uuid.New()

	var mu sync.Mutex
	var upsertRefCalls []database.UpsertChatDiffStatusReferenceParams
	var publishedIDs []uuid.UUID

	ctrl := gomock.NewController(t)
	store := dbmock.NewMockStore(ctrl)

	// Broadcast path: GetChatsByWorkspaceIDs scopes the query to
	// the workspace directly; no post-filtering needed.
	store.EXPECT().GetChatsByWorkspaceIDs(gomock.Any(), gomock.Any()).
		DoAndReturn(func(_ context.Context, ids []uuid.UUID) ([]database.Chat, error) {
			require.Equal(t, []uuid.UUID{workspaceID}, ids)
			return []database.Chat{
				{ID: chat1, OwnerID: ownerID, WorkspaceID: uuid.NullUUID{UUID: workspaceID, Valid: true}},
			}, nil
		})
	store.EXPECT().UpsertChatDiffStatusReference(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, arg database.UpsertChatDiffStatusReferenceParams) (database.ChatDiffStatus, error) {
		mu.Lock()
		upsertRefCalls = append(upsertRefCalls, arg)
		mu.Unlock()
		return database.ChatDiffStatus{ChatID: arg.ChatID}, nil
	}).Times(1)

	pub := func(_ context.Context, chatID uuid.UUID) error {
		mu.Lock()
		publishedIDs = append(publishedIDs, chatID)
		mu.Unlock()
		return nil
	}

	mClock := quartz.NewMock(t)
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
	refresher := newTestRefresher(t, mClock)
	worker := gitsync.NewWorker(store, refresher, pub, mClock, logger)

	// Zero-value ChatID (uuid.Nil) triggers broadcast.
	worker.MarkStale(ctx, gitsync.MarkStaleParams{
		WorkspaceID: workspaceID,
		Branch:      "main",
		Origin:      "https://github.com/org/repo",
	})

	mu.Lock()
	defer mu.Unlock()

	require.Len(t, upsertRefCalls, 1)
	assert.Equal(t, chat1, upsertRefCalls[0].ChatID)
	assert.Equal(t, "main", upsertRefCalls[0].GitBranch)

	require.Len(t, publishedIDs, 1)