import sys import unittest from aprsd.plugins import email if sys.version_info >= (3, 2): from unittest import mock else: from unittest import mock class TestMain(unittest.TestCase): @mock.patch("aprsd.plugins.email._imap_connect") @mock.patch("aprsd.plugins.email._smtp_connect") def test_validate_email(self, imap_mock, smtp_mock): """Test to make sure we fail.""" imap_mock.return_value = None smtp_mock.return_value = {"smaiof": "fire"} config = mock.MagicMock() email.validate_email_config(config, True)