def test_failed_update_settings(self):
notifiers_manager = NotifierManager({'test': NotifierCollectionTest.TestNotifier()})
notifiers_manager.update_settings = MagicMock(return_value=False)
notifier = Notifier(notifiers_manager)
self.api.add_route('/api/notifiers/{notifier}', notifier)
self.simulate_request('/api/notifiers/{0}'.format(1), method="PUT",
body=json.dumps({'login': '******', 'password': '******'}))
self.assertEqual(self.srmock.status, falcon.HTTP_BAD_REQUEST)
def test_not_found_update_settings(self):
notifiers_manager = NotifierManager({'test': NotifierCollectionTest.TestNotifier()})
notifiers_manager.update_settings = MagicMock(side_effect=KeyError)
notifier = Notifier(notifiers_manager)
self.api.add_route('/api/notifiers/{notifier}', notifier)
self.simulate_request('/api/notifiers/{0}'.format(1), method="PUT",
body=json.dumps({'login': '******', 'password': '******'}))
self.assertEqual(self.srmock.status, falcon.HTTP_NOT_FOUND)
def test_failed_update_settings(self):
# noinspection PyTypeChecker
notifiers_manager = NotifierManager(Mock(), {'test': NotifierCollectionTest.TestNotifier()})
notifiers_manager.update_settings = Mock(return_value=False)
notifier = Notifier(notifiers_manager)
self.api.add_route('/api/notifiers/{notifier}', notifier)
self.simulate_request('/api/notifiers/{0}'.format(1), method="PUT",
body=json.dumps({'login': '******', 'password': '******'}))
self.assertEqual(self.srmock.status, falcon.HTTP_BAD_REQUEST)
def test_not_found_update_settings(self):
# noinspection PyTypeChecker
notifiers_manager = NotifierManager(Mock(), {'test': NotifierCollectionTest.TestNotifier()})
notifiers_manager.update_settings = Mock(side_effect=KeyError)
notifier = Notifier(notifiers_manager)
self.api.add_route('/api/notifiers/{notifier}', notifier)
self.simulate_request('/api/notifiers/{0}'.format(1), method="PUT",
body=json.dumps({'login': '******', 'password': '******'}))
self.assertEqual(self.srmock.status, falcon.HTTP_NOT_FOUND)
def test_successful_update_settings(self):
notifiers_manager = NotifierManager(
{'test': NotifierCollectionTest.TestNotifier()})
notifiers_manager.update_settings = MagicMock(return_value=True)
notifier = Notifier(notifiers_manager)
self.api.add_route('/api/notifiers/{notifier}', notifier)
self.simulate_request('/api/notifiers/{0}'.format('test'),
method="PUT",
body=json.dumps({
'login': '******',
'password': '******'
}))
self.assertEqual(self.srmock.status, falcon.HTTP_NO_CONTENT)
class NotifierManagerTest(TestCase):
NOTIFIER1_NAME = 'notifier1'
NOTIFIER2_NAME = 'notifier2'
def setUp(self):
super(NotifierManagerTest, self).setUp()
self.notifier1 = Mock()
self.notifier2 = Mock()
self.notifier_manager = NotifierManager({
self.NOTIFIER1_NAME:
self.notifier1,
self.NOTIFIER2_NAME:
self.notifier2
})
def test_init(self):
notifier_manager = NotifierManager()
self.assertIsNotNone(notifier_manager.notifiers)
def test_get_settings(self):
settings1 = {'uri': 'uri1', 'login': '******', 'password': '******'}
settings2 = {'uri': 'uri2', 'login': '******', 'password': '******'}
self.notifier1.get_settings = MagicMock(return_value=settings1)
self.notifier2.get_settings = MagicMock(return_value=settings2)
settings = self.notifier_manager.get_settings(self.NOTIFIER1_NAME)
self.assertEqual(settings1, settings)
self.notifier1.get_settings.assert_called_with()
self.notifier2.get_settings.assert_not_called()
settings = self.notifier_manager.get_settings(self.NOTIFIER2_NAME)
self.assertEqual(settings2, settings)
self.notifier2.get_settings.assert_called_with()
def test_get_notifier(self):
notifier = self.notifier_manager.get_notifier(
self.NOTIFIER1_NAME).get('notifier')
self.assertEqual(self.notifier1, notifier)
notifier = self.notifier_manager.get_notifier(
self.NOTIFIER2_NAME).get('notifier')
self.assertEqual(self.notifier2, notifier)
def test_update_settings(self):
settings1 = {'uri': 'uri1', 'login': '******', 'password': '******'}
settings2 = {'uri': 'uri2', 'login': '******', 'password': '******'}
self.notifier1.update_settings = MagicMock()
self.notifier2.update_settings = MagicMock()
self.notifier_manager.update_settings(self.NOTIFIER1_NAME, settings1)
self.notifier1.update_settings.assert_called_with(settings1)
self.notifier2.update_settings.assert_not_called()
self.notifier_manager.update_settings(self.NOTIFIER2_NAME, settings2)
self.notifier2.update_settings.assert_called_with(settings2)
def test_send_test_message(self):
notify1 = MagicMock()
notify2 = MagicMock()
self.notifier1.notify = notify1
self.notifier2.notify = notify2
self.notifier_manager.send_test_message(self.NOTIFIER1_NAME)
notify1.assert_called_once_with(
"Test Message", "This is monitorrent test message",
"https://github.com/werwolfby/monitorrent")
notify2.assert_not_called()
self.notifier_manager.send_test_message(self.NOTIFIER2_NAME)
notify2.assert_called_once_with(
"Test Message", "This is monitorrent test message",
"https://github.com/werwolfby/monitorrent")
def test_get_enabled_new_settings(self):
# new should be false
self.notifier1.get_settings = MagicMock(return_value=None)
self.assertFalse(self.notifier_manager.get_enabled(
self.NOTIFIER1_NAME))
def test_set_enabled(self):
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, True)
self.assertTrue(self.notifier_manager.get_enabled(self.NOTIFIER1_NAME))
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, False)
self.assertFalse(self.notifier_manager.get_enabled(
self.NOTIFIER1_NAME))
def test_set_enabled_new(self):
self.notifier1.get_settings = MagicMock(return_value=None)
self.notifier1.settings_class = MagicMock
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, True)
class NotifierManagerTest(TestCase):
NOTIFIER1_NAME = 'notifier1'
NOTIFIER2_NAME = 'notifier2'
def setUp(self):
super(NotifierManagerTest, self).setUp()
self.notifier1 = Mock()
self.notifier2 = Mock()
self.notifier_manager = NotifierManager(
{self.NOTIFIER1_NAME: self.notifier1, self.NOTIFIER2_NAME: self.notifier2})
def test_init(self):
notifier_manager = NotifierManager()
self.assertIsNotNone(notifier_manager.notifiers)
def test_get_settings(self):
settings1 = {'uri': 'uri1', 'login': '******', 'password': '******'}
settings2 = {'uri': 'uri2', 'login': '******', 'password': '******'}
self.notifier1.get_settings = MagicMock(return_value=settings1)
self.notifier2.get_settings = MagicMock(return_value=settings2)
settings = self.notifier_manager.get_settings(self.NOTIFIER1_NAME)
self.assertEqual(settings1, settings)
self.notifier1.get_settings.assert_called_with()
self.notifier2.get_settings.assert_not_called()
settings = self.notifier_manager.get_settings(self.NOTIFIER2_NAME)
self.assertEqual(settings2, settings)
self.notifier2.get_settings.assert_called_with()
def test_get_notifier(self):
notifier = self.notifier_manager.get_notifier(self.NOTIFIER1_NAME).get('notifier')
self.assertEqual(self.notifier1, notifier)
notifier = self.notifier_manager.get_notifier(self.NOTIFIER2_NAME).get('notifier')
self.assertEqual(self.notifier2, notifier)
def test_update_settings(self):
settings1 = {'uri': 'uri1', 'login': '******', 'password': '******'}
settings2 = {'uri': 'uri2', 'login': '******', 'password': '******'}
self.notifier1.update_settings = MagicMock()
self.notifier2.update_settings = MagicMock()
self.notifier_manager.update_settings(self.NOTIFIER1_NAME, settings1)
self.notifier1.update_settings.assert_called_with(settings1)
self.notifier2.update_settings.assert_not_called()
self.notifier_manager.update_settings(self.NOTIFIER2_NAME, settings2)
self.notifier2.update_settings.assert_called_with(settings2)
def test_send_test_message(self):
notify1 = MagicMock()
notify2 = MagicMock()
self.notifier1.notify = notify1
self.notifier2.notify = notify2
self.notifier_manager.send_test_message(self.NOTIFIER1_NAME)
notify1.assert_called_once_with("Test Message", "This is monitorrent test message",
"https://github.com/werwolfby/monitorrent")
notify2.assert_not_called()
self.notifier_manager.send_test_message(self.NOTIFIER2_NAME)
notify2.assert_called_once_with("Test Message", "This is monitorrent test message",
"https://github.com/werwolfby/monitorrent")
def test_get_enabled_new_settings(self):
# new should be false
self.notifier1.get_settings = MagicMock(return_value=None)
self.assertFalse(self.notifier_manager.get_enabled(self.NOTIFIER1_NAME))
def test_set_enabled(self):
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, True)
self.assertTrue(self.notifier_manager.get_enabled(self.NOTIFIER1_NAME))
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, False)
self.assertFalse(self.notifier_manager.get_enabled(self.NOTIFIER1_NAME))
def test_set_enabled_new(self):
self.notifier1.get_settings = MagicMock(return_value=None)
self.notifier1.settings_class = MagicMock
self.notifier_manager.set_enabled(self.NOTIFIER1_NAME, True)
