Add tests for YamlParamType

Author: erindru

sqlmesh_dbt/cli.py

@@ -91,6 +91,7 @@ def run(ctx: click.Context, vars: t.Optional[t.Dict[str, t.Any]], **kwargs: t.An
 @dbt.command(name="list")
 @select_option
 @exclude_option
+@vars_option
 @click.pass_context
 def list_(ctx: click.Context, vars: t.Optional[t.Dict[str, t.Any]], **kwargs: t.Any) -> None:
     """List the resources in your project"""

tests/dbt/cli/test_list.py

@@ -59,4 +59,4 @@ def test_list_with_vars(jaffle_shop_duckdb: Path, invoke_cli: t.Callable[..., Re
     assert result.exit_code == 0
     assert not result.exception
 
-    assert "model_bar" in result.output
+    assert "model_bar" in result.output

tests/dbt/cli/test_options.py

@@ -0,0 +1,23 @@
+import typing as t
+import pytest
+from sqlmesh_dbt.options import YamlParamType
+from click.exceptions import BadParameter
+
+
+@pytest.mark.parametrize(
+    "input,expected",
+    [
+        (1, BadParameter("Input value '1' should be a string")),
+        ("", BadParameter("String '' is not valid YAML")),
+        ("['a', 'b']", BadParameter("String.*did not evaluate to a dict, got.*")),
+        ("foo: bar", {"foo": "bar"}),
+        ('{"key": "value", "date": 20180101}', {"key": "value", "date": 20180101}),
+        ("{key: value, date: 20180101}", {"key": "value", "date": 20180101}),
+    ],
+)
+def test_yaml_param_type(input: str, expected: t.Union[BadParameter, t.Dict[str, t.Any]]):
+    if isinstance(expected, BadParameter):
+        with pytest.raises(BadParameter, match=expected.message):
+            YamlParamType().convert(input, None, None)
+    else:
+        assert YamlParamType().convert(input, None, None) == expected