Function TestRecoverStaleRequiresActionChat

coderd/x/chatd/chatd_test.go:3778–3844

(t *testing.T)

}

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

	db, ps, rawDB := dbtestutil.NewDBWithSQLDB(t)

	ctx := testutil.Context(t, testutil.WaitLong)
	user, org, model := seedChatDependencies(t, db)

	// Use a very short stale threshold so the periodic recovery
	// kicks in quickly during the test.
	staleAfter := 500 * time.Millisecond

	// Create a chat and set it to requires_action to simulate a
	// client that disappeared while the chat was waiting for
	// dynamic tool results.
	chat := dbgen.Chat(t, db, database.Chat{
		OrganizationID:    org.ID,
		OwnerID:           user.ID,
		Title:             "stale-requires-action",
		LastModelConfigID: model.ID,
	})

	_, err := db.UpdateChatStatus(ctx, database.UpdateChatStatusParams{
		ID:     chat.ID,
		Status: database.ChatStatusRequiresAction,
	})
	require.NoError(t, err)

	// Backdate updated_at so the chat appears stale to the
	// recovery loop without needing time.Sleep.
	_, err = rawDB.ExecContext(ctx,
		"UPDATE chats SET updated_at = $1 WHERE id = $2",
		time.Now().Add(-time.Hour), chat.ID)
	require.NoError(t, err)

	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
	server := chatd.New(chatd.Config{
		Logger:                     logger,
		Database:                   db,
		ReplicaID:                  uuid.New(),
		Pubsub:                     ps,
		PendingChatAcquireInterval: testutil.WaitLong,
		InFlightChatStaleAfter:     staleAfter,
	})
	server.Start()
	t.Cleanup(func() {
		require.NoError(t, server.Close())
	})

	// The stale recovery should transition the requires_action
	// chat to error with the timeout message.
	var chatResult database.Chat
	require.Eventually(t, func() bool {
		chatResult, err = db.GetChatByID(ctx, chat.ID)
		if err != nil {
			return false
		}
		return chatResult.Status == database.ChatStatusError