Send tests for enabled & disabled.

Author: Tom Galloway

market/tests/test_smtpnotification.py

@@ -1,6 +1,7 @@
 from twisted.trial import unittest
 from twisted.python import log
-from mock import MagicMock
+from mock import patch, MagicMock
+import mock
 
 from market.smtpnotification import SMTPNotification
 
@@ -12,17 +13,42 @@ def setUp(self):
         log.addObserver(observer)
         self.addCleanup(log.removeObserver, observer)
         self.db = MagicMock()
-
-    def test_MarketSmtp_settings_success(self):
-        '''SMTP Notification settings correctly set.'''
         self.db.settings.get.return_value = ['0', '1', '2', '3', '4', '5', '6',
                                              '7', '8', '9', '10', '11', '12',
-                                             '13', '14', 'test_server',
+                                             '13', 1, 'test_server',
                                              'test_sender', 'test_recipient',
                                              'test_username', 'test_password']
+
+    def test_MarketSmtp_settings_success(self):
+        '''SMTP Notification settings correctly set.'''
         s = SMTPNotification(self.db)
         self.assertEqual('test_server', s.server)
         self.assertEqual('test_sender', s.sender)
         self.assertEqual('test_recipient', s.recipient)
         self.assertEqual('test_username', s.username)
         self.assertEqual('test_password', s.password)
+
+    @patch("smtplib.SMTP")
+    def test_MarketSmtp_send_enabled_success(self, mock_smtp):
+        '''Email sent when enabled'''
+        instance = mock_smtp.return_value
+        s = SMTPNotification(self.db)
+        s.send('test_subject', 'test_body')
+        mock_smtp.assert_called_once_with('test_server')
+        instance.login.assert_called_once_with('test_username', 'test_password')
+        instance.sendmail.assert_called_once_with('test_sender', 'test_recipient', mock.ANY)
+
+    @patch("smtplib.SMTP")
+    def test_MarketSmtp_send_disabled_not_sent(self, mock_smtp):
+        '''Email not sent when disabled'''
+        instance = mock_smtp.return_value
+        self.db.settings.get.return_value = ['0', '1', '2', '3', '4', '5', '6',
+                                             '7', '8', '9', '10', '11', '12',
+                                             '13', 0, 'test_server',
+                                             'test_sender', 'test_recipient',
+                                             'test_username', 'test_password']
+        s = SMTPNotification(self.db)
+        s.send('test_subject', 'test_body')
+        assert mock_smtp.call_count == 0
+        assert instance.login.call_count == 0
+        assert instance.sendmail.call_count == 0