Fixed a unicode encoding case

googleapis · Jan 7, 2015 · 58a7962 · 58a7962
1 parent 391df3c
commit 58a7962
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_json_model.py b/tests/test_json_model.py
@@ -123,7 +123,12 @@ def test_json_build_query(self):
 
     query_dict = parse_qs(query[1:])
     self.assertEqual(query_dict['foo'], ['1'])
-    self.assertEqual(query_dict['bar'], [u'\N{COMET}'.encode('utf-8')])
+    if isinstance(u'\N{COMET}', str):
+      # Python 3, no need to encode
+      self.assertEqual(query_dict['bar'], [u'\N{COMET}'])
+    else:
+      # Python 2, encode string
+      self.assertEqual(query_dict['bar'], [u'\N{COMET}'.encode('utf-8')])
     self.assertEqual(query_dict['baz'], ['fe', 'fi', 'fo', 'fum'])
     self.assertTrue('qux' not in query_dict)
     self.assertEqual(body, '{}')