From 93c3246b558d07c5c876313578c337464fd90885 Mon Sep 17 00:00:00 2001 From: Samuel Marks <807580+SamuelMarks@users.noreply.github.com> Date: Mon, 25 Sep 2023 21:07:08 -0400 Subject: [PATCH] [cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py] Increase test coverage via introducing `test_column_call_to_param_server_default` --- .../test_parse_sqlalchemy_utils.py | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py b/cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py index 31f11907..15735661 100644 --- a/cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py +++ b/cdd/tests/test_sqlalchemy/test_parse_sqlalchemy_utils.py @@ -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