diff --git a/tests/testlib.py b/tests/testlib.py index 729461f8..4e8f7fc6 100644 --- 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')