Python StreamingClientMock.get_sent示例

示例#1

文件： test_output_writer.py 项目： microsoft/agogosml

def test_broadcast_success(*fixtures):
    mock1 = StreamingClientMock()
    mock2 = StreamingClientMock()
    broadcast = BroadcastStreamingClient({'BROADCAST_CLIENTS': [mock1, mock2]})

    success = broadcast.send({'test': 'message'})

    assert success
    assert mock1.get_sent() and mock2.get_sent()

示例#2

文件： test_output_writer.py 项目： microsoft/agogosml

def test_broadcast_failure(*fixtures):
    mock1 = StreamingClientMock()
    mock2 = StreamingClientMock()
    mock3 = StreamingClientMock()
    broadcast = BroadcastStreamingClient(
        {'BROADCAST_CLIENTS': [mock1, mock2, mock3]})

    mock2.set_fail_send(True)
    success = broadcast.send({'test': 'message'})

    assert not success
    assert mock1.get_sent() and not mock2.get_sent() and mock3.get_sent()