Example #1
0
    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
Example #2
0
    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