8bedecd2dd
The composer classmap autoloader loads the real PRAutoBlogger_Encryption before our eval() stub runs, so class_exists() is always true and the stub is never installed. Two consequences: 1. Provider tests set 'enc:fake-key' and the real decrypt() path ran, calling wp_salt() which isn't stubbed. Switched to plaintext keys so is_encrypted() short-circuits and no crypto runs. 2. Support test for the decrypt path now uses a real encrypt/decrypt round-trip with a stubbed wp_salt() returning a deterministic 32-char salt. This exercises the actual code path users hit in production, not a lying stub. Removes the dead eval() stubs that were being skipped anyway.
240 lines
9.2 KiB
PHP
240 lines
9.2 KiB
PHP
<?php
|
|
/**
|
|
* Tests for PRAutoBlogger_Runware_Image_Provider.
|
|
*
|
|
* Validates interface compliance, two-step POST-then-GET flow, retry
|
|
* behavior on 5xx, immediate-fail on 4xx, and dimension/prompt guards.
|
|
* All HTTP calls are mocked — no real API traffic.
|
|
*
|
|
* @package PRAutoBlogger\Tests\Providers
|
|
*/
|
|
|
|
namespace PRAutoBlogger\Tests\Providers;
|
|
|
|
use PRAutoBlogger\Tests\BaseTestCase;
|
|
use Brain\Monkey\Functions;
|
|
|
|
class RunwareImageProviderTest extends BaseTestCase {
|
|
|
|
protected function setUp(): void {
|
|
parent::setUp();
|
|
|
|
Functions\when( 'get_option' )->alias( function ( $key, $default = false ) {
|
|
static $options = [
|
|
'prautoblogger_runware_api_key' => 'rw-plain-test-key',
|
|
'prautoblogger_image_model' => 'runware:100@1',
|
|
];
|
|
return $options[ $key ] ?? $default;
|
|
} );
|
|
|
|
Functions\when( 'is_wp_error' )->alias( function ( $thing ) {
|
|
return $thing instanceof \WP_Error;
|
|
} );
|
|
|
|
Functions\when( 'wp_generate_uuid4' )->justReturn( '11111111-2222-3333-4444-555555555555' );
|
|
|
|
|
|
}
|
|
|
|
/** Provider name is 'runware'. */
|
|
public function test_get_provider_name(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->assertSame( 'runware', $provider->get_provider_name() );
|
|
}
|
|
|
|
/** estimate_cost returns a positive float for schnell. */
|
|
public function test_estimate_cost_schnell(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$cost = $provider->estimate_cost( 1216, 640, [ 'model' => 'runware:100@1' ] );
|
|
$this->assertIsFloat( $cost );
|
|
$this->assertEqualsWithDelta( 0.0006, $cost, 0.000001 );
|
|
}
|
|
|
|
/** estimate_cost returns higher value for dev. */
|
|
public function test_estimate_cost_dev_higher_than_schnell(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$schnell = $provider->estimate_cost( 1216, 640, [ 'model' => 'runware:100@1' ] );
|
|
$dev = $provider->estimate_cost( 1216, 640, [ 'model' => 'runware:101@1' ] );
|
|
$this->assertGreaterThan( $schnell, $dev );
|
|
}
|
|
|
|
/** Empty prompt throws InvalidArgumentException. */
|
|
public function test_generate_image_empty_prompt_throws(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \InvalidArgumentException::class );
|
|
$provider->generate_image( '', 1200, 640 );
|
|
}
|
|
|
|
/** Whitespace-only prompt throws InvalidArgumentException. */
|
|
public function test_generate_image_whitespace_prompt_throws(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \InvalidArgumentException::class );
|
|
$provider->generate_image( " \t\n ", 1200, 640 );
|
|
}
|
|
|
|
/** Dimension below minimum throws. */
|
|
public function test_generate_image_tiny_dimensions_throws(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \InvalidArgumentException::class );
|
|
$provider->generate_image( 'A test prompt', 10, 10 );
|
|
}
|
|
|
|
/** Dimension above maximum throws. */
|
|
public function test_generate_image_huge_dimensions_throws(): void {
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \InvalidArgumentException::class );
|
|
$provider->generate_image( 'A test prompt', 5000, 5000 );
|
|
}
|
|
|
|
/** Happy path: POST returns URL, GET returns bytes, result shape correct. */
|
|
public function test_generate_image_success(): void {
|
|
$image_url = 'https://im.runware.ai/image/abc.png';
|
|
$post_body = wp_json_encode( [
|
|
'data' => [
|
|
[ 'taskType' => 'imageInference', 'imageURL' => $image_url ],
|
|
],
|
|
] );
|
|
$fake_bytes = 'fake-png-bytes';
|
|
|
|
Functions\when( 'wp_remote_post' )->justReturn( [
|
|
'response' => [ 'code' => 200 ],
|
|
'headers' => [],
|
|
'body' => $post_body,
|
|
] );
|
|
Functions\when( 'wp_remote_get' )->justReturn( [
|
|
'response' => [ 'code' => 200 ],
|
|
'body' => $fake_bytes,
|
|
] );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->alias( function ( $r ) {
|
|
return isset( $r['response']['code'] ) ? (int) $r['response']['code'] : 0;
|
|
} );
|
|
Functions\when( 'wp_remote_retrieve_body' )->alias( function ( $r ) {
|
|
return isset( $r['body'] ) ? (string) $r['body'] : '';
|
|
} );
|
|
Functions\when( 'wp_remote_retrieve_header' )->justReturn( '' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$result = $provider->generate_image( 'A gym bro holding a vial', 1200, 632 );
|
|
|
|
$this->assertArrayHasKey( 'bytes', $result );
|
|
$this->assertArrayHasKey( 'mime_type', $result );
|
|
$this->assertArrayHasKey( 'width', $result );
|
|
$this->assertArrayHasKey( 'height', $result );
|
|
$this->assertArrayHasKey( 'model', $result );
|
|
$this->assertArrayHasKey( 'seed', $result );
|
|
$this->assertArrayHasKey( 'cost_usd', $result );
|
|
$this->assertArrayHasKey( 'latency_ms', $result );
|
|
|
|
$this->assertSame( $fake_bytes, $result['bytes'] );
|
|
$this->assertSame( 'image/png', $result['mime_type'] );
|
|
// Snapped to multiples of 64: 1200 → 1216, 632 → 640.
|
|
$this->assertSame( 1216, $result['width'] );
|
|
$this->assertSame( 640, $result['height'] );
|
|
$this->assertSame( 'runware:100@1', $result['model'] );
|
|
$this->assertGreaterThan( 0, $result['seed'] );
|
|
}
|
|
|
|
/** 4xx (non-429) fails immediately with RuntimeException. */
|
|
public function test_generate_image_4xx_throws_immediately(): void {
|
|
Functions\when( 'wp_remote_post' )->justReturn( [
|
|
'response' => [ 'code' => 401 ],
|
|
'headers' => [],
|
|
'body' => '{"errors":[{"message":"invalid key"}]}',
|
|
] );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->justReturn( 401 );
|
|
Functions\when( 'wp_remote_retrieve_body' )->justReturn( '{"errors":[{"message":"invalid key"}]}' );
|
|
Functions\when( 'wp_remote_retrieve_header' )->justReturn( '' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \RuntimeException::class );
|
|
$this->expectExceptionMessage( 'HTTP 401' );
|
|
$provider->generate_image( 'A test prompt', 1200, 632 );
|
|
}
|
|
|
|
/** 5xx triggers retry loop and eventually throws after PRAUTOBLOGGER_MAX_RETRIES. */
|
|
public function test_generate_image_5xx_retries_then_throws(): void {
|
|
$call_count = 0;
|
|
Functions\when( 'wp_remote_post' )->alias( function () use ( &$call_count ) {
|
|
$call_count++;
|
|
return [
|
|
'response' => [ 'code' => 503 ],
|
|
'headers' => [],
|
|
'body' => '{"errors":[{"message":"temporarily unavailable"}]}',
|
|
];
|
|
} );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->justReturn( 503 );
|
|
Functions\when( 'wp_remote_retrieve_body' )->justReturn( '{"errors":[{"message":"temporarily unavailable"}]}' );
|
|
Functions\when( 'wp_remote_retrieve_header' )->justReturn( '' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
try {
|
|
$provider->generate_image( 'A test prompt', 1200, 632 );
|
|
$this->fail( 'Expected RuntimeException after max retries' );
|
|
} catch ( \RuntimeException $e ) {
|
|
$this->assertSame(
|
|
PRAUTOBLOGGER_MAX_RETRIES,
|
|
$call_count,
|
|
'Provider must attempt exactly PRAUTOBLOGGER_MAX_RETRIES times before giving up.'
|
|
);
|
|
}
|
|
}
|
|
|
|
/** errors[] in a 200 body is treated as an upstream error. */
|
|
public function test_generate_image_errors_in_200_body_throws(): void {
|
|
$body = wp_json_encode( [
|
|
'errors' => [ [ 'message' => 'moderation rejected' ] ],
|
|
] );
|
|
Functions\when( 'wp_remote_post' )->justReturn( [
|
|
'response' => [ 'code' => 200 ],
|
|
'headers' => [],
|
|
'body' => $body,
|
|
] );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->justReturn( 200 );
|
|
Functions\when( 'wp_remote_retrieve_body' )->justReturn( $body );
|
|
Functions\when( 'wp_remote_retrieve_header' )->justReturn( '' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$this->expectException( \RuntimeException::class );
|
|
$provider->generate_image( 'A test prompt', 1200, 632 );
|
|
}
|
|
|
|
/** validate_credentials_detailed returns ok for a 200 response with no errors. */
|
|
public function test_validate_credentials_detailed_ok(): void {
|
|
Functions\when( 'wp_remote_post' )->justReturn( [
|
|
'response' => [ 'code' => 200 ],
|
|
'body' => '{"data":[{"taskType":"authentication"}]}',
|
|
] );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->justReturn( 200 );
|
|
Functions\when( 'wp_remote_retrieve_body' )->justReturn( '{"data":[{"taskType":"authentication"}]}' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$result = $provider->validate_credentials_detailed();
|
|
$this->assertSame( 'ok', $result['status'] );
|
|
}
|
|
|
|
/** validate_credentials_detailed returns error when key missing. */
|
|
public function test_validate_credentials_detailed_missing_key(): void {
|
|
Functions\when( 'get_option' )->alias( function ( $key, $default = false ) {
|
|
return $default;
|
|
} );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$result = $provider->validate_credentials_detailed();
|
|
$this->assertSame( 'error', $result['status'] );
|
|
}
|
|
|
|
/** validate_credentials_detailed returns error when response contains errors[]. */
|
|
public function test_validate_credentials_detailed_rejected_key(): void {
|
|
Functions\when( 'wp_remote_post' )->justReturn( [
|
|
'response' => [ 'code' => 200 ],
|
|
'body' => '{"errors":[{"message":"invalid key"}]}',
|
|
] );
|
|
Functions\when( 'wp_remote_retrieve_response_code' )->justReturn( 200 );
|
|
Functions\when( 'wp_remote_retrieve_body' )->justReturn( '{"errors":[{"message":"invalid key"}]}' );
|
|
|
|
$provider = new \PRAutoBlogger_Runware_Image_Provider();
|
|
$result = $provider->validate_credentials_detailed();
|
|
$this->assertSame( 'error', $result['status'] );
|
|
}
|
|
}
|