Quick fix

davidfischer-ch · Feb 5, 2018 · ad91378 · ad91378
1 parent b2c72c1
commit ad91378
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_subprocess.py b/tests/test_subprocess.py
@@ -71,12 +71,12 @@ def test_cmd_missing_binary(self):
         self.equal(cmd('hfuejnvwqkdivengz', fail=False)['returncode'], 2)
 
     def test_retry_first_try(self):
-        log = Mock()
+        log = mock.Mock()
         cmd('ls', log=log, tries=5, delay_min=1, delay_max=1)
         validate_list(log.call_args_list, [r"call\(u*'Execute ls'\)"])
 
     def test_retry_missing_binary_no_retry(self):
-        log = Mock()
+        log = mock.Mock()
         with self.raises(OSError):
             cmd('hfuejnvwqkdivengz', log=log, tries=5)
         validate_list(log.call_args_list, [
@@ -85,7 +85,7 @@ def test_retry_missing_binary_no_retry(self):
         ])
 
     def test_retry_no_success(self):
-        log = Mock()
+        log = mock.Mock()
         cmd('ls hfuejnvwqkdivengz', log=log, fail=False, tries=5, delay_min=0.0, delay_max=0.95)
         validate_list(log.call_args_list, [
             r"call\(u*'Execute ls hfuejnvwqkdivengz'\)",
@@ -113,7 +113,7 @@ def test_screen(self):
             self.true(screens[0].endswith('my_2nd_screen'))
         finally:
             # Cleanup
-            log = Mock()
+            log = mock.Mock()
             screen_kill(name='my_1st_screen', log=log)
             screen_kill(name='my_2nd_screen', log=log)
             if log.call_args_list: