fixed test setup

StorjOld · Feb 25, 2016 · 2a5bd2d · 2a5bd2d
1 parent d8f8691
commit 2a5bd2d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/storjnet/api.py b/storjnet/api.py
@@ -68,7 +68,7 @@ def dht_put(self, key, value):
 
         @crochet.wait_for(timeout=self._call_timeout)
         def func():
-            return self.put_async(key, value)
+            return self.dht_put_async(key, value)
         return func()
 
     def dht_get_async(self, key):

diff --git a/swarm.py b/swarm.py
@@ -25,7 +25,7 @@
 # log.startLogging(sys.stdout)
 
 
-SWARMSIZE = 50
+SWARMSIZE = 5
 START_NET_PORT = 6000
 START_USER_PORT = 5000
 
@@ -54,9 +54,12 @@
         while True:
             time.sleep(1)
 
-    finally:
+    # expected exit mode
+    except KeyboardInterrupt:
+        pass
 
-        # shutdown swarm
+    # shutdown swarm
+    finally:
         for api, thread in nodes:
             api.stopserver()
             thread.join()