Add more unit tests

holgern · Apr 8, 2020 · 5d22500 · 5d22500
1 parent 0fda10d
commit 5d22500
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tests/beem/test_cli.py b/tests/beem/test_cli.py
@@ -388,6 +388,14 @@ def test_updatenodes(self):
         runner.invoke(cli, ['-o', 'set', 'nodes', self.node_list])
         result = runner.invoke(cli, ['updatenodes', '--hive', '--test'])
         self.assertEqual(result.exit_code, 0)
+        result = runner.invoke(cli, ['updatenodes', '--steem'])
+        self.assertEqual(result.exit_code, 0)
+        result = runner.invoke(cli, ['updatenodes'])
+        self.assertEqual(result.exit_code, 0)
+        result = runner.invoke(cli, ['updatenodes', '--hive'])
+        self.assertEqual(result.exit_code, 0)        
+        result = runner.invoke(cli, ['updatenodes'])
+        self.assertEqual(result.exit_code, 0)        
         runner.invoke(cli, ['-o', 'set', 'nodes', str(self.node_list)])
 
     def test_currentnode(self):

diff --git a/tests/beem/test_steem.py b/tests/beem/test_steem.py
@@ -455,3 +455,10 @@ def test_broadcast(self):
             exceptions.MissingKeyError
         ):
             bts.broadcast(tx=tx)
+
+    def test_switch_blockchain(self):
+        bts = self.bts
+        bts.switch_blockchain("steem", update_nodes=True)
+        assert not bts.is_hive
+        bts.switch_blockchain("hive", update_nodes=True)
+        assert bts.is_hive