investbrain/tests/ChatWithTest.php

<?php

declare(strict_types=1);

namespace Tests\Feature;

use App\Ai\Agents\ChatWithPortfolioAgent;
use App\Ai\Agents\ChatWithSuggestedPromptsAgent;
use App\Models\ChatWithConversation;
use App\Models\Portfolio;
use App\Models\User;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\RateLimiter;
use Illuminate\Support\Str;
use Livewire\Volt\Volt;
use Tests\TestCase;

class ChatWithTest extends TestCase
{
    use RefreshDatabase;

    public function test_mount_creates_chat_with_conversation_record(): void
    {
        ChatWithPortfolioAgent::fake(['Test response']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs($user = User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        $this->assertDatabaseCount('agent_conversations', 0);

        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);

        $this->assertDatabaseHas('agent_conversations', [
            'chatable_type' => Portfolio::class,
            'chatable_id' => $portfolio->id,
            'user_id' => $user->id,
        ]);
    }

    public function test_mount_reuses_existing_chat_with_conversation(): void
    {
        ChatWithPortfolioAgent::fake(['Test response']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs(User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);
        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);

        $this->assertDatabaseCount('agent_conversations', 1);
    }

    public function test_mount_loads_existing_messages(): void
    {
        ChatWithPortfolioAgent::fake(['Test response']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs($user = User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        $conversation = ChatWithConversation::create([
            'chatable_type' => Portfolio::class,
            'chatable_id' => $portfolio->id,
            'user_id' => $user->id,
            'title' => 'Chat with investments',
        ]);

        DB::table('agent_conversation_messages')->insert([
            'id' => (string) Str::uuid7(),
            'conversation_id' => $conversation->id,
            'user_id' => $user->id,
            'agent' => ChatWithPortfolioAgent::class,
            'role' => 'user',
            'content' => 'Previous question',
            'attachments' => '[]',
            'tool_calls' => '[]',
            'tool_results' => '[]',
            'usage' => '[]',
            'meta' => '[]',
            'created_at' => now(),
            'updated_at' => now(),
        ]);

        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);

        $messages = $component->get('messages');
        $this->assertCount(1, $messages);
        $this->assertEquals('user', $messages[0]['role']);
        $this->assertEquals('Previous question', $messages[0]['content']);
    }

    public function test_start_completion_adds_user_message_and_sets_streaming(): void
    {
        ChatWithPortfolioAgent::fake(['The portfolio looks great!']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs(User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
            ->set('prompt', 'How is my portfolio doing?')
            ->call('startCompletion')
            ->assertSet('streaming', true)
            ->assertSee('How is my portfolio doing?');
    }

    public function test_generate_completion_calls_agent_and_stores_response(): void
    {
        ChatWithPortfolioAgent::fake(['The portfolio looks great!']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs(User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
            ->set('messages', [['role' => 'user', 'content' => 'How is my portfolio doing?', 'created_at' => now()]])
            ->call('generateCompletion')
            ->assertSet('streaming', false);

        ChatWithPortfolioAgent::assertPrompted('How is my portfolio doing?');

        $this->assertDatabaseHas('agent_conversation_messages', [
            'role' => 'assistant',
            'content' => 'The portfolio looks great!',
        ]);
    }

    public function test_empty_prompt_returns_guidance_without_calling_agent(): void
    {
        ChatWithPortfolioAgent::fake(['Test response']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs(User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
            ->set('prompt', '   ')
            ->call('startCompletion');

        $lastMessage = collect($component->get('messages'))->last()['content'];
        $this->assertStringContainsString('Feel free to ask me a question!', $lastMessage);
        ChatWithPortfolioAgent::assertNeverPrompted();
    }

    public function test_rate_limiting_blocks_excessive_requests(): void
    {
        ChatWithPortfolioAgent::fake(['Test response']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs($user = User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        for ($i = 0; $i < 20; $i++) {
            RateLimiter::hit($user->id.'/'.$portfolio->id, 60);
        }

        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
            ->set('prompt', 'Am I rate limited?')
            ->call('startCompletion');

        $lastMessage = collect($component->get('messages'))->last()['content'];
        $this->assertStringContainsString("Hang on! You're doing that too much.", $lastMessage);
        ChatWithPortfolioAgent::assertNeverPrompted();
    }

    public function test_suggested_prompt_is_used_as_prompt(): void
    {
        ChatWithPortfolioAgent::fake(['Here is the best holding!']);
        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);

        $this->actingAs(User::factory()->create());
        $portfolio = Portfolio::factory()->create();

        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
            ->call('startCompletion', 'Which holding is most successful in this portfolio?')
            ->assertSet('streaming', true)
            ->assertSee('Which holding is most successful in this portfolio?');
    }
}