Added conditional import to ensure tests will work under Python 2.5

ade@google.com · ade@google.com · commit d69e5e4b5af7 · 2010-08-31T15:28:20.000+01:00
diff --git a/tests/test_json_model.py b/tests/test_json_model.py
@@ -24,7 +24,13 @@
 from apiclient.discovery import JsonModel
 import os
 import unittest
-import urlparse
+
+# Python 2.5 requires different modules
+try:
+  from urlparse import parse_qs
+except ImportError:
+  from cgi import parse_qs
+
 
 
 class Model(unittest.TestCase):
@@ -71,7 +77,7 @@ def test_json_build_query(self):
     self.assertEqual(headers['accept'], 'application/json')
     self.assertEqual(headers['content-type'], 'application/json')
 
-    query_dict = urlparse.parse_qs(query)
+    query_dict = parse_qs(query)
     self.assertEqual(query_dict['foo'], ['1'])
     self.assertEqual(query_dict['bar'], [u'\N{COMET}'.encode('utf-8')])
     self.assertEqual(body, '{"data": {}}')
