diff options
Diffstat (limited to 'kazoo/tests/test_lock.py')
| -rw-r--r-- | kazoo/tests/test_lock.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/kazoo/tests/test_lock.py b/kazoo/tests/test_lock.py index 2301e9d..397e971 100644 --- a/kazoo/tests/test_lock.py +++ b/kazoo/tests/test_lock.py @@ -1,7 +1,7 @@ import collections -import mock import threading import unittest +from unittest.mock import MagicMock import uuid import pytest @@ -803,7 +803,7 @@ class TestSequence(unittest.TestCase): goLock = "_c_8eb60557ba51e0da67eefc47467d3f34-lock-0000000031" pyLock = "514e5a831836450cb1a56c741e990fd8__lock__0000000032" children = ["hello", goLock, "world", pyLock] - client = mock.MagicMock() + client = MagicMock() client.get_children.return_value = children lock = Lock(client, "test") assert lock._get_predecessor(pyLock) is None @@ -815,7 +815,7 @@ class TestSequence(unittest.TestCase): goLock = "_c_8eb60557ba51e0da67eefc47467d3f34-lock-0000000031" pyLock = "514e5a831836450cb1a56c741e990fd8__lock__0000000032" children = ["hello", goLock, "world", pyLock] - client = mock.MagicMock() + client = MagicMock() client.get_children.return_value = children lock = Lock(client, "test", extra_lock_patterns=["-lock-"]) assert lock._get_predecessor(pyLock) == goLock |