chore(main): init subcommand args for GAM, GYB

Author: thekaveman

compiler_admin/main.py

@@ -35,10 +35,12 @@ def _subcmd(name, help, add_username_arg=True) -> argparse.ArgumentParser:
 
     _subcmd("info", help="Print configuration and debugging information.", add_username_arg=False)
 
-    _subcmd(
+    init_parser = _subcmd(
         "init",
         help="Initialize a new admin project. This command should be run once before any others.",
     )
+    init_parser.add_argument("--gam", action="store_true", help="If provided, initialize a new GAM project.")
+    init_parser.add_argument("--gyb", action="store_true", help="If provided, initialize a new GYB project.")
 
     _subcmd("create", help="Create a new user in the Compiler domain.")
 
@@ -70,7 +72,7 @@ def _subcmd(name, help, add_username_arg=True) -> argparse.ArgumentParser:
     elif args.command == "delete":
         return delete(args.username)
     elif args.command == "init":
-        return init(args.username)
+        return init(args.username, gam=args.gam, gyb=args.gyb)
     elif args.command == "offboard":
         return offboard(args.username, args.alias)
     elif args.command == "restore":

tests/test_main.py

@@ -108,12 +108,28 @@ def test_main_info_default(mock_commands_info):
     mock_commands_info.assert_called_once()
 
 
-def test_main_init(mock_commands_init):
+def test_main_init_default(mock_commands_init):
     main(argv=["init", "username"])
 
     mock_commands_init.assert_called_once()
-    call_args = mock_commands_init.call_args.args
-    assert "username" in call_args
+    assert mock_commands_init.call_args.args == ("username",)
+    assert mock_commands_init.call_args.kwargs == {"gam": False, "gyb": False}
+
+
+def test_main_init_gam(mock_commands_init):
+    main(argv=["init", "username", "--gam"])
+
+    mock_commands_init.assert_called_once()
+    assert mock_commands_init.call_args.args == ("username",)
+    assert mock_commands_init.call_args.kwargs == {"gam": True, "gyb": False}
+
+
+def test_main_init_gyb(mock_commands_init):
+    main(argv=["init", "username", "--gyb"])
+
+    mock_commands_init.assert_called_once()
+    assert mock_commands_init.call_args.args == ("username",)
+    assert mock_commands_init.call_args.kwargs == {"gam": False, "gyb": True}
 
 
 def test_main_init_no_username(mock_commands_init):