From d427c27e7ea37902a1a6e3b96c9047b0330aa587 Mon Sep 17 00:00:00 2001
From: Dmitri Bogomolov <4glitch@gmail.com>
Date: Mon, 6 Dec 2021 20:09:10 +0200
Subject: [PATCH] Started a test case for address generator

---
 src/tests/test_addressgenerator.py | 54 ++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 src/tests/test_addressgenerator.py

diff --git a/src/tests/test_addressgenerator.py b/src/tests/test_addressgenerator.py
new file mode 100644
index 00000000..0549242f
--- /dev/null
+++ b/src/tests/test_addressgenerator.py
@@ -0,0 +1,54 @@
+import unittest
+from binascii import unhexlify
+
+from pybitmessage import pathmagic
+
+from .samples import (
+    sample_seed, sample_deterministic_addr3, sample_deterministic_addr4,
+    sample_deterministic_ripe)
+
+
+class TestAddressGenerator(unittest.TestCase):
+    """Test case for AddressGenerator (with thread or not)"""
+
+    @classmethod
+    def setUpClass(cls):
+        pathmagic.setup()  # need this because of import state in network ):
+        import queues
+        import state
+
+        state.enableGUI = False
+        from class_addressGenerator import addressGenerator
+
+        # import helper_startup
+        # from bmconfigparser import BMConfigParser
+
+        # helper_startup.loadConfig()
+        # config = BMConfigParser()
+
+        cls.command_queue = queues.addressGeneratorQueue
+        cls.return_queue = queues.apiAddressGeneratorReturnQueue
+        cls.worker_queue = queues.workerQueue
+
+        thread = addressGenerator()
+        thread.daemon = True
+        thread.start()
+
+    def _execute(self, command, *args):
+        self.command_queue.put((command,) + args)
+        try:
+            return self.return_queue.get()[0]
+        except IndexError:
+            self.fail('Failed to execute command %s' % command)
+
+    def test_createChan(self):
+        """Test createChan command"""
+        self.assertEqual(
+            sample_deterministic_addr3,
+            self._execute('createChan', 3, 1, 'test', sample_seed, True))
+        self.assertEqual(
+            self.worker_queue.get(),
+            ('sendOutOrStoreMyV3Pubkey', unhexlify(sample_deterministic_ripe)))
+        self.assertEqual(
+            sample_deterministic_addr4,
+            self._execute('createChan', 4, 1, 'test', sample_seed, True))