1 files changed, 4 insertions, 0 deletions
diff --git a/test/testutil.py b/test/testutil.py
index fba3869..7661cbc 100644
--- a/test/testutil.py
+++ b/test/testutil.py
@@ -89,6 +89,10 @@ class KafkaIntegrationTestCase(unittest.TestCase):
 
         return self._messages[s].encode('utf-8')
 
+    def key(self, k):
+        return k.encode('utf-8')
+
+
 class Timer(object):
     def __enter__(self):
         self.start = time.time()