Allow call_function_test to succeed depite testlib import.

7 years ago · 2ed39c411f
parent 2659cf0ae0
commit 2ed39c411f
1 changed files with 2 additions and 3 deletions
--- a/tests/testlib.py
+++ b/tests/testlib.py
@ -5,10 +5,9 @@ import sys
 import unittest
 import urlparse

-import mock
-
 import mitogen.master
-import docker
+if mitogen.is_master:  # TODO: shouldn't be necessary.
+    import docker


 DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')