fixed tests and .env issue

project-mk-ultra · Jun 16, 2018 · 6d3fa36 · 6d3fa36
1 parent 303ad02
commit 6d3fa36
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 14 deletions.
diff --git a/dht/dht.py b/dht/dht.py
@@ -14,10 +14,10 @@
 
 load_dotenv(find_dotenv())
 
-K_BUCKET_SIZE = int(os.getenv("K_BUCKET_SIZE"))
-ALPHA = int(os.getenv("ALPHA"))
-ID_BITS = int(os.getenv("ID_BITS"))
-ITERATION_SLEEP = int(os.getenv("ITERATION_SLEEP"))
+K_BUCKET_SIZE = 20 if os.getenv("K_BUCKET_SIZE") is None else int(os.getenv("K_BUCKET_SIZE"))
+ALPHA = 3 if os.getenv("ALPHA") is None else int(os.getenv("ALPHA"))
+ID_BITS = 128 if os.getenv("ID_BITS") is None else int(os.getenv("ID_BITS"))
+ITERATION_SLEEP = 1 if os.getenv("ITERATION_SLEEP") is None else int(os.getenv("ITERATION_SLEEP"))
 
 
 class DHT:

diff --git a/tests/dht_test.py b/tests/dht_test.py
@@ -6,9 +6,7 @@
 
 class DHTTest(unittest.TestCase):
     def setUp(self):
-        from dotenv import load_dotenv, find_dotenv
-        env_path = Path('..') / '.env.dist'
-        load_dotenv(dotenv_path=env_path)
+        pass
 
     def test_dht(self):
         """

diff --git a/tests/peer_test.py b/tests/peer_test.py
@@ -6,11 +6,8 @@
 
 class PeerTest(unittest.TestCase):
     def setUp(self):
-        from dotenv import load_dotenv
-        env_path = Path('..') / '.env.dist'
-        load_dotenv(dotenv_path=env_path)
+        pass
 
-
-def test_peer(self):
-    peer = Peer("localhost", 9789, "foo", "bar")
-    self.assertEqual(str(peer), "localhost:9789")
+    def test_peer(self):
+        peer = Peer("localhost", 9789, "foo", "bar")
+        self.assertEqual(str(peer), "localhost:9789")