mirror of
https://github.com/OMGeeky/flucto-heisskleber.git
synced 2025-12-28 07:18:09 +01:00
774e0d8496
This commit is a lot and adds breaking changes. Also adds a UDP subscriber and publisher.
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
import json
|
|
import socket
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from heisskleber.udp.config import UdpConf
|
|
from heisskleber.udp.publisher import UdpPublisher
|
|
from heisskleber.udp.subscriber import UdpSubscriber
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_socket():
|
|
with patch("heisskleber.udp.publisher.socket.socket") as mock_socket:
|
|
yield mock_socket
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_conf():
|
|
return UdpConf(ip="127.0.0.1", port=12345, packer="json")
|
|
|
|
|
|
def test_connects_to_socket(mock_socket, mock_conf) -> None:
|
|
_ = UdpPublisher(mock_conf)
|
|
|
|
# constructor was called
|
|
mock_socket.assert_called_with(socket.AF_INET, socket.SOCK_DGRAM)
|
|
|
|
|
|
def test_closes_socket(mock_socket, mock_conf) -> None:
|
|
pub = UdpPublisher(mock_conf)
|
|
del pub
|
|
|
|
# instace was closed
|
|
mock_socket.return_value.close.assert_called()
|
|
|
|
|
|
def test_packs_and_sends_message(mock_socket, mock_conf) -> None:
|
|
pub = UdpPublisher(mock_conf)
|
|
|
|
# explicitly define packer to be json.dumps
|
|
assert pub.pack == json.dumps
|
|
|
|
pub.send({"key": "val", "intkey": 1, "floatkey": 1.0}, "test")
|
|
|
|
mock_socket.return_value.sendto.assert_called_with(
|
|
b'{"key": "val", "intkey": 1, "floatkey": 1.0, "topic": "test"}',
|
|
(str(mock_conf.ip), mock_conf.port),
|
|
)
|
|
|
|
|
|
def test_subscriber_receives_message_from_queue(mock_conf) -> None:
|
|
sub = UdpSubscriber(mock_conf)
|
|
|
|
test_topic, test_data = ("test", {"key": "val", "intkey": 1, "floatkey": 1.0})
|
|
|
|
sub._queue.put((test_topic, test_data))
|
|
|
|
topic, data = sub.receive()
|
|
assert test_topic == topic
|
|
assert test_data == data
|
|
|
|
|
|
@pytest.fixture
|
|
def udp_sub(mock_conf):
|
|
sub = UdpSubscriber(mock_conf)
|
|
sub.start_loop()
|
|
yield sub
|
|
|
|
|
|
def test_sends_message_between_pub_and_sub(udp_sub, mock_conf):
|
|
pub = UdpPublisher(mock_conf)
|
|
test_data = {"key": "val", "intkey": 1, "floatkey": 1.0}
|
|
test_topic = "test_topic"
|
|
|
|
# Need to copy the dict, because the publisher will mutate it
|
|
pub.send(test_data.copy(), test_topic)
|
|
topic, data = udp_sub.receive()
|
|
assert test_topic == topic
|
|
assert test_data == data
|