Fix python 2.6 failures

googleapis · Mar 10, 2015 · 1aa4b50 · 1aa4b50
1 parent e63b36a
commit 1aa4b50
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/test_devshell.py b/tests/test_devshell.py
@@ -81,8 +81,7 @@ def run(self):
 class DevshellCredentialsTests(unittest.TestCase):
 
   def test_signals_no_server(self):
-    with self.assertRaises(NoDevshellServer):
-      DevshellCredentials()
+    self.assertRaises(NoDevshellServer, DevshellCredentials)
 
   def test_request_response(self):
     with _AuthReferenceServer():
@@ -94,7 +93,7 @@ def test_request_response(self):
   def test_no_refresh_token(self):
     with _AuthReferenceServer():
       creds = DevshellCredentials()
-      self.assertIsNone(creds.refresh_token)
+      self.assertEquals(None, creds.refresh_token)
 
   def test_reads_credentials(self):
     with _AuthReferenceServer():
@@ -121,6 +120,4 @@ def test_handles_ignores_extra_fields(self):
   def test_refuses_to_save_to_well_known_file(self):
     with _AuthReferenceServer():
       creds = DevshellCredentials()
-      with self.assertRaises(NotImplementedError):
-        save_to_well_known_file(creds)
-
+      self.assertRaises(NotImplementedError, save_to_well_known_file, creds)