From 45aa606a30471f1dc6f76c2cbb5fa2540e1b91bb Mon Sep 17 00:00:00 2001 From: Ephraim Anierobi Date: Tue, 14 Jan 2025 08:55:47 +0100 Subject: [PATCH] fixup! Update tests/cli/commands/remote_commands/test_dag_command.py --- tests/cli/commands/remote_commands/test_dag_command.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/cli/commands/remote_commands/test_dag_command.py b/tests/cli/commands/remote_commands/test_dag_command.py index 8fc52eee769ed..04b0ef88104e7 100644 --- a/tests/cli/commands/remote_commands/test_dag_command.py +++ b/tests/cli/commands/remote_commands/test_dag_command.py @@ -109,13 +109,14 @@ def test_reserialize_should_support_bundle_name_argument(self, configure_testing path_to_parse = TEST_DAGS_FOLDER / "test_dag_with_no_tags.py" with configure_testing_dag_bundle(path_to_parse): + # reserializes only the above path dag_command.dag_reserialize( self.parser.parse_args(["dags", "reserialize", "--bundle-name", "testing"]) ) # Check serialized DAG are back serialized_dags_after_reserialize = session.query(SerializedDagModel).all() - assert len(serialized_dags_after_reserialize) == 1 # Serialized DAG back + assert len(serialized_dags_after_reserialize) == 1 def test_reserialize_should_support_more_than_one_bundle(self, configure_testing_dag_bundle, session): # Run clear of serialized dags @@ -128,6 +129,7 @@ def test_reserialize_should_support_more_than_one_bundle(self, configure_testing path_to_parse = TEST_DAGS_FOLDER / "test_dag_with_no_tags.py" with configure_testing_dag_bundle(path_to_parse): + # The command will now serialize the above bundle and the example dag bundle dag_command.dag_reserialize(self.parser.parse_args(["dags", "reserialize"])) # Check serialized DAG are back