prautoblogger/tests/unit/Providers/RunwareModelCatalogTest.php

<?php
/**
 * Tests for PRAutoBlogger_Runware_Model_Catalog.
 *
 * Validates live sync, fallback logic, cache staleness, and pricing merge.
 *
 * @package PRAutoBlogger\Tests\Providers
 */

namespace PRAutoBlogger\Tests\Providers;

use PRAutoBlogger\Tests\BaseTestCase;
use Brain\Monkey\Functions;
use Brain\Monkey\Actions;

class RunwareModelCatalogTest extends BaseTestCase {

	protected function setUp(): void {
		parent::setUp();
		// Mock WP options for cache simulation.
		Functions\when( 'get_option' )->alias( function ( $key, $default = false ) {
			static $options = array();
			return $options[ $key ] ?? $default;
		} );
		Functions\when( 'update_option' )->alias( function ( $key, $value ) {
			static $options = array();
			$options[ $key ] = $value;
			return true;
		} );
		Functions\when( 'time' )->justReturn( 1704067200 ); // 2024-01-01 00:00:00
	}

	/** Sync with empty or missing API key returns false. */
	public function test_sync_no_api_key(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$result = $catalog->sync();
		$this->assertFalse( $result );
	}

	/** get_models returns fallback if no cache and sync fails. */
	public function test_get_models_fallback_on_sync_failure(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$models = $catalog->get_models();

		// Should have the fallback list (15 Runware models).
		$this->assertIsArray( $models );
		$this->assertGreaterThan( 0, count( $models ) );

		// Check at least one expected fallback model exists.
		$schnell_exists = false;
		foreach ( $models as $model ) {
			if ( 'runware:100@1' === ( $model['id'] ?? '' ) ) {
				$schnell_exists = true;
				$this->assertSame( 'FLUX.1 schnell (Runware)', $model['name'] );
				$this->assertSame( 'runware', $model['provider'] );
				break;
			}
		}
		$this->assertTrue( $schnell_exists, 'Fallback should include FLUX.1 schnell' );
	}

	/** is_stale returns true if never synced. */
	public function test_is_stale_never_synced(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$this->assertTrue( $catalog->is_stale() );
	}

	/** get_last_synced_at returns null if never synced. */
	public function test_get_last_synced_at_never(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$this->assertNull( $catalog->get_last_synced_at() );
	}

	/** Fallback list is never empty. */
	public function test_get_models_never_empty(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();

		// Even with no cache and failed sync, get_models should return the fallback.
		$models = $catalog->get_models();
		$this->assertIsArray( $models );
		$this->assertNotEmpty( $models );

		// All models should have required fields.
		foreach ( $models as $model ) {
			$this->assertArrayHasKey( 'id', $model );
			$this->assertArrayHasKey( 'name', $model );
			$this->assertArrayHasKey( 'provider', $model );
			$this->assertArrayHasKey( 'capabilities', $model );
		}
	}

	/** Fallback models have correct pricing. */
	public function test_fallback_models_have_pricing(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$models = $catalog->get_models();

		$schnell = null;
		$dev = null;
		foreach ( $models as $model ) {
			if ( 'runware:100@1' === $model['id'] ) {
				$schnell = $model;
			}
			if ( 'runware:101@1' === $model['id'] ) {
				$dev = $model;
			}
		}

		$this->assertNotNull( $schnell );
		$this->assertEqualsWithDelta( 0.0006, $schnell['cost_per_image'], 0.000001 );

		$this->assertNotNull( $dev );
		$this->assertEqualsWithDelta( 0.02, $dev['cost_per_image'], 0.000001 );
	}

	/** Fallback models all have 'image_generation' capability. */
	public function test_fallback_models_have_capabilities(): void {
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$models = $catalog->get_models();

		foreach ( $models as $model ) {
			$this->assertIsArray( $model['capabilities'] );
			$this->assertContains( 'image_generation', $model['capabilities'] );
		}
	}

	/** Happy path: successful sync caches models and get_models returns them. */
	public function test_sync_success_caches_and_returns_models(): void {
		// Mock a valid Runware API response.
		Functions\when( 'wp_remote_post' )->justReturn(
			array(
				'response' => array( 'code' => 200 ),
				'body'     => wp_json_encode(
					array(
						'data' => array(
							array(
								'id'       => 'test-model-1',
								'name'     => 'Test Model 1',
								'cost'     => 0.001,
								'provider' => 'runware',
							),
							array(
								'id'       => 'test-model-2',
								'name'     => 'Test Model 2',
								'cost'     => 0.002,
								'provider' => 'runware',
							),
						),
					)
				),
			)
		);

		// Mock the API key to exist.
		Functions\when( 'get_option' )->alias( function ( $key, $default = false ) {
			static $options = array();
			if ( 'prautoblogger_runware_api_key' === $key ) {
				return 'test-api-key';
			}
			return $options[ $key ] ?? $default;
		} );

		Functions\when( 'update_option' )->alias( function ( $key, $value ) {
			static $options = array();
			$options[ $key ] = $value;
			return true;
		} );

		// Perform the sync.
		$catalog = new \PRAutoBlogger_Runware_Model_Catalog();
		$result = $catalog->sync();
		$this->assertTrue( $result, 'Sync should return true on success' );

		// Verify get_models returns the synced models (not fallback).
		$models = $catalog->get_models();
		$this->assertIsArray( $models );
		$this->assertCount( 2, $models );
		$this->assertSame( 'test-model-1', $models[0]['id'] );
		$this->assertSame( 'test-model-2', $models[1]['id'] );
	}
}