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
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_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
