Resolve time import error in exception_handler

locust/main.py

@@ -19,7 +19,7 @@
 from .runners import LocalLocustRunner, MasterLocustRunner, SlaveLocustRunner
 from .stats import (print_error_report, print_percentile_stats, print_stats,
                     stats_printer, stats_writer, write_stat_csvs)
-from .util.time import parse_timespan
+from .util.timespan import parse_timespan
 
 _internals = [Locust, HttpLocust]
 version = locust.__version__

locust/test/test_util.py

@@ -1,5 +1,5 @@
 import unittest
-from locust.util.time import parse_timespan
+from locust.util.timespan import parse_timespan
 from locust.util.rounding import proper_round
 
 

locust/test/test_zmqrpc.py

@@ -30,3 +30,9 @@ def test_client_recv(self):
         self.assertEqual(msg.type, 'test')
         self.assertEqual(msg.data, 'message')
         self.assertEqual(msg.node_id, 'identity')
+
+    def test_client_retry(self):
+        server = zmqrpc.Server('0.0.0.0', 8888)
+        server.socket.close()
+        with self.assertRaises(zmq.error.ZMQError):
+            server.recv_from_client()