def test_updated_unconfigured(self):
args = mock.Mock(group='group', klass='new_class', delete=False)
result = rs_limits._report_group_class(args, None)
self.assertEqual(result, None)
self.assertEqual(sys.stdout.getvalue(),
"Group group:\n"
" New rate-limit class: new_class\n")
def test_configured(self):
args = mock.Mock(group='group', klass=None, delete=False)
result = rs_limits._report_group_class(args, 'old_class')
self.assertEqual(result, None)
self.assertEqual(sys.stdout.getvalue(),
"Group group:\n"
" Configured rate-limit class: old_class\n")
def test_deleted_unconfigured(self):
args = mock.Mock(group='group', klass=None, delete=True)
result = rs_limits._report_group_class(args, None)
self.assertEqual(result, None)
self.assertEqual(sys.stdout.getvalue(),
"Group group:\n"
" Deleted from database\n")
def test_unconfigured(self):
args = mock.Mock(group='group', klass=None, delete=False)
result = rs_limits._report_group_class(args, None)
self.assertEqual(result, None)
self.assertEqual(sys.stdout.getvalue(),
"Group group:\n"
" Not currently configured in database.\n")
def test_deleted(self):
args = mock.Mock(group='group', klass=None, delete=True)
result = rs_limits._report_group_class(args, 'old_class')
self.assertEqual(result, None)
self.assertEqual(sys.stdout.getvalue(),
"Group group:\n"
" Previous rate-limit class: old_class\n"
" Deleted from database\n")
