Fix test and make nargs configurable

vkottler · Sep 8, 2023 · a670f02 · a670f02
1 parent 2525823
commit a670f02
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 13 deletions.
diff --git a/runtimepy/commands/common.py b/runtimepy/commands/common.py
@@ -6,12 +6,14 @@
 from argparse import ArgumentParser as _ArgumentParser
 
 
-def arbiter_args(parser: _ArgumentParser) -> None:
+def arbiter_args(parser: _ArgumentParser, nargs: str = "+") -> None:
     """Add common connection-arbiter parameters.."""
 
     parser.add_argument(
         "--init_only",
         action="store_true",
         help="exit after completing initialization",
     )
-    parser.add_argument("configs", nargs="+", help="the configuration to load")
+    parser.add_argument(
+        "configs", nargs=nargs, help="the configuration to load"
+    )
diff --git a/tests/commands/test_arbiter.py b/tests/commands/test_arbiter.py
@@ -13,19 +13,14 @@
 def test_arbiter_command_basic():
     """Test basic usages of the 'arbiter' command."""
 
-    for entry in ["basic.yaml"]:
-        assert (
-            runtimepy_main(
-                [
-                    PKG_NAME,
-                    "arbiter",
-                    "--init_only",
-                    str(resource("connection_arbiter", entry)),
-                ]
-            )
-            == 0
+    assert (
+        runtimepy_main(
+            [PKG_NAME, "arbiter", "--init_only", str(resource("empty.yaml"))]
         )
+        == 0
+    )
 
+    for entry in ["basic.yaml"]:
         assert (
             runtimepy_main(
                 [

diff --git a/tests/data/valid/empty.yaml b/tests/data/valid/empty.yaml
@@ -0,0 +1 @@
+---