From 1b21e25f7e7f78a73e1519c06e230c722187e53c Mon Sep 17 00:00:00 2001
From: Kevin F <kevin@voidnet>
Date: Sun, 15 May 2022 23:21:45 -0500
Subject: [PATCH] Finished basic get_new_peers tests

---
 tests/gossip-unittests/test_client_get_new_peers.py | 6 ++++--
 tests/release-tests/test_big_brother.py             | 5 +++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/tests/gossip-unittests/test_client_get_new_peers.py b/tests/gossip-unittests/test_client_get_new_peers.py
index cfde8935..7220997b 100644
--- a/tests/gossip-unittests/test_client_get_new_peers.py
+++ b/tests/gossip-unittests/test_client_get_new_peers.py
@@ -51,7 +51,7 @@ fake_peer_addresses = [MockPeer().transport_address for i in range(10)]
 
 
 def _server():
-   
+
     with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
         s.bind(server_file)
         s.listen(1)
@@ -70,12 +70,14 @@ class OnionrGossipClientGetNewPeers(unittest.TestCase):
     def test_get_new_peers_no_peers(self):
         gossip_peer_set.clear()
         self.assertRaises(ValueError, get_new_peers)
+        self.assertFalse(len(gossip_peer_set))
 
     def test_get_new_peers(self):
         p = MockPeer()
         gossip_peer_set.add(p)
         get_new_peers()
+        assert len(gossip_peer_set)
         self.assertTrue(len(gossip_peer_set), len(fake_peer_addresses) + 1)
 
-
+sleep(0.5)
 unittest.main()
diff --git a/tests/release-tests/test_big_brother.py b/tests/release-tests/test_big_brother.py
index 7e9befda..4f7a2928 100644
--- a/tests/release-tests/test_big_brother.py
+++ b/tests/release-tests/test_big_brother.py
@@ -17,6 +17,11 @@ createdirs.create_dirs()
 setup.setup_config()
 
 class TestBigBrother(unittest.TestCase):
+    def test_system(self):
+        import bigbrother
+        bigbrother.enable_ministries()
+        self.assertRaises(onionrexceptions.ArbitraryCodeExec, os.system, "echo test")
+
     def test_requests_connect(self):
         import bigbrother
         bigbrother.enable_ministries()