Fixing assert contexts for <= 2.6 / unittest2

Author: joshmarshall

tests.py

@@ -107,7 +107,9 @@ def test_notification(self):
         self.assertTrue(response == verify_response)
 
     def test_non_existent_method(self):
-        self.assertRaises(ProtocolError, self.client.foobar)
+        with self.assertRaises(ProtocolError):
+            self.client.foobar()
+
         request = json.loads(history.request)
         response = json.loads(history.response)
         verify_request = {
@@ -304,7 +306,8 @@ def test_single_kwargs(self):
 
     def test_single_kwargs_and_args(self):
         client = self.get_client()
-        self.assertRaises(ProtocolError, client.add, (5,), {'y': 10})
+        with self.assertRaises(ProtocolError):
+            client.add(5, y=10)
 
     def test_single_notify(self):
         client = self.get_client()
@@ -351,7 +354,8 @@ def test_multicall_failure(self):
             else:
                 def func():
                     return result[i]
-                self.assertRaises(raises[i], func)
+                with self.assertRaises(raises[i]):
+                    func()
 
 
 if jsonrpc.USE_UNIX_SOCKETS:
@@ -401,11 +405,8 @@ def setUp(self):
 
     def test_client(self):
         address = "unix://shouldnt/work.sock"
-        self.assertRaises(
-            jsonrpc.UnixSocketMissing,
-            Server,
-            address
-        )
+        with self.assertRaises(jsonrpc.UnixSocketMissing):
+            Server(address)
 
     def tearDown(self):
         jsonrpc.USE_UNIX_SOCKETS = self.original_value