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