def test_success_messages(self, mock_warning, mock_success):
# foo-archiver enabled and stored adds success message.
_add_archival_messages(['foo-archiver'], [], {'foo-archiver': True}, self.request)
self.assertTrue('activated new archivers' in mock_success.call_args[0][1])
self.assertTrue('foo-archiver' in mock_success.call_args[0][1])
# messages.warning should not have been called.
self.assertEqual(mock_warning.call_count, 0)
def test_warning_messages(self, mock_warning, mock_success):
# foo-archiver enabled, but not stored adds warning message.
_add_archival_messages(['foo-archiver'], [], {'foo-archiver': False}, self.request)
self.assertTrue('could not be enabled' in mock_warning.call_args[0][1])
self.assertTrue('foo-archiver' in mock_warning.call_args[0][1])
# messages.success should not have been called.
self.assertEqual(mock_success.call_count, 0)
