def test_handle_remove_notifications_no_args(self, sys_stderr, sys_exit):
sys_exit.side_effect = SystemExit()
with self.assertRaises(SystemExit):
handle_remove_notification(self.options, self.session, [])
self.session.deleteNotification.assert_not_called()
def test_handle_remove_notification_bogus(self, stderr):
expected = """Usage: %s remove-notification [options] <notification_id> [<notification_id> ...]
(Specify the --help global option for a list of other help options)
%s: error: All notification ids has to be integers
""" % (self.progname, self.progname)
with self.assertRaises(SystemExit) as ex:
handle_remove_notification(self.options, self.session, ['bogus'])
self.assertExitCode(ex, 2)
self.assert_console_message(stderr, expected)
self.session.deleteNotification.assert_not_called()
def test_handle_remove_notification_not_quiet(self, stdout,
activate_session_mock):
self.options.quiet = False
expected = "Notification 1 successfully removed.\n" \
"Notification 3 successfully removed.\n" \
"Notification 5 successfully removed.\n"
handle_remove_notification(self.options, self.session, ['1', '3', '5'])
self.session.deleteNotification.assert_has_calls(
[mock.call(1), mock.call(3),
mock.call(5)])
activate_session_mock.assert_called_once_with(self.session,
self.options)
actual = stdout.getvalue()
print(actual)
self.assertMultiLineEqual(actual, expected)
def test_handle_remove_notification(self, activate_session_mock):
handle_remove_notification(self.options, self.session, ['1', '3', '5'])
self.session.deleteNotification.assert_has_calls([mock.call(1), mock.call(3), mock.call(5)])