Merge lp:~roadmr/canonical-identity-provider/suspend-user-mgmt-command into lp:canonical-identity-provider/release

[Daniel Manrique]

Implement a suspend_account management command.

Implement a suspend_account management command.

[Maximiliano Bertacchini (maxiberta)]

Looks good to me! (with a question)

review: Approve

[Daniel Manrique (roadmr)]

yay

review: Approve

=== added file 'src/identityprovider/management/commands/suspend_account.py'
--- src/identityprovider/management/commands/suspend_account.py	1970-01-01 00:00:00 +0000
+++ src/identityprovider/management/commands/suspend_account.py	2018-07-11 21:00:51 +0000
@@ -0,0 +1,24 @@
+from django.core.management.base import BaseCommand, CommandError
+
+from identityprovider.models import Account
+from identityprovider.models.const import AccountStatus
+
+
+class Command(BaseCommand):
+
+    help = ("Suspend an account identified by displayname.")
+
+    def add_arguments(self, parser):
+        parser.add_argument('username')
+
+    def handle(self, *args, **options):
+        name = options['username']
+
+        try:
+            account = Account.objects.get(displayname=name)
+        except Account.DoesNotExist:
+            raise CommandError("Account {} not found".format(name))
+        account.status = AccountStatus.SUSPENDED
+        account.save()
+
+        print("Suspended account {}".format(account))
 
=== added file 'src/identityprovider/tests/test_command_suspend_account.py'
--- src/identityprovider/tests/test_command_suspend_account.py	1970-01-01 00:00:00 +0000
+++ src/identityprovider/tests/test_command_suspend_account.py	2018-07-11 21:00:51 +0000
@@ -0,0 +1,46 @@
+from django.core.management import CommandError, call_command
+
+from identityprovider.models import Account
+from identityprovider.models.const import AccountStatus
+from identityprovider.tests.utils import SSOBaseTestCase
+
+
+class SuspendAccountCommandTestCase(SSOBaseTestCase):
+
+    def setUp(self):
+        super(SuspendAccountCommandTestCase, self).setUp()
+        self.test_name = 'foo-user'
+
+        self.account = self.factory.make_account(
+            status=AccountStatus.ACTIVE,
+            email="[e-mail]",
+            displayname=self.test_name,
+            email_validated=True)
+
+    def test_missing_arg(self):
+        with self.assertRaises(CommandError) as err:
+            call_command('suspend_account')
+        self.assertEqual("Error: too few arguments", str(err.exception))
+
+    def test_extra_arg(self):
+        with self.assertRaises(CommandError) as err:
+            call_command(
+                'suspend_account', self.test_name, "somethingelse")
+        self.assertEqual("Error: unrecognized arguments: somethingelse",
+                         str(err.exception))
+
+    def test_suspend_account(self):
+        call_command('suspend_account', self.test_name)
+
+        self.assertEqual(
+            AccountStatus.SUSPENDED,
+            Account.objects.get(displayname=self.test_name).status)
+
+    def test_suspend_account_not_found(self):
+        bogusname = self.test_name + "bogus"
+
+        with self.assertRaises(CommandError) as err:
+            call_command('suspend_account', bogusname)
+        self.assertEqual(
+            "Account {} not found".format(bogusname),
+            str(err.exception))