Use self.assertRaises as context manager

2018-08-28 17:45:58 +02:00
parent 1649591d75
commit a8a7166e78
1 changed files with 2 additions and 2 deletions
--- a/tests/test_api/test_nodes.py
+++ b/tests/test_api/test_nodes.py
@@ -52,8 +52,8 @@ class NodeContentTypeTest(AbstractPillarTest):
                self.assertNotIn('content_type', db_node['properties'])

                # PUT it again, without a file -- should be blocked.
-                self.assertRaises(UnprocessableEntity, self.app.put_internal, 'nodes', node_doc,
-                                  _id=node_id)
+                with self.assertRaises(UnprocessableEntity):
+                    self.app.put_internal('nodes', node_doc, _id=node_id)

                # PUT it with a file.
                node_doc['properties']['file'] = str(file_id)