Allow call_function_test to succeed depite testlib import.

7 years ago · bc801bc2ed
parent 118fe01be5
commit bc801bc2ed
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')