def test_from_iterable():
stream = Stream.from_iterable(range(10))
L = stream.sink_to_list()
stream.start()
time.sleep(0.01)
assert L == list(range(10))
def test_ws_roundtrip():
pytest.importorskip("websockets")
s0 = Stream.from_websocket("localhost", 8989, start=True)
l = s0.sink_to_list()
data = [b'0123'] * 4
s = Stream.from_iterable(data)
s.to_websocket("ws://localhost:8989")
s.start()
wait_for(lambda: data == l, timeout=1)
s.stop()
s0.stop()
def test_mqtt_roundtrip():
pytest.importorskip("paho.mqtt.client")
s0 = Stream.from_mqtt("mqtt.eclipseprojects.io", 1883,
"streamz/sensor/temperature")
l = s0.map(lambda msg: msg.payload).sink_to_list()
s0.start()
data = [b'0123'] * 4
s = Stream.from_iterable(data)
s.to_mqtt("mqtt.eclipseprojects.io", 1883, "streamz/sensor/temperature")
s.start()
wait_for(lambda: data == l, timeout=1)
s.stop()
s0.stop()
