Fix test to not user assertRaises context manager

praekeltfoundation · Apr 7, 2016 · 9771515 · 9771515
1 parent ecf0da1
commit 9771515
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/vumi/persist/tests/test_redis_base.py b/vumi/persist/tests/test_redis_base.py
@@ -37,8 +37,9 @@ def test_no_key_prefix_sub_manager(self):
     def test_client_and_client_proxy_disallowed(self):
         '''If both the client and the client proxy are specified when creating
         a manager, then an exception should be raised.'''
-        with self.assertRaises(AssertionError) as e:
-            Manager(object(), None, None, client_proxy=object())
+        e = self.assertRaises(
+            AssertionError, Manager, object(), None, None,
+            client_proxy=object())
         self.assertEqual(
-            str(e.exception),
+            str(e),
             'Only one of client or client_proxy may be specified')