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)