Fourth characterisation test

Siddharta Govindaraj · Siddharta Govindaraj · commit 08fdb5fe14ff · 2015-03-23T18:24:29.000+05:30
diff --git a/stock_alerter/legacy.py b/stock_alerter/legacy.py
@@ -5,8 +5,11 @@
 
 
 class AlertProcessor:
-    def __init__(self, autorun=True):
-        self.exchange = {"GOOG": Stock("GOOG"), "AAPL": Stock("AAPL")}
+    def __init__(self, autorun=True, exchange=None):
+        if exchange is None:
+            self.exchange = {"GOOG": Stock("GOOG"), "AAPL": Stock("AAPL")}
+        else:
+            self.exchange = exchange
         rule_1 = PriceRule("GOOG", lambda stock: stock.price > 10)
         rule_2 = PriceRule("AAPL", lambda stock: stock.price > 5)
         self.exchange["GOOG"].updated.connect(
diff --git a/stock_alerter/tests/test_legacy.py b/stock_alerter/tests/test_legacy.py
@@ -30,3 +30,12 @@ def test_processor_characterization_3(self):
         updates = [("GOOG", datetime(2014, 12, 8), 5)]
         processor.do_updates(updates)
         mock_goog.update.assert_called_with(datetime(2014, 12, 8), 5)
+
+    def test_processor_characterization_4(self):
+        mock_goog = mock.Mock()
+        mock_aapl = mock.Mock()
+        exchange = {"GOOG": mock_goog, "AAPL": mock_aapl}
+        processor = AlertProcessor(autorun=False, exchange=exchange)
+        updates = [("GOOG", datetime(2014, 12, 8), 5)]
+        processor.do_updates(updates)
+        mock_goog.update.assert_called_with(datetime(2014, 12, 8), 5)
