Python download Examples

Example #1

def test_client_download_timeout():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
        }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = False
        with pytest.raises(CompoundPiSendTimeout):
            client.download('192.168.0.1', 0, io.BytesIO())

Example #2

def test_client_download_ok():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
        }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = True
        client.download('192.168.0.1', 0, io.BytesIO())
        l.assert_called_once_with('SEND 0,5647', ['192.168.0.1'])

Example #3

File: test_client.py Project: 4leavedclover/compoundpi

def test_client_download_timeout():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
            }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = False
        with pytest.raises(CompoundPiSendTimeout):
            client.download('192.168.0.1', 0, io.BytesIO())

Example #4

File: test_client.py Project: 4leavedclover/compoundpi

def test_client_download_ok():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
            }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = True
        client.download('192.168.0.1', 0, io.BytesIO())
        l.assert_called_once_with('SEND 0,5647', ['192.168.0.1'])

Example #5

def test_client_download_exception():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
        }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = True
        client._server.exception = ValueError('Foo')
        with pytest.raises(ValueError) as excinfo:
            client.download('192.168.0.1', 0, io.BytesIO())
            assert excinfo.value.args == ('Foo', )

Example #6

File: test_client.py Project: 4leavedclover/compoundpi

def test_client_download_exception():
    def download_server_effect(bind, handler):
        return Mock(**{'socket.getsockname.return_value': bind})
    with patch('compoundpi.client.CompoundPiDownloadServer', side_effect=download_server_effect), \
            patch('compoundpi.client.CompoundPiServerList.transact') as l:
        l.return_value = {
            compoundpi.client.IPv4Address('192.168.0.1'): None,
            }
        client = compoundpi.client.CompoundPiClient()
        client._server.event = Mock()
        client._server.event.wait.return_value = True
        client._server.exception = ValueError('Foo')
        with pytest.raises(ValueError) as excinfo:
            client.download('192.168.0.1', 0, io.BytesIO())
            assert excinfo.value.args == ('Foo',)