def test_connection_made_and_lost(self): i = ICAPProtocol(None) transport = MagicMock() assert not i.connected i.connection_made(transport) assert i.transport == transport assert i.connected i.connection_lost(None) assert not i.connected
def test_write_response__when_disconnected(self): i = ICAPProtocol(False) with patch('icap.asyncio.Serializer') as mock_serializer: i.write_response(MagicMock(), MagicMock()) assert not mock_serializer.mock_calls i.connection_made(MagicMock()) with patch('icap.asyncio.Serializer') as mock_serializer: i.write_response(MagicMock(), MagicMock()) assert mock_serializer.mock_calls
def test_validate_request_aborts_400_for_non_icap(self): request_line = RequestLine("REQMOD", "/", "HTTP/1.1") request = ICAPRequest(request_line) try: ICAPProtocol(None).validate_request(request) except ICAPAbort as e: assert e.status_code == 400 else: assert False m = MagicMock(is_request=False) try: ICAPProtocol(None).validate_request(m) except ICAPAbort as e: assert e.status_code == 400 else: assert False