Initial commit: Runware-powered content generator foundation
Laravel app with a cost-tracked Runware.ai integration (image, video, text generation), a hard monthly EUR budget cap with live USD/EUR FX conversion, and a locally-maintained model/pricing table since Runware's model-search API doesn't expose pricing.
This commit is contained in:
@@ -0,0 +1,109 @@
|
||||
<?php
|
||||
|
||||
namespace Tests\Feature;
|
||||
|
||||
use App\Exceptions\BudgetExceededException;
|
||||
use App\Models\Generation;
|
||||
use App\Models\RunwareModel;
|
||||
use App\Services\Runware\GenerationService;
|
||||
use Illuminate\Foundation\Testing\RefreshDatabase;
|
||||
use Illuminate\Support\Facades\Http;
|
||||
use Illuminate\Support\Facades\Storage;
|
||||
use Tests\TestCase;
|
||||
|
||||
class GenerationServiceTest extends TestCase
|
||||
{
|
||||
use RefreshDatabase;
|
||||
|
||||
protected function service(): GenerationService
|
||||
{
|
||||
return $this->app->make(GenerationService::class);
|
||||
}
|
||||
|
||||
public function test_image_generation_records_actual_cost_and_downloads_file(): void
|
||||
{
|
||||
Storage::fake('public');
|
||||
Http::fake([
|
||||
'https://api.runware.ai/v1' => Http::response(['data' => [[
|
||||
'taskType' => 'imageInference',
|
||||
'taskUUID' => '11111111-1111-1111-1111-111111111111',
|
||||
'imageURL' => 'https://cdn.example.com/out.jpg',
|
||||
'cost' => 0.01,
|
||||
]]]),
|
||||
'https://cdn.example.com/out.jpg' => Http::response('FAKEBYTES'),
|
||||
]);
|
||||
|
||||
RunwareModel::create([
|
||||
'modality' => 'image', 'air_id' => 'a:cheap@1', 'label' => 'Cheap',
|
||||
'unit' => 'per_image', 'price_usd' => 0.01, 'active' => true, 'is_default' => true,
|
||||
]);
|
||||
|
||||
$generation = $this->service()->createImage('a cat wearing a hat');
|
||||
|
||||
$this->assertSame('completed', $generation->status);
|
||||
$this->assertEquals(0.01, $generation->cost_usd);
|
||||
$this->assertNotNull($generation->cost_eur);
|
||||
$this->assertFalse($generation->cost_is_estimated);
|
||||
Storage::disk('public')->assertExists($generation->output_path);
|
||||
}
|
||||
|
||||
public function test_failed_api_call_marks_generation_failed_without_cost(): void
|
||||
{
|
||||
Http::fake([
|
||||
'https://api.runware.ai/v1' => Http::response(['errors' => [['message' => 'boom']]], 400),
|
||||
]);
|
||||
|
||||
RunwareModel::create([
|
||||
'modality' => 'image', 'air_id' => 'a:cheap@1', 'label' => 'Cheap',
|
||||
'unit' => 'per_image', 'price_usd' => 0.01, 'active' => true, 'is_default' => true,
|
||||
]);
|
||||
|
||||
$this->expectException(\RuntimeException::class);
|
||||
|
||||
try {
|
||||
$this->service()->createImage('a cat wearing a hat');
|
||||
} finally {
|
||||
$generation = Generation::first();
|
||||
$this->assertSame('failed', $generation->status);
|
||||
$this->assertNull($generation->cost_usd);
|
||||
}
|
||||
}
|
||||
|
||||
public function test_budget_cap_blocks_generation_that_would_exceed_it(): void
|
||||
{
|
||||
config(['runware.monthly_cap_eur' => 0.001]);
|
||||
|
||||
RunwareModel::create([
|
||||
'modality' => 'image', 'air_id' => 'a:pricey@1', 'label' => 'Pricey',
|
||||
'unit' => 'per_image', 'price_usd' => 10, 'active' => true, 'is_default' => true,
|
||||
]);
|
||||
|
||||
$this->expectException(BudgetExceededException::class);
|
||||
|
||||
$this->service()->createImage('this should never call the API');
|
||||
|
||||
Http::assertNothingSent();
|
||||
}
|
||||
|
||||
public function test_text_generation_falls_back_to_estimated_cost_when_usage_cost_missing(): void
|
||||
{
|
||||
Http::fake([
|
||||
'https://api.runware.ai/v1/chat/completions' => Http::response([
|
||||
'choices' => [['message' => ['content' => 'Hello there']]],
|
||||
'usage' => ['total_tokens' => 100],
|
||||
]),
|
||||
]);
|
||||
|
||||
RunwareModel::create([
|
||||
'modality' => 'text', 'air_id' => 't:cheap@1', 'label' => 'Cheap',
|
||||
'unit' => 'per_million_tokens', 'price_usd' => 1, 'active' => true, 'is_default' => true,
|
||||
]);
|
||||
|
||||
$generation = $this->service()->createText('say hello');
|
||||
|
||||
$this->assertSame('completed', $generation->status);
|
||||
$this->assertSame('Hello there', $generation->output_text);
|
||||
$this->assertTrue($generation->cost_is_estimated);
|
||||
$this->assertEqualsWithDelta(0.0001, $generation->cost_usd, 0.00001);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user