Improve management test branch coverage

Signed-off-by: Aleksi Häkli <aleksi.hakli@iki.fi>
This commit is contained in:
Aleksi Häkli 2019-02-10 20:03:57 +02:00
parent b8f417d485
commit e3e69006a4
No known key found for this signature in database
GPG key ID: 3E7146964D726BBE

View file

@ -34,6 +34,16 @@ class ManagementCommandTestCase(TestCase):
expected = '2 attempts removed.\n'
self.assertEqual(expected, out.getvalue())
def test_axes_reset_not_found(self):
out = StringIO()
call_command('axes_reset', stdout=out)
out = StringIO()
call_command('axes_reset', stdout=out)
expected = 'No attempts found.\n'
self.assertEqual(expected, out.getvalue())
def test_axes_reset_ip(self):
out = StringIO()
call_command('axes_reset_ip', '10.0.0.1', stdout=out)