Add --value-type option to the CLI

python/enumecg/cli.py

@@ -27,8 +27,9 @@
     type=click.Choice(PRIMARY_TYPE_CHOICES),
     help="Primary enumeration type",
 )
+@click.option("--value-type", help="Enumerator value type")
 @click.argument("file", type=click.File())
-def cli(file=sys.stdin, documentation=None, primary_type=None):
+def cli(file=sys.stdin, documentation=None, primary_type=None, value_type=None):
     """Generate C++ boilerplate for an Enhanced Enum definition
 
     This executable is a part of the Enhanced Enum library. It is used
@@ -46,4 +47,11 @@ def cli(file=sys.stdin, documentation=None, primary_type=None):
 
     """
     enum = yaml.safe_load(file)
-    click.echo(generate(enum, documentation=documentation, primary_type=primary_type))
+    click.echo(
+        generate(
+            enum,
+            documentation=documentation,
+            primary_type=primary_type,
+            value_type=value_type,
+        )
+    )

python/tests/test_cli.py

@@ -43,3 +43,10 @@ def test_cli_should_have_primeray_type_option(
         result.output
         == generate(status_definition_dict, primary_type=primary_type) + "\n"
     )
+
+
+def test_cli_should_have_value_type_option(
+    cli_runner, enum_file, status_definition_dict
+):
+    result = cli_runner.invoke(cli, ["--value-type", "MyType", str(enum_file)])
+    assert result.output == generate(status_definition_dict, value_type="MyType") + "\n"