use six.b() in test string

istresearch · May 22, 2017 · 0238192 · 0238192
1 parent 80b734c
commit 0238192
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/utils/tests/online.py b/utils/tests/online.py
@@ -10,6 +10,7 @@
 from mock import MagicMock
 import time
 import sys
+import six
 
 import redis
 import argparse
@@ -128,8 +129,8 @@ def test_purge_old(self):
         self.redis_conn.set('default_counter:2015-10', 'stuff2')
 
         tc.purge_old()
-        self.assertEqual([b'default_counter:2015-10'],
-            self.redis_conn.keys(tc.get_key() + ':*'))
+        self.assertEqual([six.b('default_counter:2015-10')],
+                         self.redis_conn.keys(tc.get_key() + ':*'))
 
         self.redis_conn.delete('default_counter:2015-10')