refactor: clean up imports and improve session key assertion formatting

This commit is contained in:
Abhimanyu Saharan
2026-02-11 10:47:32 +05:30
parent c09116b8f0
commit c3490630a4
5 changed files with 11 additions and 22 deletions

View File

@@ -16,7 +16,9 @@ from app.services.openclaw.shared import GatewayAgentIdentity
def test_gateway_main_session_key_matches_gateway_identity() -> None:
gateway_id = UUID("00000000-0000-0000-0000-000000000123")
assert gateway_main_session_key(gateway_id) == GatewayAgentIdentity.session_key_for_id(gateway_id)
assert gateway_main_session_key(gateway_id) == GatewayAgentIdentity.session_key_for_id(
gateway_id
)
def test_board_lead_session_key_format() -> None:
@@ -32,16 +34,14 @@ def test_board_agent_session_key_format() -> None:
def test_board_scoped_session_key_selects_lead() -> None:
agent_id = UUID("00000000-0000-0000-0000-000000000001")
board_id = UUID("00000000-0000-0000-0000-000000000002")
assert (
board_scoped_session_key(agent_id=agent_id, board_id=board_id, is_board_lead=True)
== board_lead_session_key(board_id)
)
assert board_scoped_session_key(
agent_id=agent_id, board_id=board_id, is_board_lead=True
) == board_lead_session_key(board_id)
def test_board_scoped_session_key_selects_non_lead() -> None:
agent_id = UUID("00000000-0000-0000-0000-000000000001")
board_id = UUID("00000000-0000-0000-0000-000000000002")
assert (
board_scoped_session_key(agent_id=agent_id, board_id=board_id, is_board_lead=False)
== board_agent_session_key(agent_id)
)
assert board_scoped_session_key(
agent_id=agent_id, board_id=board_id, is_board_lead=False
) == board_agent_session_key(agent_id)