EmailAction - testing exception handling

Siddharta Govindaraj · Siddharta Govindaraj · commit 9e0b0a143bdb · 2015-03-23T15:23:56.000+05:30
diff --git a/stock_alerter/tests/test_action.py b/stock_alerter/tests/test_action.py
@@ -30,3 +30,12 @@ def test_connection_closed_after_sending_mail(self, mock_smtp_class):
         mock_smtp.assert_has_calls([
             mock.call.send_message(mock.ANY),
             mock.call.quit()])
+
+    def test_connection_closed_if_send_gives_error(self, mock_smtp_class):
+        mock_smtp = mock_smtp_class.return_value
+        mock_smtp.send_message.side_effect = smtplib.SMTPServerDisconnected()
+        try:
+            self.action.execute("MSFT has crossed $10 price level")
+        except Exception:
+            pass
+        self.assertTrue(mock_smtp.quit.called)
