diff --git a/tests/test_consul.py b/tests/test_consul.py index 8e55a3eb..8b35457c 100644 --- a/tests/test_consul.py +++ b/tests/test_consul.py @@ -197,7 +197,7 @@ class TestConsul(unittest.TestCase): with patch.object(KV, 'put', Mock(side_effect=ConsulException)): self.assertFalse(self.c.update_leader(cluster, 12345)) mock_time = Mock(side_effect=[0, 0, 0, 0, 100, 200, 300]) - with patch('time.time', mock_time), patch('time.time_ns', mock_time): + with patch('time.time', mock_time), patch('time.time_ns', mock_time, create=True): self.assertRaises(ConsulError, self.c.update_leader, cluster, 12345) with patch('time.time', Mock(side_effect=[0, 100, 200, 300])): self.assertRaises(ConsulError, self.c.update_leader, cluster, 12345) diff --git a/tests/test_kubernetes.py b/tests/test_kubernetes.py index becef826..b2cb95a4 100644 --- a/tests/test_kubernetes.py +++ b/tests/test_kubernetes.py @@ -442,7 +442,7 @@ class TestKubernetesEndpoints(BaseTestKubernetes): k8s_client.rest.ApiException(409, ''), mock_namespaced_kind()] mock_read.return_value.metadata.resource_version = '2' mock_time = Mock(side_effect=[0, 0, 100, 200, 0, 0, 0, 0, 0, 100, 200]) - with patch('time.time', mock_time), patch('time.time_ns', mock_time): + with patch('time.time', mock_time), patch('time.time_ns', mock_time, create=True): self.assertFalse(self.k.update_leader(cluster, '123')) self.assertFalse(self.k.update_leader(cluster, '123')) mock_patch.side_effect = k8s_client.rest.ApiException(409, '')