test: skeleton for an end-to-end unittest

[loctrkd.git] / test / test_storage.py

diff --git a/test/test_storage.py b/test/test_storage.py
new file mode 100644 (file)
index 0000000..0c5e260
--- /dev/null
+++ b/test/test_storage.py
@@ -0,0 +1,42 @@
+""" Send junk to the collector """
+
+from random import Random
+from socket import getaddrinfo, socket, AF_INET6, SOCK_STREAM
+from sqlite3 import connect
+from time import sleep
+import unittest
+from .common import send_and_drain, TestWithServers
+
+
+class Storage(TestWithServers):
+    def setUp(self, *args: str) -> None:
+        super().setUp("collector", "storage", "lookaside", "termconfig")
+        for fam, typ, pro, cnm, skadr in getaddrinfo(
+            "::1",
+            self.conf.getint("collector", "port"),
+            family=AF_INET6,
+            type=SOCK_STREAM,
+        ):
+            break  # Just take the first element
+        self.sock = socket(AF_INET6, SOCK_STREAM)
+        self.sock.connect(skadr)
+
+    def tearDown(self) -> None:
+        sleep(1)  # give collector some time
+        send_and_drain(self.sock, None)
+        self.sock.close()
+        super().tearDown()
+
+    def test_storage(self) -> None:
+        buf = b"xx\r\x01\x03Y3\x90w\x19q\x85\x05\r\n"
+        send_and_drain(self.sock, buf)
+        # TODO: make a proper sequence
+        with connect(self.conf.get("storage", "dbfn")) as db:
+            c = db.cursor()
+            c.execute("select * from events")
+            events = [dict(row) for row in c]
+        print(events)
+
+
+if __name__ == "__main__":
+    unittest.main()