Class TestUserSocketFunctionality

tests/test_user_socket_integration.py:102–129

Tests verifying user socket functionality works correctly.

class TestUserSocketFunctionality:
    """Tests verifying user socket functionality works correctly."""

    @pytest.mark.asyncio
    async def test_user_socket_recv_timeout(self, clientAsync, socket_manager):
        """User socket recv() should timeout gracefully when no events."""
        user_socket = socket_manager.user_socket()

        async with user_socket:
            # recv() should timeout without errors (no events on quiet account)
            with pytest.raises(asyncio.TimeoutError):
                await asyncio.wait_for(user_socket.recv(), timeout=2)

    @pytest.mark.asyncio
    async def test_user_socket_context_manager(self, clientAsync, socket_manager):
        """User socket should work as async context manager."""
        user_socket = socket_manager.user_socket()

        # Should not be connected initially
        assert user_socket._subscription_id is None

        async with user_socket:
            # Should be connected inside context
            assert user_socket._subscription_id is not None
            assert user_socket._uses_ws_api_subscription is True

        # Subscription ID is cleared after unsubscribe
        assert user_socket._subscription_id is None


class TestNonUserSockets: