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

@@ -32,8 +32,8 @@ from app.schemas.board_onboarding import (
BoardOnboardingUserProfile,
)
from app.schemas.boards import BoardRead
from app.services.openclaw.gateway_resolver import get_gateway_for_board
from app.services.openclaw.gateway_dispatch import GatewayDispatchService
from app.services.openclaw.gateway_resolver import get_gateway_for_board
from app.services.openclaw.onboarding_service import BoardOnboardingMessagingService
from app.services.openclaw.policies import OpenClawAuthorizationPolicy
from app.services.openclaw.provisioning_db import (

View File

@@ -20,19 +20,9 @@ from app.core.time import utcnow
from app.db import crud
from app.db.pagination import paginate
from app.db.session import get_session
from app.models.activity_events import ActivityEvent
from app.models.agents import Agent
from app.models.approvals import Approval
from app.models.board_groups import BoardGroup
from app.models.board_memory import BoardMemory
from app.models.board_onboarding import BoardOnboardingSession
from app.models.boards import Board
from app.models.gateways import Gateway
from app.models.organization_board_access import OrganizationBoardAccess
from app.models.organization_invite_board_access import OrganizationInviteBoardAccess
from app.models.task_dependencies import TaskDependency
from app.models.task_fingerprints import TaskFingerprint
from app.models.tasks import Task
from app.schemas.boards import BoardCreate, BoardRead, BoardUpdate
from app.schemas.common import OkResponse
from app.schemas.pagination import DefaultLimitOffsetPage

View File

@@ -109,4 +109,3 @@ async def delete_board(session: AsyncSession, *, board: Board) -> OkResponse:
await session.delete(board)
await session.commit()
return OkResponse()

View File

@@ -12,8 +12,8 @@ from app.models.boards import Board
from app.models.gateways import Gateway
from app.services.openclaw.db_service import OpenClawDBService
from app.services.openclaw.gateway_resolver import (
get_gateway_for_board,
gateway_client_config,
get_gateway_for_board,
optional_gateway_client_config,
require_gateway_for_board,
)

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)