feat(ai-edits): add output storage backfill flow and coverage
This commit is contained in:
121
tests/Feature/Console/AiEditsBackfillStorageCommandTest.php
Normal file
121
tests/Feature/Console/AiEditsBackfillStorageCommandTest.php
Normal file
@@ -0,0 +1,121 @@
|
||||
<?php
|
||||
|
||||
namespace Tests\Feature\Console;
|
||||
|
||||
use App\Models\AiEditOutput;
|
||||
use App\Models\AiEditRequest;
|
||||
use App\Models\AiStyle;
|
||||
use App\Models\Event;
|
||||
use App\Models\Photo;
|
||||
use Illuminate\Foundation\Testing\RefreshDatabase;
|
||||
use Illuminate\Support\Facades\Http;
|
||||
use Illuminate\Support\Facades\Storage;
|
||||
use Tests\TestCase;
|
||||
|
||||
class AiEditsBackfillStorageCommandTest extends TestCase
|
||||
{
|
||||
use RefreshDatabase;
|
||||
|
||||
public function test_it_backfills_storage_for_outputs_without_local_path(): void
|
||||
{
|
||||
config([
|
||||
'ai-editing.outputs.enabled' => true,
|
||||
'ai-editing.outputs.allowed_hosts' => ['cdn.runware.ai'],
|
||||
'filesystems.default' => 'public',
|
||||
'watermark.base.asset' => 'branding/test-watermark.png',
|
||||
]);
|
||||
|
||||
Storage::fake('public');
|
||||
Storage::disk('public')->put('branding/test-watermark.png', $this->tinyPngBinary());
|
||||
|
||||
Http::fake([
|
||||
'https://cdn.runware.ai/*' => Http::response($this->tinyPngBinary(), 200, [
|
||||
'Content-Type' => 'image/png',
|
||||
]),
|
||||
]);
|
||||
|
||||
[$request, $output] = $this->createOutputWithoutStoragePath();
|
||||
|
||||
$this->artisan('ai-edits:backfill-storage --limit=50')
|
||||
->assertExitCode(0);
|
||||
|
||||
$request->refresh();
|
||||
$output->refresh();
|
||||
|
||||
$this->assertNotNull($output->storage_disk);
|
||||
$this->assertNotNull($output->storage_path);
|
||||
$this->assertNotSame('', trim((string) $output->storage_path));
|
||||
$this->assertTrue(Storage::disk((string) $output->storage_disk)->exists((string) $output->storage_path));
|
||||
}
|
||||
|
||||
public function test_it_supports_pretend_mode_without_writing_changes(): void
|
||||
{
|
||||
config([
|
||||
'ai-editing.outputs.enabled' => true,
|
||||
'ai-editing.outputs.allowed_hosts' => ['cdn.runware.ai'],
|
||||
'filesystems.default' => 'public',
|
||||
]);
|
||||
|
||||
[, $output] = $this->createOutputWithoutStoragePath();
|
||||
|
||||
$this->artisan('ai-edits:backfill-storage --pretend --limit=10')
|
||||
->assertExitCode(0);
|
||||
|
||||
$output->refresh();
|
||||
|
||||
$this->assertNull($output->storage_disk);
|
||||
$this->assertNull($output->storage_path);
|
||||
}
|
||||
|
||||
/**
|
||||
* @return array{0: AiEditRequest, 1: AiEditOutput}
|
||||
*/
|
||||
private function createOutputWithoutStoragePath(): array
|
||||
{
|
||||
$event = Event::factory()->create(['status' => 'published']);
|
||||
$photo = Photo::factory()->for($event)->create([
|
||||
'tenant_id' => $event->tenant_id,
|
||||
'status' => 'approved',
|
||||
]);
|
||||
$style = AiStyle::query()->create([
|
||||
'key' => 'backfill-style',
|
||||
'name' => 'Backfill Style',
|
||||
'provider' => 'runware',
|
||||
'provider_model' => 'runware-default',
|
||||
'requires_source_image' => true,
|
||||
'is_active' => true,
|
||||
]);
|
||||
|
||||
$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' => 'Transform image style.',
|
||||
'idempotency_key' => 'backfill-request-1',
|
||||
'queued_at' => now()->subMinutes(2),
|
||||
'started_at' => now()->subMinute(),
|
||||
'completed_at' => now()->subSeconds(20),
|
||||
]);
|
||||
|
||||
$output = AiEditOutput::query()->create([
|
||||
'request_id' => $request->id,
|
||||
'provider_asset_id' => 'backfill-asset-1',
|
||||
'provider_url' => 'https://cdn.runware.ai/outputs/backfill-image.png',
|
||||
'is_primary' => true,
|
||||
'safety_state' => 'passed',
|
||||
'generated_at' => now(),
|
||||
]);
|
||||
|
||||
return [$request, $output];
|
||||
}
|
||||
|
||||
private function tinyPngBinary(): string
|
||||
{
|
||||
return (string) base64_decode('iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mP8/x8AAwMCAO+XnV0AAAAASUVORK5CYII=');
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user