ca1a41f6f9
Add initial plugin implementation per the CTO engineering handoff (geomonitor/threads/2026-06-ai-visibility-v1/01-cto-eng-handoff.md). ## What - PGM_Probe_Provider interface + PGM_Probe_Result value object - PGM_Perplexity_Provider (sonar via OpenRouter/CF AI Gateway) - PGM_OpenRouter_Provider (GPT-search + Gemini-search, generic parameterised) - PGM_Gateway_Client: shared HTTP + exponential-backoff retry (mirrors PRAB) - PGM_Citation_Detector: domain extraction, is_cited(), get_our_position() - PGM_Cost_Ledger: MTD cost from DB, can_afford() pre-check, hard monthly cap - PGM_Probe_Runner: peptides×intents×models run, budget abort, in-run cache - PGM_Table_Manager: pgm_ai_visibility dbDelta, schema-version, drop_table() - PGM_Config: typed getters, seed_defaults() (12 peptides, 3 intents, 3 models) - PGM_Cron: weekly pgm_weekly_probe schedule/clear - PGM_Data_Collector + PGM_Dashboard_Panel interfaces (seam for future panels) - PGM_Collector_Registry: singleton registry - PGM_Ai_Visibility_Collector: trendline, standings, MTD cost (score formula) - PGM_Ai_Visibility_Panel: proxy note, Chart.js trendline, standings table - PGM_Admin_Page: GEO Monitor menu, Run now POST handler (nonce+capability) - uninstall.php: DROP TABLE + DELETE pgm_* options + wp_clear_scheduled_hook - Unit tests (6 files): citation detection, score calc, cost ledger + budget abort, provider response parsing (mock HTTP), cron, uninstall purge - ARCHITECTURE.md, CONVENTIONS.md, CONTEXT.md (score formula), README.md, CHANGELOG.md — same-commit per DoD - CI workflow: PHP lint (8.1/8.2/8.3) + PHPCS + 300-line check ## Security - Secrets from wp-config constants only; none hardcoded - manage_options + nonce on every admin action - Input sanitized; output escaped Agent-Session: geomonitor v0.1.0
86 lines
3.6 KiB
PHP
86 lines
3.6 KiB
PHP
<?php
|
|
/**
|
|
* Tests for PGM_Cost_Ledger: MTD cost retrieval, budget pre-check, and
|
|
* the budget-cap abort behaviour.
|
|
*
|
|
* @package PeptideGeoMonitor
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
|
|
require_once __DIR__ . '/../bootstrap.php';
|
|
|
|
echo "=== PGM Cost Ledger Tests ===\n";
|
|
|
|
// ── Test: can_afford when below cap ─────────────────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '2.50'; // MTD spent
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert( $ledger->can_afford( 1.0 ), 'can_afford: 2.50 spent + 1.00 = 3.50 < 5.00 → true' );
|
|
|
|
// ── Test: can_afford when at cap ─────────────────────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '5.00';
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert( ! $ledger->can_afford( 0.01 ), 'can_afford: 5.00 spent + 0.01 > 5.00 → false' );
|
|
|
|
// ── Test: can_afford when exactly at cap ─────────────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '4.99';
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert( $ledger->can_afford( 0.01 ), 'can_afford: 4.99 + 0.01 = 5.00 exactly → true (at cap is allowed)' );
|
|
|
|
// ── Test: get_remaining_budget_usd ──────────────────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '3.25';
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert_equals( 1.75, round( $ledger->get_remaining_budget_usd(), 2 ), 'get_remaining_budget_usd: 5.00 - 3.25 = 1.75' );
|
|
|
|
// ── Test: get_remaining_budget_usd never negative ────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '10.00'; // over cap
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert_equals( 0.0, $ledger->get_remaining_budget_usd(), 'get_remaining_budget_usd: clamps to 0.0 when over cap' );
|
|
|
|
// ── Test: get_month_to_date_usd with null DB result (no rows yet) ────────
|
|
|
|
pgm_test_reset();
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 5.0;
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = null;
|
|
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert_equals( 0.0, $ledger->get_month_to_date_usd(), 'get_month_to_date_usd: null DB result → 0.0' );
|
|
|
|
// ── Test: update_row_cost returns true ──────────────────────────────────
|
|
|
|
pgm_test_reset();
|
|
$ledger = new PGM_Cost_Ledger();
|
|
pgm_assert( $ledger->update_row_cost( 42, 0.00123456 ), 'update_row_cost: returns true on success' );
|
|
|
|
// ── Test: budget abort mid-run (runner stops when cap hit) ──────────────
|
|
|
|
pgm_test_reset();
|
|
// Simulate a nearly-exhausted budget.
|
|
$GLOBALS['pgm_test_state']['options']['pgm_monthly_budget_usd'] = 0.001; // $0.001 cap
|
|
$GLOBALS['pgm_test_state']['wpdb_var'] = '0.001'; // fully spent
|
|
|
|
$ledger2 = new PGM_Cost_Ledger();
|
|
pgm_assert( ! $ledger2->can_afford( 0.000001 ), 'budget abort: can_afford returns false at cap → runner would stop' );
|
|
|
|
exit( pgm_test_summary() );
|