125 lines
4.4 KiB
PHP
125 lines
4.4 KiB
PHP
<?php
|
|
|
|
namespace Tests\Unit\Services;
|
|
|
|
use App\Models\AiEditRequest;
|
|
use App\Models\AiStyle;
|
|
use App\Models\Event;
|
|
use App\Models\EventPackage;
|
|
use App\Models\EventPackageAddon;
|
|
use App\Models\Package;
|
|
use App\Models\Photo;
|
|
use App\Services\AiEditing\AiUsageLedgerService;
|
|
use Illuminate\Foundation\Testing\RefreshDatabase;
|
|
use Tests\TestCase;
|
|
|
|
class AiUsageLedgerServiceTest extends TestCase
|
|
{
|
|
use RefreshDatabase;
|
|
|
|
public function test_it_records_package_included_context_when_feature_is_in_package(): void
|
|
{
|
|
$event = Event::factory()->create();
|
|
$package = Package::factory()->endcustomer()->create([
|
|
'features' => ['ai_styling'],
|
|
]);
|
|
EventPackage::query()->create([
|
|
'event_id' => $event->id,
|
|
'package_id' => $package->id,
|
|
'purchased_price' => $package->price,
|
|
'purchased_at' => now(),
|
|
'gallery_expires_at' => now()->addDays(14),
|
|
]);
|
|
|
|
$style = AiStyle::query()->create([
|
|
'key' => 'ledger-style-package',
|
|
'name' => 'Ledger Package',
|
|
'provider' => 'runware',
|
|
'provider_model' => 'runware-default',
|
|
'is_active' => true,
|
|
]);
|
|
$photo = Photo::factory()->for($event)->create([
|
|
'tenant_id' => $event->tenant_id,
|
|
'status' => 'approved',
|
|
]);
|
|
$request = AiEditRequest::query()->create([
|
|
'tenant_id' => $event->tenant_id,
|
|
'event_id' => $event->id,
|
|
'photo_id' => $photo->id,
|
|
'style_id' => $style->id,
|
|
'provider' => 'runware',
|
|
'provider_model' => 'runware-default',
|
|
'status' => AiEditRequest::STATUS_SUCCEEDED,
|
|
'safety_state' => 'passed',
|
|
'prompt' => 'Package context.',
|
|
'idempotency_key' => 'ledger-package-context-1',
|
|
'queued_at' => now(),
|
|
'completed_at' => now(),
|
|
]);
|
|
|
|
$ledger = app(AiUsageLedgerService::class)->recordDebitForRequest($request, 0.01234);
|
|
|
|
$this->assertSame('package_included', $ledger->package_context);
|
|
$this->assertSame('0.01234', $ledger->amount_usd);
|
|
$this->assertSame('runware', $ledger->metadata['provider'] ?? null);
|
|
}
|
|
|
|
public function test_it_records_addon_context_and_is_idempotent_per_request(): void
|
|
{
|
|
$event = Event::factory()->create();
|
|
$package = Package::factory()->endcustomer()->create([
|
|
'features' => ['basic_uploads'],
|
|
]);
|
|
$eventPackage = EventPackage::query()->create([
|
|
'event_id' => $event->id,
|
|
'package_id' => $package->id,
|
|
'purchased_price' => $package->price,
|
|
'purchased_at' => now(),
|
|
'gallery_expires_at' => now()->addDays(14),
|
|
]);
|
|
EventPackageAddon::query()->create([
|
|
'event_package_id' => $eventPackage->id,
|
|
'event_id' => $event->id,
|
|
'tenant_id' => $event->tenant_id,
|
|
'addon_key' => 'ai_styling_unlock',
|
|
'quantity' => 1,
|
|
'status' => 'completed',
|
|
'purchased_at' => now(),
|
|
]);
|
|
|
|
$style = AiStyle::query()->create([
|
|
'key' => 'ledger-style-addon',
|
|
'name' => 'Ledger Addon',
|
|
'provider' => 'runware',
|
|
'provider_model' => 'runware-default',
|
|
'is_active' => true,
|
|
]);
|
|
$photo = Photo::factory()->for($event)->create([
|
|
'tenant_id' => $event->tenant_id,
|
|
'status' => 'approved',
|
|
]);
|
|
$request = AiEditRequest::query()->create([
|
|
'tenant_id' => $event->tenant_id,
|
|
'event_id' => $event->id,
|
|
'photo_id' => $photo->id,
|
|
'style_id' => $style->id,
|
|
'provider' => 'runware',
|
|
'provider_model' => 'runware-default',
|
|
'status' => AiEditRequest::STATUS_SUCCEEDED,
|
|
'safety_state' => 'passed',
|
|
'prompt' => 'Addon context.',
|
|
'idempotency_key' => 'ledger-addon-context-1',
|
|
'queued_at' => now(),
|
|
'completed_at' => now(),
|
|
]);
|
|
|
|
$service = app(AiUsageLedgerService::class);
|
|
$first = $service->recordDebitForRequest($request, 0.01);
|
|
$second = $service->recordDebitForRequest($request, 0.99);
|
|
|
|
$this->assertSame($first->id, $second->id);
|
|
$this->assertSame('addon_unlock', $first->package_context);
|
|
$this->assertSame(1, $request->usageLedgers()->count());
|
|
}
|
|
}
|