feat: update openclaw workspace integration
This commit is contained in:
@@ -311,6 +311,17 @@ class TestRiskAgent:
|
||||
|
||||
|
||||
class TestStorageService:
|
||||
def test_storage_service_defaults_to_live_config(self):
|
||||
from backend.services.storage import StorageService
|
||||
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
storage = StorageService(
|
||||
dashboard_dir=Path(tmpdir),
|
||||
initial_cash=100000.0,
|
||||
)
|
||||
|
||||
assert storage.config_name == "live"
|
||||
|
||||
def test_calculate_portfolio_value_cash_only(self):
|
||||
from backend.services.storage import StorageService
|
||||
|
||||
|
||||
74
backend/tests/test_openclaw_websocket_client.py
Normal file
74
backend/tests/test_openclaw_websocket_client.py
Normal file
@@ -0,0 +1,74 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""Tests for the OpenClaw WebSocket client session helpers."""
|
||||
|
||||
import pytest
|
||||
|
||||
from shared.client.openclaw_websocket_client import OpenClawWebSocketClient
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_resolve_session_parses_gateway_key_response():
|
||||
client = OpenClawWebSocketClient(gateway_token="test-token")
|
||||
|
||||
async def fake_send_request(method, params=None, _allow_handshake=False):
|
||||
assert method == "sessions.resolve"
|
||||
assert params["agentId"] == "main"
|
||||
return {"ok": True, "key": "agent:main:main"}
|
||||
|
||||
client._send_request = fake_send_request # type: ignore[method-assign]
|
||||
|
||||
resolved = await client.resolve_session(agent_id="main")
|
||||
|
||||
assert resolved == "agent:main:main"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_message_uses_session_send_payload():
|
||||
client = OpenClawWebSocketClient(gateway_token="test-token")
|
||||
|
||||
async def fake_send_request(method, params=None, _allow_handshake=False):
|
||||
assert method == "sessions.send"
|
||||
assert params == {
|
||||
"key": "agent:main:main",
|
||||
"message": "hello",
|
||||
"thinking": "medium",
|
||||
}
|
||||
return {"ok": True, "runId": "run-1"}
|
||||
|
||||
client._send_request = fake_send_request # type: ignore[method-assign]
|
||||
|
||||
result = await client.send_message("agent:main:main", "hello", thinking="medium")
|
||||
|
||||
assert result["runId"] == "run-1"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_session_history_uses_sessions_preview():
|
||||
client = OpenClawWebSocketClient(gateway_token="test-token")
|
||||
|
||||
async def fake_send_request(method, params=None, _allow_handshake=False):
|
||||
assert method == "sessions.preview"
|
||||
assert params == {"keys": ["agent:main:main"], "limit": 12}
|
||||
return {"previews": []}
|
||||
|
||||
client._send_request = fake_send_request # type: ignore[method-assign]
|
||||
|
||||
result = await client.get_session_history("agent:main:main", limit=12)
|
||||
|
||||
assert result == {"previews": []}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_unsubscribe_uses_session_messages_unsubscribe():
|
||||
client = OpenClawWebSocketClient(gateway_token="test-token")
|
||||
|
||||
async def fake_send_request(method, params=None, _allow_handshake=False):
|
||||
assert method == "sessions.messages.unsubscribe"
|
||||
assert params == {"key": "agent:main:main"}
|
||||
return {"subscribed": False}
|
||||
|
||||
client._send_request = fake_send_request # type: ignore[method-assign]
|
||||
|
||||
result = await client.unsubscribe("agent:main:main")
|
||||
|
||||
assert result == {"subscribed": False}
|
||||
Reference in New Issue
Block a user