Skip to content

Commit

Permalink
[cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py] Increase t…
Browse files Browse the repository at this point in the history
…est coverage via introducing `test_column_call_to_param_server_default`
  • Loading branch information
SamuelMarks committed Sep 26, 2023
1 parent 995aa60 commit 93c3246
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,31 @@ def test_column_call_to_param_pk(self) -> None:
self.assertEqual(gold_name, gen_name)
self.assertDictEqual(gold_param, gen_param)

def test_column_call_to_param_server_default(self) -> None:
"""
Tests that `parse.sqlalchemy.utils.column_call_to_param` works with server_default
"""

gold_name, gold_param = (
lambda _name: (
_name,
{
"server_default": config_schema["properties"][_name]["default"],
"typ": "str",
"doc": config_schema["properties"][_name]["description"],
"x_typ": {"sql": {"type": "String"}},
},
)
)("dataset_name")
gen_name, gen_param = column_call_to_param(
column_call_name_manipulator(
deepcopy(dataset_primary_key_column_assign.value), "add", gold_name
)
)
gen_param["server_default"] = gen_param.pop("default")
self.assertEqual(gold_name, gen_name)
self.assertDictEqual(gold_param, gen_param)

def test_column_call_to_param_fk(self) -> None:
"""
Tests that `parse.sqlalchemy.utils.column_call_to_param` works with FK
Expand Down

0 comments on commit 93c3246

Please sign in to comment.