Setting mock return values

Siddharta Govindaraj · Siddharta Govindaraj · commit ad3596d5f57c · 2015-03-23T15:10:05.000+05:30
diff --git a/stock_alerter/tests/test_alert.py b/stock_alerter/tests/test_alert.py
@@ -10,7 +10,9 @@
 class AlertTest(unittest.TestCase):
     def test_action_is_executed_when_rule_matches(self):
         exchange = {"GOOG": Stock("GOOG")}
-        rule = PriceRule("GOOG", lambda stock: stock.price > 10)
+        rule = mock.MagicMock(spec=PriceRule)
+        rule.matches.return_value = True
+        rule.depends_on.return_value = {"GOOG"}
         action = mock.MagicMock()
         alert = Alert("sample alert", rule, action)
         alert.connect(exchange)
