Using mock.patch as a context manager

Siddharta Govindaraj · Siddharta Govindaraj · commit 9812eb9f8fb7 · 2015-03-23T15:15:14.000+05:30
diff --git a/stock_alerter/tests/test_action.py b/stock_alerter/tests/test_action.py
@@ -6,11 +6,7 @@
 
 class PrintActionTest(unittest.TestCase):
     def test_executing_action_prints_message(self):
-        patcher = mock.patch('builtins.print')
-        mock_print = patcher.start()
-        try:
+        with mock.patch('builtins.print') as mock_print:
             action = PrintAction()
             action.execute("GOOG > $10")
             mock_print.assert_called_with("GOOG > $10")
-        finally:
-            patcher.stop()
