Skip to content

Commit

Permalink
fixup! Update tests/cli/commands/remote_commands/test_dag_command.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ephraimbuddy committed Jan 14, 2025
1 parent f260143 commit 45aa606
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/cli/commands/remote_commands/test_dag_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 45aa606

Please sign in to comment.