added unit test

AlisaLC · Feb 13, 2024 · 54307f6 · 54307f6
1 parent 7fa8d4b
commit 54307f6
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 3 deletions.
diff --git a/test/client.py → test/dummy_client.py b/test/client.py → test/dummy_client.py
diff --git a/test/fault_tolerance.py b/test/fault_tolerance.py
@@ -10,7 +10,7 @@
 from typing import List
 from threading import Lock
 
-from client import pull, push, subscribe
+from dummy_client import pull, push, subscribe
 
 TEST_SIZE = 1000 * 1000
 KEY_SIZE = 8

diff --git a/test/order.py b/test/order.py
@@ -5,7 +5,7 @@
 import sys
 from typing import Dict, List
 
-from client import pull, push, subscribe
+from dummy_client import pull, push, subscribe
 
 TEST_SIZE = 1000 * 1000
 KEY_SIZE = 8

diff --git a/test/scale.py b/test/scale.py
@@ -4,7 +4,7 @@
 ### press enter once more and see if you have a increase in throughput rate of the cluster
 
 import multiprocessing
-from client import pull, push, subscribe
+from dummy_client import pull, push, subscribe
 
 TEST_SIZE = 1000 * 1000
 KEY_SIZE = 8

diff --git a/test/unit_test.py b/test/unit_test.py
@@ -0,0 +1,13 @@
+import unittest
+
+from client.python import kafka_client as python_client
+from client.go import client as go_client
+
+
+# simple test which gets passed for all clients
+class TestClient(unittest.TestCase):
+    def always_pass(self):
+        self.assertTrue(True)
+
+if __name__ == '__main__':
+    unittest.main()