refector old update_password tests

awx/main/tests/functional/commands/test_commands.py

@@ -0,0 +1,52 @@
+import sys
+import pytest
+
+try:
+    from cStringIO import StringIO
+except ImportError:
+    from StringIO import StringIO
+
+from django.core.management import call_command
+
+from awx.main.management.commands.update_password import UpdatePassword
+
+def run_command(name, *args, **options):
+    command_runner = options.pop('command_runner', call_command)
+    stdin_fileobj = options.pop('stdin_fileobj', None)
+    options.setdefault('verbosity', 1)
+    options.setdefault('interactive', False)
+    original_stdin = sys.stdin
+    original_stdout = sys.stdout
+    original_stderr = sys.stderr
+    if stdin_fileobj:
+        sys.stdin = stdin_fileobj
+    sys.stdout = StringIO()
+    sys.stderr = StringIO()
+    result = None
+    try:
+        result = command_runner(name, *args, **options)
+    except Exception as e:
+        result = e
+    finally:
+        captured_stdout = sys.stdout.getvalue()
+        captured_stderr = sys.stderr.getvalue()
+        sys.stdin = original_stdin
+        sys.stdout = original_stdout
+        sys.stderr = original_stderr
+    return result, captured_stdout, captured_stderr
+
+@pytest.mark.parametrize(
+    "command,username,password,expected,changed", [
+        ('update_password', 'admin', 'dingleberry', 'Password updated\n', True),
+        ('update_password', 'admin', 'admin', 'Password not updated\n', False),
+        ('update_password', None, 'foo', 'username required', False),
+        ('update_password', 'admin', None, 'password required', False),
+    ]
+)
+def test_run_command(mocker, command, username, password, expected, changed):
+    with mocker.patch.object(UpdatePassword, 'update_password', return_value=changed):
+        result, stdout, stderr = run_command(command, username=username, password=password)
+        if result is None:
+            assert stdout == expected
+        else:
+            assert str(result) == expected