Function TestWorker_MarkStale_WithChatID

coderd/x/gitsync/worker_test.go:816–870

(t *testing.T)

}

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

	targetChat := uuid.New()

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

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

	// GetChatsByWorkspaceIDs should NOT be called when a specific chat ID is provided.
	store.EXPECT().GetChatsByWorkspaceIDs(gomock.Any(), gomock.Any()).Times(0)
	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)
	now := mClock.Now()
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
	refresher := newTestRefresher(t, mClock)
	worker := gitsync.NewWorker(store, refresher, pub, mClock, logger)

	worker.MarkStale(ctx, gitsync.MarkStaleParams{
		WorkspaceID: uuid.New(),
		Branch:      "my-branch",
		Origin:      "https://github.com/org/repo",
		ChatID:      targetChat,
	})

	mu.Lock()
	defer mu.Unlock()

	require.Len(t, upsertRefCalls, 1)
	assert.Equal(t, targetChat, upsertRefCalls[0].ChatID)
	assert.Equal(t, "my-branch", upsertRefCalls[0].GitBranch)
	assert.Equal(t, "https://github.com/org/repo", upsertRefCalls[0].GitRemoteOrigin)
	assert.True(t, upsertRefCalls[0].StaleAt.Before(now),
		"stale_at should be in the past, got %v vs now %v", upsertRefCalls[0].StaleAt, now)

	require.Len(t, publishedIDs, 1)
	assert.Equal(t, targetChat, publishedIDs[0])
}

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