Fix test to use key_prefix option

2014-04-16 18:11:29 +09:00
parent 3762ed5ddc
commit 4eb1c89cde
1 changed files with 2 additions and 6 deletions
--- a/pymemcache/test/test_client.py
+++ b/pymemcache/test/test_client.py
@@ -634,13 +634,9 @@ class TestMockClient(ClientTestMixin, unittest.TestCase):
    Client = MockMemcacheClient


-class PrefixedClient(Client):
-    def check_key(self, key):
-        return b'xyz:' + Client.check_key(self, key)
-
-
 class TestPrefixedClient(ClientTestMixin, unittest.TestCase):
-    Client = PrefixedClient
+    def Client(self, *args, **kwargs):
+        return Client(*args, key_prefix=b'xyz:', **kwargs)

    def test_get_found(self):
        client = self.Client(None)