412 lines
14 KiB
PHP
412 lines
14 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
use App\Ai\Agents\SocialMediaAssistant;
|
|
use App\Ai\Tools\AttachmentCollector;
|
|
use App\Enums\AiMessage\Status;
|
|
use App\Enums\UserWorkspace\Role;
|
|
use App\Events\Ai\AssistantMessageUpdated;
|
|
use App\Jobs\Ai\GenerateAssistantResponse;
|
|
use App\Models\AiMessage;
|
|
use App\Models\AiUsageLog;
|
|
use App\Models\Post;
|
|
use App\Models\User;
|
|
use App\Models\Workspace;
|
|
use App\Services\Ai\HumanizerService;
|
|
use App\Services\Ai\IntentDetector;
|
|
use Illuminate\Support\Facades\Bus;
|
|
use Illuminate\Support\Facades\Event;
|
|
|
|
beforeEach(function () {
|
|
$this->user = User::factory()->create([]);
|
|
$this->workspace = Workspace::factory()->create(['user_id' => $this->user->id]);
|
|
$this->workspace->members()->attach($this->user->id, ['role' => Role::Member->value]);
|
|
$this->user->update(['current_workspace_id' => $this->workspace->id]);
|
|
|
|
$this->post = Post::factory()->create([
|
|
'workspace_id' => $this->workspace->id,
|
|
'user_id' => $this->user->id,
|
|
]);
|
|
|
|
app(AttachmentCollector::class)->clear();
|
|
});
|
|
|
|
test('index returns ai messages for a post', function () {
|
|
AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => $this->user->id,
|
|
'role' => 'user',
|
|
'content' => 'Write a caption',
|
|
]);
|
|
|
|
AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => 'Here is your caption!',
|
|
]);
|
|
|
|
$response = $this->actingAs($this->user)
|
|
->getJson(route('app.posts.assistant.index', $this->post));
|
|
|
|
$response->assertOk();
|
|
$response->assertJsonCount(2, 'messages');
|
|
});
|
|
|
|
test('index rejects access to post from other workspace', function () {
|
|
$otherUser = User::factory()->create([]);
|
|
$otherWorkspace = Workspace::factory()->create(['user_id' => $otherUser->id]);
|
|
$otherWorkspace->members()->attach($otherUser->id, ['role' => Role::Member->value]);
|
|
$otherUser->update(['current_workspace_id' => $otherWorkspace->id]);
|
|
|
|
$response = $this->actingAs($otherUser)
|
|
->getJson(route('app.posts.assistant.index', $this->post));
|
|
|
|
$response->assertForbidden();
|
|
});
|
|
|
|
test('store creates user message and pending assistant placeholder, dispatches job', function () {
|
|
Bus::fake();
|
|
|
|
$response = $this->actingAs($this->user)
|
|
->postJson(route('app.posts.assistant.store', $this->post), [
|
|
'body' => 'Write me a caption about summer vibes',
|
|
]);
|
|
|
|
$response->assertAccepted();
|
|
$response->assertJsonPath('user_message.content', 'Write me a caption about summer vibes');
|
|
$response->assertJsonPath('user_message.role', 'user');
|
|
$response->assertJsonPath('assistant_message.role', 'assistant');
|
|
$response->assertJsonPath('assistant_message.status', 'pending');
|
|
$response->assertJsonPath('assistant_message.content', '');
|
|
|
|
$this->assertDatabaseHas('ai_messages', [
|
|
'post_id' => $this->post->id,
|
|
'role' => 'user',
|
|
'content' => 'Write me a caption about summer vibes',
|
|
]);
|
|
|
|
$this->assertDatabaseHas('ai_messages', [
|
|
'post_id' => $this->post->id,
|
|
'role' => 'assistant',
|
|
'status' => 'pending',
|
|
]);
|
|
|
|
Bus::assertDispatched(GenerateAssistantResponse::class, function (GenerateAssistantResponse $job) {
|
|
return $job->prompt === 'Write me a caption about summer vibes'
|
|
&& $job->assistantMessage->post_id === $this->post->id;
|
|
});
|
|
});
|
|
|
|
test('store validates body is required', function () {
|
|
$response = $this->actingAs($this->user)
|
|
->postJson(route('app.posts.assistant.store', $this->post), [
|
|
'body' => '',
|
|
]);
|
|
|
|
$response->assertUnprocessable();
|
|
$response->assertJsonValidationErrors('body');
|
|
});
|
|
|
|
test('store rejects access to post from other workspace', function () {
|
|
$otherUser = User::factory()->create([]);
|
|
$otherWorkspace = Workspace::factory()->create(['user_id' => $otherUser->id]);
|
|
$otherWorkspace->members()->attach($otherUser->id, ['role' => Role::Member->value]);
|
|
$otherUser->update(['current_workspace_id' => $otherWorkspace->id]);
|
|
|
|
$response = $this->actingAs($otherUser)
|
|
->postJson(route('app.posts.assistant.store', $this->post), [
|
|
'body' => 'Write me a caption',
|
|
]);
|
|
|
|
$response->assertForbidden();
|
|
});
|
|
|
|
test('store blocks prohibited content via IntentDetector and does not dispatch job', function () {
|
|
Bus::fake();
|
|
|
|
$response = $this->actingAs($this->user)
|
|
->postJson(route('app.posts.assistant.store', $this->post), [
|
|
'body' => 'Create porn content for my post',
|
|
]);
|
|
|
|
$response->assertCreated();
|
|
$response->assertJsonPath('assistant_message.metadata.intent', 'blocked');
|
|
$response->assertJsonPath('assistant_message.metadata.error', true);
|
|
$response->assertJsonPath('assistant_message.status', 'completed');
|
|
|
|
Bus::assertNotDispatched(GenerateAssistantResponse::class);
|
|
|
|
$this->assertDatabaseHas('ai_messages', [
|
|
'post_id' => $this->post->id,
|
|
'role' => 'user',
|
|
'content' => 'Create porn content for my post',
|
|
]);
|
|
});
|
|
|
|
test('store blocks drug related content', function () {
|
|
Bus::fake();
|
|
|
|
$response = $this->actingAs($this->user)
|
|
->postJson(route('app.posts.assistant.store', $this->post), [
|
|
'body' => 'Write about cocaine usage',
|
|
]);
|
|
|
|
$response->assertCreated();
|
|
$response->assertJsonPath('assistant_message.metadata.intent', 'blocked');
|
|
|
|
Bus::assertNotDispatched(GenerateAssistantResponse::class);
|
|
});
|
|
|
|
test('GenerateAssistantResponse job populates message with agent response and broadcasts', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
SocialMediaAssistant::fake([['message' => 'Here is a great caption!', 'quick_actions' => []]]);
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Write a caption',
|
|
intent: 'text',
|
|
))->handle(app(IntentDetector::class), app(AttachmentCollector::class), app(HumanizerService::class));
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->content)->toBe('Here is a great caption!');
|
|
expect($assistantMessage->status)->toBe(Status::Completed);
|
|
expect($assistantMessage->metadata['intent'] ?? null)->toBe('text');
|
|
|
|
Event::assertDispatched(AssistantMessageUpdated::class, fn ($e) => $e->message->id === $assistantMessage->id);
|
|
});
|
|
|
|
test('GenerateAssistantResponse job surfaces attachments from collector', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
$collector = app(AttachmentCollector::class);
|
|
|
|
SocialMediaAssistant::fake(function () use ($collector) {
|
|
$collector->push([
|
|
'id' => 'test-media-id',
|
|
'path' => 'medias/test.png',
|
|
'url' => 'https://example.com/medias/test.png',
|
|
'mime_type' => 'image/png',
|
|
'type' => 'image',
|
|
]);
|
|
|
|
return ['message' => 'Generated an image', 'quick_actions' => []];
|
|
});
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Generate an image',
|
|
intent: 'image',
|
|
))->handle(app(IntentDetector::class), $collector, app(HumanizerService::class));
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->attachments)->toHaveCount(1);
|
|
expect($assistantMessage->attachments[0]['id'])->toBe('test-media-id');
|
|
expect($assistantMessage->metadata['intent'] ?? null)->toBe('image');
|
|
});
|
|
|
|
test('GenerateAssistantResponse failed handler marks message failed and broadcasts', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Generating,
|
|
]);
|
|
|
|
$job = new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Anything',
|
|
intent: 'text',
|
|
);
|
|
|
|
$job->failed(new RuntimeException('API quota exceeded'));
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->status)->toBe(Status::Failed);
|
|
expect($assistantMessage->content)->toBe('API quota exceeded');
|
|
expect($assistantMessage->error_message)->toBe('API quota exceeded');
|
|
expect($assistantMessage->metadata['error'] ?? null)->toBeTrue();
|
|
|
|
Event::assertDispatched(AssistantMessageUpdated::class);
|
|
});
|
|
|
|
test('GenerateAssistantResponse job builds prompt with session state', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'role' => 'assistant',
|
|
'attachments' => [
|
|
['id' => 'm1', 'type' => 'image'],
|
|
['id' => 'm2', 'type' => 'image'],
|
|
],
|
|
]);
|
|
|
|
AiUsageLog::factory()->image()->count(2)->create([
|
|
'account_id' => $this->workspace->account_id,
|
|
'workspace_id' => $this->workspace->id,
|
|
]);
|
|
|
|
$capturedPrompt = null;
|
|
SocialMediaAssistant::fake(function ($prompt) use (&$capturedPrompt) {
|
|
$capturedPrompt = (string) $prompt;
|
|
|
|
return ['message' => 'ok', 'quick_actions' => []];
|
|
});
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Write about coffee',
|
|
intent: 'text',
|
|
))->handle(app(IntentDetector::class), app(AttachmentCollector::class), app(HumanizerService::class));
|
|
|
|
expect($capturedPrompt)
|
|
->toContain('Session state')
|
|
->toContain('Images already generated in this conversation: 2')
|
|
->toContain('Monthly quota remaining:');
|
|
});
|
|
|
|
test('GenerateAssistantResponse job runs humanizer when media is generated', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
$collector = app(AttachmentCollector::class);
|
|
|
|
SocialMediaAssistant::fake([function () use ($collector) {
|
|
$collector->push(['id' => 'img-1', 'path' => 'medias/t.png', 'url' => 'https://example.com/t.png', 'mime_type' => 'image/png', 'type' => 'image']);
|
|
|
|
return ['message' => 'Original — text with em-dash.', 'quick_actions' => []];
|
|
}]);
|
|
|
|
$humanizer = Mockery::mock(HumanizerService::class);
|
|
$humanizer->shouldReceive('humanize')
|
|
->once()
|
|
->with('Original — text with em-dash.', Mockery::any())
|
|
->andReturn('Original, text without em-dash.');
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Generate an image',
|
|
intent: 'image',
|
|
))->handle(app(IntentDetector::class), $collector, $humanizer);
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->content)->toBe('Original, text without em-dash.');
|
|
});
|
|
|
|
test('GenerateAssistantResponse job skips humanizer for conversational turns without media', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
SocialMediaAssistant::fake([['message' => 'Hey! What do you want to create?', 'quick_actions' => [['label' => 'Image', 'value' => 'image']]]]);
|
|
|
|
$humanizer = Mockery::mock(HumanizerService::class);
|
|
$humanizer->shouldNotReceive('humanize');
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Hello',
|
|
intent: 'text',
|
|
))->handle(app(IntentDetector::class), app(AttachmentCollector::class), $humanizer);
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->content)->toBe('Hey! What do you want to create?');
|
|
});
|
|
|
|
test('GenerateAssistantResponse persists quick_actions in message metadata', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
SocialMediaAssistant::fake([[
|
|
'message' => 'What format do you want?',
|
|
'quick_actions' => [
|
|
['label' => '📷 Image', 'value' => 'image'],
|
|
['label' => '🎬 Video', 'value' => 'video'],
|
|
],
|
|
]]);
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Create a post',
|
|
intent: 'text',
|
|
))->handle(app(IntentDetector::class), app(AttachmentCollector::class), app(HumanizerService::class));
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->content)->toBe('What format do you want?');
|
|
expect($assistantMessage->metadata['quick_actions'] ?? null)->toBeArray()->toHaveCount(2);
|
|
expect($assistantMessage->metadata['quick_actions'][0]['label'])->toBe('📷 Image');
|
|
expect($assistantMessage->metadata['quick_actions'][0]['value'])->toBe('image');
|
|
});
|
|
|
|
test('GenerateAssistantResponse handles empty quick_actions gracefully', function () {
|
|
Event::fake([AssistantMessageUpdated::class]);
|
|
|
|
SocialMediaAssistant::fake([['message' => 'Polished caption.', 'quick_actions' => []]]);
|
|
|
|
$assistantMessage = AiMessage::factory()->create([
|
|
'post_id' => $this->post->id,
|
|
'user_id' => null,
|
|
'role' => 'assistant',
|
|
'content' => '',
|
|
'status' => Status::Pending,
|
|
]);
|
|
|
|
(new GenerateAssistantResponse(
|
|
assistantMessage: $assistantMessage,
|
|
prompt: 'Polish my caption',
|
|
intent: 'text',
|
|
))->handle(app(IntentDetector::class), app(AttachmentCollector::class), app(HumanizerService::class));
|
|
|
|
$assistantMessage->refresh();
|
|
expect($assistantMessage->metadata['quick_actions'] ?? null)->toBe([]);
|
|
});
|