def test_announcement_notifier(self, mock_load_config,
mock_twitter, mock_open):
"""Assert that AnnouncementNotifier works"""
mock_open.return_value = "config.yaml"
def load_config(filename):
if "config.yaml" in filename:
return self.base_auth
elif "statuses.yaml" in filename:
return self.statuses
mock_load_config.side_effect = load_config
self.announcement_notifier = AnnouncementNotifier()
mock_open.return_value = "statuses.yaml"
self.announcement_notifier.generate_message()
self.assertEqual(self.announcement_notifier.message,
'I am the MOTD')
