コード例 #1
0
ファイル: test_ipc.py プロジェクト: AbelMon/librarian
def test_send_timeout(open_socket):
    mocked_socket = mock.Mock()
    mocked_socket.send.side_effect = mod.socket.timeout
    ctx_manager = mock.MagicMock()
    ctx_manager.__enter__.return_value = mocked_socket
    open_socket.return_value = ctx_manager

    result = mod.send('some data')
    assert result is None
コード例 #2
0
ファイル: test_ipc.py プロジェクト: AbelMon/librarian
def test_send_success(open_socket):
    data = '<xml />'

    def mocked_recv(size):
        if hasattr(mocked_recv, 'called'):
            return '\0'

        mocked_recv.called = True
        return data

    mocked_socket = mock.Mock()
    mocked_socket.recv.side_effect = mocked_recv
    ctx_manager = mock.MagicMock()
    ctx_manager.__enter__.return_value = mocked_socket
    open_socket.return_value = ctx_manager

    result = mod.send(data)
    assert ET.tostring(result) == data