Function TestService_PublishesEvents

coderd/x/chatd/chatdebug/service_test.go:1016–1065

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitLong)
	db, _ := dbtestutil.NewDB(t)
	_, owner, chat, model := seedChat(t, db)
	require.NotEqual(t, uuid.Nil, owner.ID)

	memoryPubsub := dbpubsub.NewInMemory()
	svc := chatdebug.NewService(db, testutil.Logger(t), memoryPubsub)
	type eventResult struct {
		event chatdebug.DebugEvent
		err   error
	}
	events := make(chan eventResult, 1)
	cancel, err := memoryPubsub.Subscribe(chatdebug.PubsubChannel(chat.ID),
		func(_ context.Context, message []byte) {
			var event chatdebug.DebugEvent
			unmarshalErr := json.Unmarshal(message, &event)
			events <- eventResult{event: event, err: unmarshalErr}
		},
	)
	require.NoError(t, err)
	defer cancel()

	run, err := svc.CreateRun(ctx, chatdebug.CreateRunParams{
		ChatID:        chat.ID,
		ModelConfigID: model.ID,
		Kind:          chatdebug.KindChatTurn,
		Status:        chatdebug.StatusInProgress,
	})
	require.NoError(t, err)

	select {
	case received := <-events:
		require.NoError(t, received.err)
		require.Equal(t, chatdebug.EventKindRunUpdate, received.event.Kind)
		require.Equal(t, chat.ID, received.event.ChatID)
		require.Equal(t, run.ID, received.event.RunID)
		require.Equal(t, uuid.Nil, received.event.StepID)
	case <-time.After(testutil.WaitShort):
		t.Fatal("timed out waiting for debug event")
	}

	select {
	case received := <-events:
		t.Fatalf("unexpected extra event: %+v", received.event)
	default:
	}
}

func newFixture(t *testing.T) testFixture {
	t.Helper()