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()