Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 792327afd6 |
@@ -1,7 +1,12 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"testing"
|
||||
|
||||
sqlmock "github.com/DATA-DOG/go-sqlmock"
|
||||
"github.com/Molecule-AI/molecule-monorepo/platform/internal/db"
|
||||
)
|
||||
|
||||
// TestExtractExpiresInSeconds covers the JSON parser used at enqueue time
|
||||
@@ -58,3 +63,207 @@ func TestExtractExpiresInSeconds(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// ── QueueStatusByID ─────────────────────────────────────────────────────────────
|
||||
|
||||
func setupQueueStatusDB(t *testing.T) sqlmock.Sqlmock {
|
||||
t.Helper()
|
||||
mockDB, mock, err := sqlmock.New()
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create sqlmock: %v", err)
|
||||
}
|
||||
prevDB := db.DB
|
||||
db.DB = mockDB
|
||||
t.Cleanup(func() { db.DB = prevDB; mockDB.Close() })
|
||||
return mock
|
||||
}
|
||||
|
||||
func TestQueueStatusByID_Success(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
wsID := "cccccccc-cccc-cccc-cccc-cccccccccccc"
|
||||
|
||||
rows := sqlmock.NewRows([]string{
|
||||
"id", "workspace_id", "status", "priority", "attempts",
|
||||
"last_error", "enqueued_at", "dispatched_at", "completed_at", "expires_at",
|
||||
"response_body",
|
||||
}).AddRow(
|
||||
queueID, wsID, "queued", 50, 0,
|
||||
nil, // last_error
|
||||
"2026-01-01T00:00:00Z", // enqueued_at
|
||||
nil, // dispatched_at
|
||||
nil, // completed_at
|
||||
nil, // expires_at
|
||||
nil, // response_body
|
||||
)
|
||||
mock.ExpectQuery(`SELECT`).
|
||||
WithArgs(queueID).
|
||||
WillReturnRows(rows)
|
||||
|
||||
qs, err := QueueStatusByID(context.Background(), queueID)
|
||||
if err != nil {
|
||||
t.Fatalf("QueueStatusByID returned error: %v", err)
|
||||
}
|
||||
if qs.ID != queueID {
|
||||
t.Errorf("ID = %q, want %q", qs.ID, queueID)
|
||||
}
|
||||
if qs.WorkspaceID != wsID {
|
||||
t.Errorf("WorkspaceID = %q, want %q", qs.WorkspaceID, wsID)
|
||||
}
|
||||
if qs.Status != "queued" {
|
||||
t.Errorf("Status = %q, want %q", qs.Status, "queued")
|
||||
}
|
||||
if qs.Priority != 50 {
|
||||
t.Errorf("Priority = %d, want 50", qs.Priority)
|
||||
}
|
||||
if qs.LastError != nil {
|
||||
t.Errorf("LastError = %v, want nil", qs.LastError)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueStatusByID_NotFound(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
|
||||
mock.ExpectQuery(`SELECT`).
|
||||
WithArgs(queueID).
|
||||
WillReturnError(sql.ErrNoRows)
|
||||
|
||||
qs, err := QueueStatusByID(context.Background(), queueID)
|
||||
if err != sql.ErrNoRows {
|
||||
t.Errorf("expected sql.ErrNoRows, got %v", err)
|
||||
}
|
||||
if qs != nil {
|
||||
t.Errorf("expected nil queue status, got %+v", qs)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueStatusByID_DBError(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
|
||||
mock.ExpectQuery(`SELECT`).
|
||||
WithArgs(queueID).
|
||||
WillReturnError(sql.ErrConnDone)
|
||||
|
||||
qs, err := QueueStatusByID(context.Background(), queueID)
|
||||
if err == nil {
|
||||
t.Error("expected error, got nil")
|
||||
}
|
||||
if qs != nil {
|
||||
t.Errorf("expected nil queue status, got %+v", qs)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueStatusByID_CompletedWithResponse(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
wsID := "cccccccc-cccc-cccc-cccc-cccccccccccc"
|
||||
|
||||
respBody := []byte(`{"text":"delegation result"}`)
|
||||
rows := sqlmock.NewRows([]string{
|
||||
"id", "workspace_id", "status", "priority", "attempts",
|
||||
"last_error", "enqueued_at", "dispatched_at", "completed_at", "expires_at",
|
||||
"response_body",
|
||||
}).AddRow(
|
||||
queueID, wsID, "completed", 50, 1,
|
||||
nil,
|
||||
"2026-01-01T00:00:00Z",
|
||||
"2026-01-01T00:01:00Z",
|
||||
"2026-01-01T00:02:00Z",
|
||||
nil,
|
||||
respBody,
|
||||
)
|
||||
mock.ExpectQuery(`SELECT`).
|
||||
WithArgs(queueID).
|
||||
WillReturnRows(rows)
|
||||
|
||||
qs, err := QueueStatusByID(context.Background(), queueID)
|
||||
if err != nil {
|
||||
t.Fatalf("QueueStatusByID returned error: %v", err)
|
||||
}
|
||||
if qs.Status != "completed" {
|
||||
t.Errorf("Status = %q, want completed", qs.Status)
|
||||
}
|
||||
if qs.ResponseBody == nil {
|
||||
t.Fatal("ResponseBody should be set for completed status")
|
||||
}
|
||||
if string(qs.ResponseBody) != `{"text":"delegation result"}` {
|
||||
t.Errorf("ResponseBody = %q, want %q", string(qs.ResponseBody), `{"text":"delegation result"}`)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// ── queueRowAuthFields ──────────────────────────────────────────────────────────
|
||||
|
||||
func TestQueueRowAuthFields_Success(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
callerID := "bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb"
|
||||
wsID := "cccccccc-cccc-cccc-cccc-cccccccccccc"
|
||||
|
||||
rows := sqlmock.NewRows([]string{"caller_id", "workspace_id"}).
|
||||
AddRow(callerID, wsID)
|
||||
mock.ExpectQuery(`SELECT caller_id, workspace_id FROM a2a_queue WHERE id`).
|
||||
WithArgs(queueID).
|
||||
WillReturnRows(rows)
|
||||
|
||||
gotCaller, gotWs, err := queueRowAuthFields(context.Background(), queueID)
|
||||
if err != nil {
|
||||
t.Fatalf("queueRowAuthFields returned error: %v", err)
|
||||
}
|
||||
if gotCaller != callerID {
|
||||
t.Errorf("callerID = %q, want %q", gotCaller, callerID)
|
||||
}
|
||||
if gotWs != wsID {
|
||||
t.Errorf("workspaceID = %q, want %q", gotWs, wsID)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueRowAuthFields_NotFound(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
|
||||
mock.ExpectQuery(`SELECT caller_id, workspace_id FROM a2a_queue WHERE id`).
|
||||
WithArgs(queueID).
|
||||
WillReturnError(sql.ErrNoRows)
|
||||
|
||||
_, _, err := queueRowAuthFields(context.Background(), queueID)
|
||||
if err != sql.ErrNoRows {
|
||||
t.Errorf("expected sql.ErrNoRows, got %v", err)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueRowAuthFields_DBError(t *testing.T) {
|
||||
mock := setupQueueStatusDB(t)
|
||||
queueID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
|
||||
mock.ExpectQuery(`SELECT caller_id, workspace_id FROM a2a_queue WHERE id`).
|
||||
WithArgs(queueID).
|
||||
WillReturnError(sql.ErrConnDone)
|
||||
|
||||
_, _, err := queueRowAuthFields(context.Background(), queueID)
|
||||
if err == nil {
|
||||
t.Error("expected error, got nil")
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -516,3 +516,51 @@ func TestDrainQueueForWorkspace_ClaimGuarding_SecondDrainGetsEmpty(t *testing.T)
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// ── QueueDepth ──────────────────────────────────────────────────────────────────
|
||||
|
||||
func TestQueueDepth_ReturnsCount(t *testing.T) {
|
||||
mockDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create sqlmock: %v", err)
|
||||
}
|
||||
prevDB := db.DB
|
||||
db.DB = mockDB
|
||||
t.Cleanup(func() { db.DB = prevDB; mockDB.Close() })
|
||||
|
||||
wsID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
mock.ExpectQuery(`SELECT COUNT(*) FROM a2a_queue WHERE workspace_id = $1 AND status = 'queued'`).
|
||||
WithArgs(wsID).
|
||||
WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(42))
|
||||
|
||||
got := QueueDepth(context.Background(), wsID)
|
||||
if got != 42 {
|
||||
t.Errorf("QueueDepth returned %d, want 42", got)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueDepth_ZeroWhenEmpty(t *testing.T) {
|
||||
mockDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create sqlmock: %v", err)
|
||||
}
|
||||
prevDB := db.DB
|
||||
db.DB = mockDB
|
||||
t.Cleanup(func() { db.DB = prevDB; mockDB.Close() })
|
||||
|
||||
wsID := "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
|
||||
mock.ExpectQuery(`SELECT COUNT(*) FROM a2a_queue WHERE workspace_id = $1 AND status = 'queued'`).
|
||||
WithArgs(wsID).
|
||||
WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(0))
|
||||
|
||||
got := QueueDepth(context.Background(), wsID)
|
||||
if got != 0 {
|
||||
t.Errorf("QueueDepth returned %d, want 0", got)
|
||||
}
|
||||
if err := mock.ExpectationsWereMet(); err != nil {
|
||||
t.Errorf("unmet sqlmock expectations: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -947,6 +947,73 @@ func TestVerifyDiscordSignature_WrongLengthPubKey(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
// ==================== matchesChatID pure function ====================
|
||||
|
||||
func TestMatchesChatID_ExactMatch(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": "123456"}
|
||||
if !matchesChatID(cfg, "123456") {
|
||||
t.Error("expected true for exact match")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_NoMatch(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": "123456"}
|
||||
if matchesChatID(cfg, "654321") {
|
||||
t.Error("expected false for non-matching chat ID")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_PrefixNoMatch(t *testing.T) {
|
||||
// "123" is a prefix of "123456" but not an exact match.
|
||||
cfg := map[string]interface{}{"chat_id": "123456"}
|
||||
if matchesChatID(cfg, "123") {
|
||||
t.Error("expected false for prefix of stored chat ID")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_CommaSeparatedMultiple(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": "111,222,333"}
|
||||
for _, id := range []string{"111", "222", "333"} {
|
||||
if !matchesChatID(cfg, id) {
|
||||
t.Errorf("expected true for %q in comma-separated list", id)
|
||||
}
|
||||
}
|
||||
if matchesChatID(cfg, "444") {
|
||||
t.Error("expected false for ID not in list")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_WhitespaceTrimmed(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": "111, 222 , 333"}
|
||||
if !matchesChatID(cfg, "222") {
|
||||
t.Error("expected true for whitespace-trimmed match")
|
||||
}
|
||||
if matchesChatID(cfg, " 222") {
|
||||
t.Error("expected false for whitespace in query (not trimmed from query)")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_EmptyChatID(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": ""}
|
||||
if matchesChatID(cfg, "123456") {
|
||||
t.Error("expected false for empty chat_id in config")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_MissingChatIDKey(t *testing.T) {
|
||||
cfg := map[string]interface{}{}
|
||||
if matchesChatID(cfg, "123456") {
|
||||
t.Error("expected false when chat_id key is missing")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesChatID_NonStringChatID(t *testing.T) {
|
||||
cfg := map[string]interface{}{"chat_id": 123456} // wrong type
|
||||
if matchesChatID(cfg, "123456") {
|
||||
t.Error("expected false when chat_id is not a string")
|
||||
}
|
||||
}
|
||||
|
||||
// TestChannelHandler_Webhook_Discord_NoKey_Returns401 verifies that a Discord
|
||||
// webhook request is rejected with 401 when no public key is configured in the
|
||||
// DB and DISCORD_APP_PUBLIC_KEY env var is not set.
|
||||
|
||||
@@ -1,432 +0,0 @@
|
||||
"""BaseAdapter coverage gap tests — fills uncovered branches in adapter_base.py.
|
||||
|
||||
Covers:
|
||||
- resolve_provider_routing(): all URL-precedence branches + unknown prefix
|
||||
- RuntimeCapabilities.to_dict(): all flag combinations
|
||||
- BaseAdapter.capabilities(): returns RuntimeCapabilities() (platform-owns-everything)
|
||||
- BaseAdapter.idle_timeout_override(): returns None (use platform default)
|
||||
- BaseAdapter.get_config_schema(): returns {} (override per-subclass)
|
||||
- BaseAdapter.memory_filename(): returns "CLAUDE.md"
|
||||
- BaseAdapter.register_tool_hook(): no-op (override for dynamic registry)
|
||||
- BaseAdapter.register_subagent_hook(): no-op (override for DeepAgents)
|
||||
- BaseAdapter.transcript_lines(): returns supported=False dict
|
||||
- BaseAdapter.append_to_memory_hook(): idempotent append, marker deduplication
|
||||
- BaseAdapter.pre_stop_state(): captures session_id from executor + transcript_lines
|
||||
- BaseAdapter.restore_state(): stores session_id + transcript_lines from snapshot
|
||||
- BaseAdapter.inject_plugins(): delegates to install_plugins_via_registry
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
WORKSPACE_DIR = Path(__file__).parent.parent
|
||||
if str(WORKSPACE_DIR) not in sys.path:
|
||||
sys.path.insert(0, str(WORKSPACE_DIR))
|
||||
|
||||
from a2a.server.agent_execution import AgentExecutor
|
||||
|
||||
from adapter_base import (
|
||||
AdapterConfig,
|
||||
BaseAdapter,
|
||||
ProviderRegistry,
|
||||
RuntimeCapabilities,
|
||||
resolve_provider_routing,
|
||||
)
|
||||
|
||||
|
||||
class _StubAdapter(BaseAdapter):
|
||||
"""Minimal concrete adapter for testing base-class default behaviour."""
|
||||
|
||||
@staticmethod
|
||||
def name() -> str:
|
||||
return "stub"
|
||||
|
||||
@staticmethod
|
||||
def display_name() -> str:
|
||||
return "Stub"
|
||||
|
||||
@staticmethod
|
||||
def description() -> str:
|
||||
return "test stub"
|
||||
|
||||
async def setup(self, config: AdapterConfig) -> None:
|
||||
return None
|
||||
|
||||
async def create_executor(self, config: AdapterConfig) -> AgentExecutor: # pragma: no cover
|
||||
raise NotImplementedError
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# resolve_provider_routing tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_resolve_provider_routing_parses_prefix_and_model():
|
||||
"""'anthropic:claude-sonnet-4-6' splits into prefix + bare model."""
|
||||
api_key, base_url, model_id = resolve_provider_routing(
|
||||
"anthropic:claude-sonnet-4-6",
|
||||
{"ANTHROPIC_API_KEY": "sk-ant-test"},
|
||||
registry={"anthropic": (("ANTHROPIC_API_KEY",), "https://api.anthropic.com")},
|
||||
)
|
||||
assert api_key == "sk-ant-test"
|
||||
assert base_url == "https://api.anthropic.com"
|
||||
assert model_id == "claude-sonnet-4-6"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_falls_back_to_openai():
|
||||
"""Bare model without colon defaults to openai prefix."""
|
||||
api_key, base_url, model_id = resolve_provider_routing(
|
||||
"gpt-4o",
|
||||
{"OPENAI_API_KEY": "sk-openai-test"},
|
||||
registry={},
|
||||
)
|
||||
assert api_key == "sk-openai-test"
|
||||
assert base_url == "https://api.openai.com/v1"
|
||||
assert model_id == "gpt-4o"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_url_from_env_var():
|
||||
"""PREFIX_BASE_URL env var takes precedence over registry default."""
|
||||
env = {
|
||||
"OPENAI_API_KEY": "sk-test",
|
||||
"OPENAI_BASE_URL": "https://my-proxy.example.com/v1",
|
||||
}
|
||||
api_key, base_url, model_id = resolve_provider_routing(
|
||||
"openai:gpt-4o", env, registry={}
|
||||
)
|
||||
assert base_url == "https://my-proxy.example.com/v1"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_url_from_runtime_config():
|
||||
"""runtime_config['provider_url'] takes precedence over registry default."""
|
||||
env = {"OPENAI_API_KEY": "sk-test"}
|
||||
api_key, base_url, model_id = resolve_provider_routing(
|
||||
"openai:gpt-4o",
|
||||
env,
|
||||
registry={},
|
||||
runtime_config={"provider_url": "https://config-proxy.example.com/v1"},
|
||||
)
|
||||
assert base_url == "https://config-proxy.example.com/v1"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_env_overrides_runtime_config():
|
||||
"""env var PREFIX_BASE_URL wins over runtime_config['provider_url']."""
|
||||
env = {
|
||||
"OPENAI_API_KEY": "sk-test",
|
||||
"OPENAI_BASE_URL": "https://env-proxy.example.com/v1",
|
||||
}
|
||||
_, base_url, _ = resolve_provider_routing(
|
||||
"openai:gpt-4o",
|
||||
env,
|
||||
registry={},
|
||||
runtime_config={"provider_url": "https://config-proxy.example.com/v1"},
|
||||
)
|
||||
assert base_url == "https://env-proxy.example.com/v1"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_falls_back_to_openai_on_unknown_prefix():
|
||||
"""Unknown provider prefix falls back to OPENAI_API_KEY + openai.com."""
|
||||
env = {"OPENAI_API_KEY": "sk-fallback"}
|
||||
api_key, base_url, model_id = resolve_provider_routing(
|
||||
"unknown:some-model", env, registry={}
|
||||
)
|
||||
assert api_key == "sk-fallback"
|
||||
assert base_url == "https://api.openai.com/v1"
|
||||
assert model_id == "some-model"
|
||||
|
||||
|
||||
def test_resolve_provider_routing_raises_when_no_api_key():
|
||||
"""RuntimeError raised when no API key env var is set for the prefix."""
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
resolve_provider_routing(
|
||||
"anthropic:claude-sonnet-4-6",
|
||||
{}, # empty env — no ANTHROPIC_API_KEY
|
||||
registry={"anthropic": (("ANTHROPIC_API_KEY",), "https://api.anthropic.com")},
|
||||
)
|
||||
assert "No API key found" in str(exc_info.value)
|
||||
assert "anthropic" in str(exc_info.value)
|
||||
|
||||
|
||||
def test_resolve_provider_routing_multiple_env_vars_first_found():
|
||||
"""registry tuple with multiple env vars — first present in env is used."""
|
||||
env = {
|
||||
# ANTHROPIC_API_KEY not set; ANTHROPIC_SECONDARY_KEY is
|
||||
"ANTHROPIC_SECONDARY_KEY": "sk-secondary",
|
||||
}
|
||||
api_key, _, _ = resolve_provider_routing(
|
||||
"anthropic:claude-sonnet-4-6",
|
||||
env,
|
||||
registry={"anthropic": (("ANTHROPIC_API_KEY", "ANTHROPIC_SECONDARY_KEY"), "https://api.anthropic.com")},
|
||||
)
|
||||
assert api_key == "sk-secondary"
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# RuntimeCapabilities tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_runtime_capabilities_to_dict_all_defaults():
|
||||
"""All flags default to False."""
|
||||
caps = RuntimeCapabilities()
|
||||
d = caps.to_dict()
|
||||
assert d == {
|
||||
"heartbeat": False,
|
||||
"scheduler": False,
|
||||
"session": False,
|
||||
"status_mgmt": False,
|
||||
"retry": False,
|
||||
"activity_decoration": False,
|
||||
"channel_dispatch": False,
|
||||
}
|
||||
|
||||
|
||||
def test_runtime_capabilities_to_dict_all_true():
|
||||
"""All flags can be set to True."""
|
||||
caps = RuntimeCapabilities(
|
||||
provides_native_heartbeat=True,
|
||||
provides_native_scheduler=True,
|
||||
provides_native_session=True,
|
||||
provides_native_status_mgmt=True,
|
||||
provides_native_retry=True,
|
||||
provides_activity_decoration=True,
|
||||
provides_channel_dispatch=True,
|
||||
)
|
||||
d = caps.to_dict()
|
||||
assert all(v is True for v in d.values())
|
||||
|
||||
|
||||
def test_runtime_capabilities_partial_flags():
|
||||
"""Partial flag set — only heartbeat and session True."""
|
||||
caps = RuntimeCapabilities(
|
||||
provides_native_heartbeat=True,
|
||||
provides_native_session=True,
|
||||
)
|
||||
d = caps.to_dict()
|
||||
assert d["heartbeat"] is True
|
||||
assert d["session"] is True
|
||||
assert d["scheduler"] is False
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# BaseAdapter method default behaviour tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_capabilities_returns_empty_runtime_capabilities():
|
||||
"""Default capabilities() returns RuntimeCapabilities() with all flags off."""
|
||||
adapter = _StubAdapter()
|
||||
caps = adapter.capabilities()
|
||||
assert isinstance(caps, RuntimeCapabilities)
|
||||
d = caps.to_dict()
|
||||
assert all(v is False for v in d.values())
|
||||
|
||||
|
||||
def test_idle_timeout_override_returns_none():
|
||||
"""Default idle_timeout_override() returns None — use platform default."""
|
||||
adapter = _StubAdapter()
|
||||
assert adapter.idle_timeout_override() is None
|
||||
|
||||
|
||||
def test_get_config_schema_returns_empty_dict():
|
||||
"""Default get_config_schema() returns {} — override per-subclass."""
|
||||
adapter = _StubAdapter()
|
||||
assert adapter.get_config_schema() == {}
|
||||
|
||||
|
||||
def test_memory_filename_returns_claude_md():
|
||||
"""Default memory_filename() returns 'CLAUDE.md'."""
|
||||
adapter = _StubAdapter()
|
||||
assert adapter.memory_filename() == "CLAUDE.md"
|
||||
|
||||
|
||||
def test_register_tool_hook_returns_none():
|
||||
"""Default register_tool_hook() is a no-op that returns None."""
|
||||
adapter = _StubAdapter()
|
||||
result = adapter.register_tool_hook("some-plugin", MagicMock())
|
||||
assert result is None
|
||||
|
||||
|
||||
def test_register_subagent_hook_returns_none():
|
||||
"""Default register_subagent_hook() is a no-op that returns None."""
|
||||
adapter = _StubAdapter()
|
||||
result = adapter.register_subagent_hook("deep-agent", {"name": "agent"})
|
||||
assert result is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_transcript_lines_returns_unsupported():
|
||||
"""Default transcript_lines() returns supported=False (runtime doesn't expose a log)."""
|
||||
adapter = _StubAdapter()
|
||||
result = await adapter.transcript_lines(since=10, limit=50)
|
||||
assert result["supported"] is False
|
||||
assert result["lines"] == []
|
||||
assert result["cursor"] == 10 # preserved from since arg
|
||||
assert result["more"] is False
|
||||
assert result["source"] is None
|
||||
assert result["runtime"] == "stub"
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# append_to_memory_hook tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_append_to_memory_hook_creates_new_file():
|
||||
"""append_to_memory_hook creates the target file if it doesn't exist."""
|
||||
adapter = _StubAdapter()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
config = AdapterConfig(model="test", config_path=tmpdir)
|
||||
content = "# Plugin: test-plugin\nsome content"
|
||||
adapter.append_to_memory_hook(config, "CLAUDE.md", content)
|
||||
|
||||
path = os.path.join(tmpdir, "CLAUDE.md")
|
||||
assert os.path.exists(path)
|
||||
with open(path) as f:
|
||||
assert content in f.read()
|
||||
|
||||
|
||||
def test_append_to_memory_hook_idempotent_with_marker():
|
||||
"""Second append with same marker is skipped (idempotent)."""
|
||||
adapter = _StubAdapter()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
config = AdapterConfig(model="test", config_path=tmpdir)
|
||||
marker_content = "# Plugin: test-plugin\nsome content"
|
||||
|
||||
adapter.append_to_memory_hook(config, "CLAUDE.md", marker_content)
|
||||
adapter.append_to_memory_hook(config, "CLAUDE.md", marker_content)
|
||||
|
||||
path = os.path.join(tmpdir, "CLAUDE.md")
|
||||
with open(path) as f:
|
||||
text = f.read()
|
||||
# Should appear only once (second append skipped)
|
||||
lines = [l for l in text.splitlines() if l.startswith("# Plugin: test-plugin")]
|
||||
assert len(lines) == 1
|
||||
|
||||
|
||||
def test_append_to_memory_hook_appends_without_marker():
|
||||
"""Appends when the marker line is not present (no deduplication needed)."""
|
||||
adapter = _StubAdapter()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
config = AdapterConfig(model="test", config_path=tmpdir)
|
||||
|
||||
adapter.append_to_memory_hook(config, "CLAUDE.md", "# First plugin\ncontent A")
|
||||
adapter.append_to_memory_hook(config, "CLAUDE.md", "# Second plugin\ncontent B")
|
||||
|
||||
path = os.path.join(tmpdir, "CLAUDE.md")
|
||||
with open(path) as f:
|
||||
text = f.read()
|
||||
assert "# First plugin" in text
|
||||
assert "# Second plugin" in text
|
||||
|
||||
|
||||
def test_append_to_memory_hook_creates_parent_dirs():
|
||||
"""append_to_memory_hook creates intermediate directories."""
|
||||
adapter = _StubAdapter()
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
config = AdapterConfig(model="test", config_path=tmpdir)
|
||||
adapter.append_to_memory_hook(config, "subdir/CLAUDE.md", "# Nested")
|
||||
|
||||
path = os.path.join(tmpdir, "subdir", "CLAUDE.md")
|
||||
assert os.path.exists(path)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# pre_stop_state tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_pre_stop_state_empty_when_no_executor():
|
||||
"""pre_stop_state returns {} when no _executor is attached."""
|
||||
adapter = _StubAdapter()
|
||||
state = adapter.pre_stop_state()
|
||||
assert state == {}
|
||||
|
||||
|
||||
def test_pre_stop_state_captures_session_id():
|
||||
"""pre_stop_state reads _executor._session_id when present."""
|
||||
adapter = _StubAdapter()
|
||||
mock_executor = MagicMock(spec=AgentExecutor)
|
||||
mock_executor._session_id = "session-abc123"
|
||||
adapter._executor = mock_executor
|
||||
|
||||
state = adapter.pre_stop_state()
|
||||
assert state["session_id"] == "session-abc123"
|
||||
|
||||
|
||||
def test_pre_stop_state_captures_transcript_lines():
|
||||
"""pre_stop_state calls transcript_lines() and includes lines when supported."""
|
||||
adapter = _StubAdapter()
|
||||
adapter._executor = None # no session_id
|
||||
|
||||
# Override transcript_lines to return supported=True
|
||||
adapter.transcript_lines = MagicMock(return_value={
|
||||
"runtime": "stub",
|
||||
"supported": True,
|
||||
"lines": [{"role": "user", "content": "hello"}],
|
||||
"cursor": 0,
|
||||
"more": False,
|
||||
"source": "/tmp/transcript.jsonl",
|
||||
})
|
||||
|
||||
state = adapter.pre_stop_state()
|
||||
assert state["transcript_lines"] == [{"role": "user", "content": "hello"}]
|
||||
|
||||
|
||||
def test_pre_stop_state_suppresses_transcript_on_exception():
|
||||
"""pre_stop_state never raises — transcript capture is best-effort."""
|
||||
adapter = _StubAdapter()
|
||||
adapter._executor = None
|
||||
|
||||
def broken_transcript(*args, **kwargs):
|
||||
raise RuntimeError("disk error")
|
||||
|
||||
adapter.transcript_lines = broken_transcript
|
||||
|
||||
# Must not raise
|
||||
state = adapter.pre_stop_state()
|
||||
assert state == {}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# restore_state tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def test_restore_state_stores_session_id():
|
||||
"""restore_state stores snapshot['session_id'] as _snapshot_session_id."""
|
||||
adapter = _StubAdapter()
|
||||
adapter.restore_state({"session_id": "restored-session-xyz"})
|
||||
assert adapter._snapshot_session_id == "restored-session-xyz"
|
||||
|
||||
|
||||
def test_restore_state_stores_transcript_lines():
|
||||
"""restore_state stores snapshot['transcript_lines'] as _snapshot_transcript."""
|
||||
adapter = _StubAdapter()
|
||||
lines = [{"role": "user", "content": "prior context"}]
|
||||
adapter.restore_state({"transcript_lines": lines})
|
||||
assert adapter._snapshot_transcript == lines
|
||||
|
||||
|
||||
def test_restore_state_handles_missing_keys():
|
||||
"""restore_state works when snapshot lacks session_id or transcript_lines."""
|
||||
adapter = _StubAdapter()
|
||||
adapter.restore_state({})
|
||||
assert adapter._snapshot_session_id is None
|
||||
assert adapter._snapshot_transcript is None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# inject_plugins tests
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_inject_plugins_delegates_to_install_plugins_via_registry():
|
||||
"""inject_plugins calls install_plugins_via_registry (default migration path)."""
|
||||
from unittest.mock import AsyncMock
|
||||
adapter = _StubAdapter()
|
||||
|
||||
with patch.object(adapter, "install_plugins_via_registry", new_callable=AsyncMock) as mock_install:
|
||||
mock_install.return_value = []
|
||||
await adapter.inject_plugins(AdapterConfig(model="test", config_path="/tmp"), MagicMock())
|
||||
mock_install.assert_called_once()
|
||||
Reference in New Issue
Block a user