test(commands/create): ensure extra args are passed along

Author: thekaveman

tests/commands/test_create.py

@@ -68,3 +68,20 @@ def test_create_user_notify(mock_google_user_exists, mock_google_CallGAMCommand,
     assert f"notify [email protected] from {USER_HELLO}" in call_args
 
     mock_google_add_user_to_group.assert_called_once()
+
+
+def test_create_user_extras(mock_google_user_exists, mock_google_CallGAMCommand, mock_google_add_user_to_group):
+    mock_google_user_exists.return_value = False
+
+    args = Namespace(username="username")
+    res = create(args, "extra1", "extra2")
+
+    assert res == RESULT_SUCCESS
+
+    mock_google_CallGAMCommand.assert_called_once()
+    call_args = " ".join(mock_google_CallGAMCommand.call_args[0][0])
+    assert "create user" in call_args
+    assert "password random changepassword" in call_args
+    assert "extra1 extra2" in call_args
+
+    mock_google_add_user_to_group.assert_called_once()

tests/test_main.py

@@ -70,6 +70,16 @@ def test_main_create_notify(mock_commands_create):
     assert Namespace(command="create", username="username", notify="notification") in call_args
 
 
+def test_main_create_extras(mock_commands_create):
+    main(argv=["create", "username", "extra1", "extra2"])
+
+    mock_commands_create.assert_called_once()
+    call_args = mock_commands_create.call_args.args
+    assert Namespace(command="create", username="username", notify=None) in call_args
+    assert "extra1" in call_args
+    assert "extra2" in call_args
+
+
 def test_main_create_no_username(mock_commands_create):
     with pytest.raises(SystemExit):
         main(argv=["create"])