feat: Opik LLM observability tracing skeleton (Wave 1) — v0.12.0
Wires PRAutoblogger into Comet Opik cloud for per-call tracing. - Thin PHP REST client (class-opik-client.php, 276 LOC) - Async fire-and-forget dispatch via wp_schedule_single_event - Feature flag PRAUTOBLOGGER_OPIK_ENABLED defaults OFF - One trace per article, 5 spans across content-generator + chief-editor - Exponential backoff 2s/4s/8s, max 3 retries, drops on failure - Admin settings page (api-key configured indicator only, never rendered) - ARCHITECTURE.md + CONVENTIONS.md updated PR #113 — QA gate: PASS (qa-reviews/prautoblogger/2026-04-23-0d8dc67.md)
This commit is contained in:
peptiderepo
241
tests/unit/Services/test-opik-dispatcher.php
Normal file
241
tests/unit/Services/test-opik-dispatcher.php
Normal file
@@ -0,0 +1,241 @@
|
||||
<?php
|
||||
declare(strict_types=1);
|
||||
|
||||
/**
|
||||
* Unit tests for PRAutoBlogger_Opik_Dispatcher.
|
||||
*
|
||||
* @see includes/services/opik/class-opik-dispatcher.php
|
||||
*/
|
||||
namespace PRAutoBlogger\Tests\Services;
|
||||
|
||||
use PRAutoBlogger\Tests\BaseTestCase;
|
||||
use Brain\Monkey\Functions;
|
||||
|
||||
/**
|
||||
* @group opik
|
||||
* @group services
|
||||
*/
|
||||
class TestOpikDispatcher extends BaseTestCase {
|
||||
|
||||
private array $option_store = array();
|
||||
|
||||
protected function setUp(): void {
|
||||
parent::setUp();
|
||||
|
||||
// Mock WP options.
|
||||
Functions\when( 'get_option' )->alias(
|
||||
function( $option, $default = false ) {
|
||||
return $this->option_store[ $option ] ?? $default;
|
||||
}
|
||||
);
|
||||
|
||||
Functions\when( 'update_option' )->alias(
|
||||
function( $option, $value ) {
|
||||
$this->option_store[ $option ] = $value;
|
||||
return true;
|
||||
}
|
||||
);
|
||||
|
||||
Functions\when( 'delete_option' )->alias(
|
||||
function( $option ) {
|
||||
unset( $this->option_store[ $option ] );
|
||||
return true;
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test dispatcher skips if Opik is not enabled.
|
||||
*/
|
||||
public function test_dispatch_skips_if_disabled(): void {
|
||||
$client = $this->create_mock_client();
|
||||
$queue = $this->create_mock_queue();
|
||||
|
||||
// Queue has items but feature is disabled.
|
||||
$this->option_store['prautoblogger_opik_enabled'] = false;
|
||||
|
||||
$dispatcher = new \PRAutoBlogger_Opik_Dispatcher( $client, $queue );
|
||||
$dispatcher->dispatch();
|
||||
|
||||
// Client should not have been called.
|
||||
$this->assertTrue( true );
|
||||
}
|
||||
|
||||
/**
|
||||
* Test dispatch() batches and posts spans.
|
||||
*/
|
||||
public function test_dispatch_posts_spans(): void {
|
||||
// Enable Opik and set credentials.
|
||||
$this->option_store['prautoblogger_opik_enabled'] = true;
|
||||
|
||||
// Mock constants.
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_API_KEY' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_API_KEY', 'test-key' );
|
||||
}
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_WORKSPACE' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_WORKSPACE', 'test-workspace' );
|
||||
}
|
||||
|
||||
$posted_spans = null;
|
||||
|
||||
$client = $this->getMockBuilder( \PRAutoBlogger_Opik_Client::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'batch_create_spans', 'batch_create_traces' ) )
|
||||
->getMock();
|
||||
|
||||
$client->expects( $this->once() )
|
||||
->method( 'batch_create_spans' )
|
||||
->willReturnCallback( function( $spans ) use ( &$posted_spans ) {
|
||||
$posted_spans = $spans;
|
||||
return array( 'inserted' => count( $spans ) );
|
||||
} );
|
||||
|
||||
$queue = $this->getMockBuilder( \PRAutoBlogger_Opik_Span_Queue::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'dequeue', 'get_queue_depth' ) )
|
||||
->getMock();
|
||||
|
||||
$queue->expects( $this->once() )
|
||||
->method( 'dequeue' )
|
||||
->with( 100 )
|
||||
->willReturn(
|
||||
array(
|
||||
array(
|
||||
'type' => 'span',
|
||||
'payload' => array( 'id' => 'span-1', 'name' => 'test' ),
|
||||
'attempt' => 0,
|
||||
'enqueued' => time(),
|
||||
),
|
||||
)
|
||||
);
|
||||
|
||||
$queue->method( 'get_queue_depth' )->willReturn( 0 );
|
||||
|
||||
$dispatcher = new \PRAutoBlogger_Opik_Dispatcher( $client, $queue );
|
||||
$dispatcher->dispatch();
|
||||
|
||||
$this->assertNotNull( $posted_spans );
|
||||
$this->assertCount( 1, $posted_spans );
|
||||
}
|
||||
|
||||
/**
|
||||
* Test dispatcher handles empty queue gracefully.
|
||||
*/
|
||||
public function test_dispatch_handles_empty_queue(): void {
|
||||
$this->option_store['prautoblogger_opik_enabled'] = true;
|
||||
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_API_KEY' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_API_KEY', 'test-key' );
|
||||
}
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_WORKSPACE' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_WORKSPACE', 'test-workspace' );
|
||||
}
|
||||
|
||||
$client = $this->create_mock_client();
|
||||
$queue = $this->create_mock_queue( array() );
|
||||
|
||||
$dispatcher = new \PRAutoBlogger_Opik_Dispatcher( $client, $queue );
|
||||
$dispatcher->dispatch();
|
||||
|
||||
// Should not error.
|
||||
$this->assertTrue( true );
|
||||
}
|
||||
|
||||
/**
|
||||
* Test dispatcher separates spans and traces.
|
||||
*/
|
||||
public function test_dispatch_separates_types(): void {
|
||||
$this->option_store['prautoblogger_opik_enabled'] = true;
|
||||
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_API_KEY' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_API_KEY', 'test-key' );
|
||||
}
|
||||
if ( ! defined( 'PRAUTOBLOGGER_OPIK_WORKSPACE' ) ) {
|
||||
define( 'PRAUTOBLOGGER_OPIK_WORKSPACE', 'test-workspace' );
|
||||
}
|
||||
|
||||
$posted_traces = null;
|
||||
$posted_spans = null;
|
||||
|
||||
$client = $this->getMockBuilder( \PRAutoBlogger_Opik_Client::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'batch_create_traces', 'batch_create_spans' ) )
|
||||
->getMock();
|
||||
|
||||
$client->method( 'batch_create_traces' )->willReturnCallback(
|
||||
function( $traces ) use ( &$posted_traces ) {
|
||||
$posted_traces = $traces;
|
||||
return array( 'inserted' => count( $traces ) );
|
||||
}
|
||||
);
|
||||
|
||||
$client->method( 'batch_create_spans' )->willReturnCallback(
|
||||
function( $spans ) use ( &$posted_spans ) {
|
||||
$posted_spans = $spans;
|
||||
return array( 'inserted' => count( $spans ) );
|
||||
}
|
||||
);
|
||||
|
||||
$queue = $this->getMockBuilder( \PRAutoBlogger_Opik_Span_Queue::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'dequeue', 'get_queue_depth' ) )
|
||||
->getMock();
|
||||
|
||||
$queue->method( 'dequeue' )->willReturn(
|
||||
array(
|
||||
array(
|
||||
'type' => 'trace',
|
||||
'payload' => array( 'id' => 'trace-1' ),
|
||||
'attempt' => 0,
|
||||
'enqueued' => time(),
|
||||
),
|
||||
array(
|
||||
'type' => 'span',
|
||||
'payload' => array( 'id' => 'span-1' ),
|
||||
'attempt' => 0,
|
||||
'enqueued' => time(),
|
||||
),
|
||||
)
|
||||
);
|
||||
|
||||
$queue->method( 'get_queue_depth' )->willReturn( 0 );
|
||||
|
||||
$dispatcher = new \PRAutoBlogger_Opik_Dispatcher( $client, $queue );
|
||||
$dispatcher->dispatch();
|
||||
|
||||
$this->assertNotNull( $posted_traces );
|
||||
$this->assertCount( 1, $posted_traces );
|
||||
$this->assertNotNull( $posted_spans );
|
||||
$this->assertCount( 1, $posted_spans );
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper to create a mock client.
|
||||
*/
|
||||
private function create_mock_client() {
|
||||
$client = $this->getMockBuilder( \PRAutoBlogger_Opik_Client::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'batch_create_spans', 'batch_create_traces' ) )
|
||||
->getMock();
|
||||
|
||||
$client->method( 'batch_create_spans' )->willReturn( array() );
|
||||
$client->method( 'batch_create_traces' )->willReturn( array() );
|
||||
|
||||
return $client;
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper to create a mock queue.
|
||||
*/
|
||||
private function create_mock_queue( array $items = array() ) {
|
||||
$queue = $this->getMockBuilder( \PRAutoBlogger_Opik_Span_Queue::class )
|
||||
->disableOriginalConstructor()
|
||||
->onlyMethods( array( 'dequeue', 'get_queue_depth' ) )
|
||||
->getMock();
|
||||
|
||||
$queue->method( 'dequeue' )->willReturn( $items );
|
||||
$queue->method( 'get_queue_depth' )->willReturn( count( $items ) );
|
||||
|
||||
return $queue;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user