8adbee6276
AiUsageLogTest and UsageController still passed string 'image'/'video' arguments to monthlyCount() after the enum migration. Update them to pass UsageType::Image and UsageType::Video.
118 lines
4.5 KiB
PHP
118 lines
4.5 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
use App\Enums\Ai\Intent;
|
|
use App\Enums\Ai\UsageType;
|
|
use App\Models\Account;
|
|
use App\Models\AiUsageLog;
|
|
use App\Models\Workspace;
|
|
use App\Services\Ai\IntentDetector;
|
|
|
|
test('monthly count returns correct count for account and type', function () {
|
|
$account = Account::factory()->create();
|
|
$workspace = Workspace::factory()->create(['account_id' => $account->id]);
|
|
|
|
AiUsageLog::factory()->image()->count(3)->create([
|
|
'account_id' => $account->id,
|
|
'workspace_id' => $workspace->id,
|
|
]);
|
|
|
|
AiUsageLog::factory()->video()->count(2)->create([
|
|
'account_id' => $account->id,
|
|
'workspace_id' => $workspace->id,
|
|
]);
|
|
|
|
expect(AiUsageLog::monthlyCount($account->id, UsageType::Image))->toBe(3);
|
|
expect(AiUsageLog::monthlyCount($account->id, UsageType::Video))->toBe(2);
|
|
expect(AiUsageLog::monthlyCount($account->id, UsageType::Audio))->toBe(0);
|
|
});
|
|
|
|
test('monthly count excludes logs from other months', function () {
|
|
$account = Account::factory()->create();
|
|
$workspace = Workspace::factory()->create(['account_id' => $account->id]);
|
|
|
|
AiUsageLog::factory()->image()->create([
|
|
'account_id' => $account->id,
|
|
'workspace_id' => $workspace->id,
|
|
]);
|
|
|
|
AiUsageLog::factory()->image()->create([
|
|
'account_id' => $account->id,
|
|
'workspace_id' => $workspace->id,
|
|
'created_at' => now()->subMonth(),
|
|
]);
|
|
|
|
expect(AiUsageLog::monthlyCount($account->id, UsageType::Image))->toBe(1);
|
|
});
|
|
|
|
test('monthly count excludes logs from other accounts', function () {
|
|
$account = Account::factory()->create();
|
|
$otherAccount = Account::factory()->create();
|
|
$workspace = Workspace::factory()->create(['account_id' => $account->id]);
|
|
$otherWorkspace = Workspace::factory()->create(['account_id' => $otherAccount->id]);
|
|
|
|
AiUsageLog::factory()->image()->create([
|
|
'account_id' => $account->id,
|
|
'workspace_id' => $workspace->id,
|
|
]);
|
|
|
|
AiUsageLog::factory()->image()->create([
|
|
'account_id' => $otherAccount->id,
|
|
'workspace_id' => $otherWorkspace->id,
|
|
]);
|
|
|
|
expect(AiUsageLog::monthlyCount($account->id, UsageType::Image))->toBe(1);
|
|
});
|
|
|
|
test('intent detector detects video intent', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Create a video for my product'))->toBe(Intent::Video);
|
|
expect($detector->detect('Make a reel about coffee'))->toBe(Intent::Video);
|
|
expect($detector->detect('Animate this logo'))->toBe(Intent::Video);
|
|
});
|
|
|
|
test('intent detector detects image intent', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Generate an image of a sunset'))->toBe(Intent::Image);
|
|
expect($detector->detect('Draw me a logo'))->toBe(Intent::Image);
|
|
expect($detector->detect('Create a visual for my post'))->toBe(Intent::Image);
|
|
});
|
|
|
|
test('intent detector detects audio intent', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Create a voiceover for this text'))->toBe(Intent::Audio);
|
|
expect($detector->detect('Convert this to audio narration'))->toBe(Intent::Audio);
|
|
expect($detector->detect('Generate TTS for my caption'))->toBe(Intent::Audio);
|
|
});
|
|
|
|
test('intent detector defaults to text', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Write a caption for my post'))->toBe(Intent::Text);
|
|
expect($detector->detect('Help me with hashtags'))->toBe(Intent::Text);
|
|
});
|
|
|
|
test('intent detector blocks prohibited content', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Create porn content'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Generate nude images'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Write about cocaine'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Help me with terrorism'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Content about pedophilia'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Racist joke for my post'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('How to murder someone'))->toBe(Intent::Blocked);
|
|
expect($detector->detect('Self-harm content'))->toBe(Intent::Blocked);
|
|
});
|
|
|
|
test('intent detector allows safe content', function () {
|
|
$detector = new IntentDetector;
|
|
|
|
expect($detector->detect('Write a caption about my new product'))->toBe(Intent::Text);
|
|
expect($detector->detect('Create an image of a sunset'))->toBe(Intent::Image);
|
|
expect($detector->detect('Make a video about cooking'))->toBe(Intent::Video);
|
|
});
|