Function TestWorker_MarkStale_UpsertAndPublish

coderd/x/gitsync/worker_test.go:600–664

(t *testing.T)

}

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

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

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

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

	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}},
				{ID: chat2, 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(2)

	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: workspaceID,
		Branch:      "feature",
		Origin:      "https://github.com/owner/repo",
	})

	mu.Lock()
	defer mu.Unlock()

	require.Len(t, upsertRefCalls, 2)
	for _, call := range upsertRefCalls {
		assert.Equal(t, "feature", call.GitBranch)
		assert.Equal(t, "https://github.com/owner/repo", call.GitRemoteOrigin)
		assert.True(t, call.StaleAt.Before(now),