peptide-repo/prautoblogger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: guard WP_Error cost access + add missing test constants

Browse Source 
- Wrap cost_usd accumulation in is_wp_error() guard so WP_Error
  returns from generate_image_a/b don't crash on array access
- Add PRAUTOBLOGGER_DEFAULT_IMAGE_STYLE_SUFFIX to test bootstrap
- Add is_wp_error, sanitize_title, esc_html__ mocks to ImagePipelineTest

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
peptiderepo
2026-04-16 17:47:03 +07:00
parent 5e7fd882bc
commit e60cb83770
3 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
includes/core/class-image-pipeline.php
View File

@@ -124,7 +124,9 @@ class PRAutoBlogger_Image_Pipeline {
? $image_a_result->get_error_message() ? $image_a_result->get_error_message()
: 'Image A generation produced no attachment ID.'; : 'Image A generation produced no attachment ID.';
} }
if ( ! is_wp_error( $image_a_result ) ) {
$result['cost_usd'] += $image_a_result['cost_usd'] ?? 0.0; $result['cost_usd'] += $image_a_result['cost_usd'] ?? 0.0;
}
// Generate Image B (source-driven prompt) if source data is available. // Generate Image B (source-driven prompt) if source data is available.
if ( null !== $source_data && ! empty( $source_data ) ) { if ( null !== $source_data && ! empty( $source_data ) ) {
@@ -136,8 +138,10 @@ class PRAutoBlogger_Image_Pipeline {
? $image_b_result->get_error_message() ? $image_b_result->get_error_message()
: 'Image B generation produced no attachment ID.'; : 'Image B generation produced no attachment ID.';
} }
if ( ! is_wp_error( $image_b_result ) ) {
$result['cost_usd'] += $image_b_result['cost_usd'] ?? 0.0; $result['cost_usd'] += $image_b_result['cost_usd'] ?? 0.0;
} }
}
return $result; return $result;
} }

5
tests/bootstrap.php
View File

@@ -55,6 +55,11 @@ if ( ! defined( 'PRAUTOBLOGGER_DEFAULT_EDITOR_MODEL' ) ) {
define( 'PRAUTOBLOGGER_DEFAULT_EDITOR_MODEL', 'google/gemini-2.5-flash-lite' ); define( 'PRAUTOBLOGGER_DEFAULT_EDITOR_MODEL', 'google/gemini-2.5-flash-lite' );
} }
// Default image style suffix used by ImagePromptBuilder and settings UI.
if ( ! defined( 'PRAUTOBLOGGER_DEFAULT_IMAGE_STYLE_SUFFIX' ) ) {
define( 'PRAUTOBLOGGER_DEFAULT_IMAGE_STYLE_SUFFIX', 'Style: test infomercial style.' );
}
// WordPress database constants used in $wpdb queries. // WordPress database constants used in $wpdb queries.
if ( ! defined( 'ARRAY_A' ) ) { if ( ! defined( 'ARRAY_A' ) ) {
define( 'ARRAY_A', 'ARRAY_A' ); define( 'ARRAY_A', 'ARRAY_A' );

5
tests/unit/Core/ImagePipelineTest.php
View File

@@ -19,6 +19,9 @@ class ImagePipelineTest extends BaseTestCase {
parent::setUp(); parent::setUp();
// Mock WordPress functions used by the pipeline and its dependencies. // Mock WordPress functions used by the pipeline and its dependencies.
Functions\when( 'is_wp_error' )->alias( function ( $thing ) {
return $thing instanceof \WP_Error;
} );
Functions\when( 'wp_generate_uuid4' )->justReturn( 'test-uuid' ); Functions\when( 'wp_generate_uuid4' )->justReturn( 'test-uuid' );
Functions\when( 'get_option' )->alias( function ( $key, $default = false ) { Functions\when( 'get_option' )->alias( function ( $key, $default = false ) {
static $options = [ static $options = [
@@ -32,6 +35,8 @@ class ImagePipelineTest extends BaseTestCase {
return trim( strip_tags( $str ) ); return trim( strip_tags( $str ) );
} ); } );
Functions\when( 'update_post_meta' )->justReturn( true ); Functions\when( 'update_post_meta' )->justReturn( true );
Functions\when( 'sanitize_title' )->returnArg();
Functions\when( 'esc_html__' )->returnArg();
} }
/** /**