Function TestRefreshChat_Success

coderd/x/gitsync/worker_test.go:1015–1062

(t *testing.T)

}

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

	chatID := uuid.New()
	ownerID := uuid.New()

	row := database.ChatDiffStatus{
		ChatID:          chatID,
		GitBranch:       "feature",
		GitRemoteOrigin: "https://github.com/owner/repo",
	}

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

	upsertedStatus := database.ChatDiffStatus{
		ChatID:       chatID,
		Url:          sql.NullString{String: "https://github.com/o/r/pull/1", Valid: true},
		Additions:    10,
		Deletions:    3,
		ChangedFiles: 2,
	}
	store.EXPECT().UpsertChatDiffStatus(gomock.Any(), gomock.Any()).
		DoAndReturn(func(_ context.Context, arg database.UpsertChatDiffStatusParams) (database.ChatDiffStatus, error) {
			assert.Equal(t, chatID, arg.ChatID)
			return upsertedStatus, nil
		})

	var publishCalled atomic.Bool
	pub := func(_ context.Context, id uuid.UUID) error {
		assert.Equal(t, chatID, id)
		publishCalled.Store(true)
		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)

	result, err := worker.RefreshChat(ctx, row, ownerID)
	require.NoError(t, err)
	require.NotNil(t, result)
	assert.Equal(t, chatID, result.ChatID)
	assert.Equal(t, upsertedStatus.Url, result.Url)
	assert.True(t, publishCalled.Load(), "publish should have been called")
}

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