Added test for disable command

tests/tests.py

@@ -1,4 +1,5 @@
 from binascii import unhexlify
+from django.core.management import call_command, CommandError
 import qrcode.image.svg
 
 try:
@@ -759,3 +760,27 @@ class TestForm(forms.Form):
         self.assertIn('number', form.errors)
         self.assertEqual(form.errors['number'],
                          [six.text_type(phone_number_validator.message)])
+
+
+class DisableCommandTest(UserMixin, TestCase):
+    def test_raises(self):
+        with self.assertRaisesMessage(CommandError, 'User "some_username" does not exist'):
+            call_command('disable', 'some_username')
+
+        with self.assertRaisesMessage(CommandError, 'User "other_username" does not exist'):
+            call_command('disable', 'other_username', 'some_username')
+
+    def test_disable_single(self):
+        user = self.create_user()
+        self.enable_otp(user)
+        call_command('disable', 'bouke@example.com')
+        self.assertEqual(list(devices_for_user(user)), [])
+
+    def test_happy_flow_multiple(self):
+        users = [self.create_user(n) for n in ['user0@example.com', 'user1@example.com']]
+        for u in users:
+            self.enable_otp(u)
+        call_command('disable', 'user0@example.com', 'user1@example.com')
+        self.assertEqual(list(devices_for_user(users[0])), [])
+        self.assertEqual(list(devices_for_user(users[1])), [])
+