stash

aircox_streamer/tests/conftest.py

@@ -0,0 +1,51 @@
+import pytest
+
+from aircox_streamer import connector
+
+
+class FakeSocket:
+    FAILING_ADDRESS = -1
+    """Connect with this address fails."""
+
+    family, type, address = None, None, None
+    sent_data = None
+    """List of data that have been `send[all]`"""
+    recv_data = None
+    """Response data to return on recv."""
+
+    def __init__(self, family, type):
+        self.family = family
+        self.type = type
+        self.sent_data = []
+
+    def connect(self, address):
+        if address == self.FAILING_ADDRESS:
+            raise RuntimeError("invalid connection")
+        self.address = address
+
+    def close(self):
+        pass
+
+    def sendall(self, data):
+        self.sent_data.append(data)
+
+    def recv(self, count):
+        data = self.recv_data[:count]
+        self.recv_data = data[count:]
+        return data.encode("utf-8") if isinstance(data, str) else data
+
+
+class Connector(connector.Connector):
+    socket_class = FakeSocket
+
+
+@pytest.fixture
+def connector(request):
+    obj = Connector("test")
+    yield obj
+    obj.close()
+
+
+@pytest.fixture
+def fail_connector():
+    return Connector(FakeSocket.FAILING_ADDRESS)