Add unit test suite: 54 tests across models, core, and providers

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-12 00:36:59 +07:00
parent 88d674d679
commit c6370cf854
19 changed files with 1585 additions and 478 deletions
--- a/tests/unit/Core/PublisherTest.php
+++ b/tests/unit/Core/PublisherTest.php
@@ -0,0 +1,159 @@
+<?php
+/**
+ * Tests for PRAutoBlogger_Publisher.
+ *
+ * Validates WordPress post creation with metadata,
+ * generation log attachment, and error handling.
+ * All WordPress functions are mocked.
+ *
+ * @package PRAutoBlogger\Tests\Core
+ */
+
+namespace PRAutoBlogger\Tests\Core;
+
+use PRAutoBlogger\Tests\BaseTestCase;
+use Brain\Monkey\Functions;
+
+class PublisherTest extends BaseTestCase {
+
+    protected function setUp(): void {
+        parent::setUp();
+        require_once PRAB_PLUGIN_DIR . 'includes/core/class-prab-publisher.php';
+    }
+
+    /**
+     * Test publish creates a draft post with correct data.
+     */
+    public function test_publish_creates_draft_post(): void {
+        $post_data = [
+            'title'   => 'BPC-157: A Complete Guide',
+            'content' => '<p>BPC-157 is a synthetic peptide...</p>',
+            'excerpt' => 'An overview of BPC-157 benefits.',
+            'category' => 'peptide-research',
+        ];
+
+        $generation_meta = [
+            'model'             => 'google/gemini-2.0-flash-001',
+            'pipeline_mode'     => 'single_pass',
+            'total_cost'        => 0.0045,
+            'prompt_tokens'     => 800,
+            'completion_tokens' => 2000,
+            'generation_logs'   => [],
+        ];
+
+        // wp_insert_post should be called with post_status = 'draft'.
+        Functions\expect( 'wp_insert_post' )
+            ->once()
+            ->with( \Mockery::on( function ( $args ) {
+                return 'draft' === $args['post_status']
+                    && 'BPC-157: A Complete Guide' === $args['post_title']
+                    && false !== strpos( $args['post_content'], 'BPC-157' );
+            } ) )
+            ->andReturn( 42 );
+
+        // Metadata should be saved.
+        Functions\expect( 'update_post_meta' )
+            ->atLeast()
+            ->once();
+
+        Functions\when( 'is_wp_error' )->justReturn( false );
+        Functions\when( 'wp_slash' )->returnArg();
+        Functions\when( 'sanitize_title' )->alias( function ( $title ) {
+            return strtolower( str_replace( ' ', '-', $title ) );
+        } );
+
+        $publisher = new \PRAutoBlogger_Publisher();
+        $post_id   = $publisher->publish( $post_data, $generation_meta );
+
+        $this->assertSame( 42, $post_id );
+    }
+
+    /**
+     * Test publish returns WP_Error when wp_insert_post fails.
+     */
+    public function test_publish_returns_error_on_insert_failure(): void {
+        $wp_error = new \stdClass();
+        $wp_error->errors = [ 'db_insert_error' => [ 'Could not insert post' ] ];
+
+        Functions\when( 'wp_insert_post' )->justReturn( $wp_error );
+        Functions\when( 'is_wp_error' )->justReturn( true );
+        Functions\when( 'wp_slash' )->returnArg();
+        Functions\when( 'sanitize_title' )->returnArg();
+
+        $publisher = new \PRAutoBlogger_Publisher();
+        $result    = $publisher->publish(
+            [ 'title' => 'Test', 'content' => 'Test content', 'excerpt' => '', 'category' => '' ],
+            []
+        );
+
+        // Should propagate the error, not the post ID.
+        $this->assertNotSame( 42, $result );
+    }
+
+    /**
+     * Test publish sanitizes title and content.
+     */
+    public function test_publish_calls_sanitization(): void {
+        Functions\expect( 'wp_slash' )
+            ->atLeast()
+            ->once()
+            ->andReturnFirstArg();
+
+        Functions\expect( 'sanitize_title' )
+            ->once()
+            ->andReturnUsing( function ( $title ) {
+                return strtolower( str_replace( ' ', '-', $title ) );
+            } );
+
+        Functions\when( 'wp_insert_post' )->justReturn( 1 );
+        Functions\when( 'is_wp_error' )->justReturn( false );
+        Functions\when( 'update_post_meta' )->justReturn( true );
+
+        $publisher = new \PRAutoBlogger_Publisher();
+        $publisher->publish(
+            [
+                'title'   => 'Test <script>alert(1)</script>',
+                'content' => '<p>Safe content</p>',
+                'excerpt' => 'Excerpt',
+                'category' => 'general',
+            ],
+            []
+        );
+
+        // If we get here without error, sanitization was called.
+        $this->assertTrue( true );
+    }
+
+    /**
+     * Test publish stores generation metadata as post meta.
+     */
+    public function test_publish_stores_generation_metadata(): void {
+        $generation_meta = [
+            'model'         => 'test/model',
+            'pipeline_mode' => 'multi_step',
+            'total_cost'    => 0.01,
+        ];
+
+        Functions\when( 'wp_insert_post' )->justReturn( 99 );
+        Functions\when( 'is_wp_error' )->justReturn( false );
+        Functions\when( 'wp_slash' )->returnArg();
+        Functions\when( 'sanitize_title' )->returnArg();
+
+        $meta_saved = [];
+        Functions\when( 'update_post_meta' )->alias(
+            function ( $post_id, $key, $value ) use ( &$meta_saved ) {
+                $meta_saved[ $key ] = $value;
+                return true;
+            }
+        );
+
+        $publisher = new \PRAutoBlogger_Publisher();
+        $publisher->publish(
+            [ 'title' => 'T', 'content' => 'C', 'excerpt' => '', 'category' => '' ],
+            $generation_meta
+        );
+
+        // Verify generation metadata was stored.
+        $this->assertNotEmpty( $meta_saved );
+    }
+}