diff options
Diffstat (limited to 'tests/test_setmulti.py')
| -rw-r--r-- | tests/test_setmulti.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_setmulti.py b/tests/test_setmulti.py index 2f7108a..39d43c9 100644 --- a/tests/test_setmulti.py +++ b/tests/test_setmulti.py @@ -53,18 +53,20 @@ class test_Memcached_Set_Multi(unittest.TestCase): self.old_socket = socket.socket socket.socket = FakeSocket + self.mc = memcache.Client(['memcached'], debug=True) + def tearDown(self): socket.socket = self.old_socket def test_Socket_Disconnect(self): - client = memcache.Client(['memcached'], debug=True) mapping = {'foo': 'FOO', 'bar': 'BAR'} - bad_keys = client.set_multi(mapping) + bad_keys = self.mc.set_multi(mapping) self.assertEqual(sorted(bad_keys), ['bar', 'foo']) if DEBUG: print('set_multi({0!r}) -> {1!r}'.format(mapping, bad_keys)) + if __name__ == '__main__': unittest.main() |