Skip to content

Commit

Permalink
add more tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ephraimbuddy committed Dec 21, 2024
1 parent 6652360 commit bcc7524
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions tests/dag_processing/test_dag_bundles.py
Original file line number Diff line number Diff line change
Expand Up @@ -288,3 +288,27 @@ def test_view_url(self, mock_has_version, mock_gitrepo, repo_url, expected_url):
)
view_url = bundle.view_url("0f0f0f")
assert view_url == expected_url

@mock.patch("airflow.dag_processing.bundles.git.Repo")
def test_view_url_raises_if_no_version(self, mock_gitrepo):
bundle = GitDagBundle(
name="test",
refresh_interval=300,
repo_url="[email protected]:apache/airflow.git",
tracking_ref="main",
)
with pytest.raises(AirflowException, match="Version is required to view the repository"):
bundle.view_url(None)

@mock.patch("airflow.dag_processing.bundles.git.Repo")
@mock.patch.object(GitDagBundle, "_has_version")
def test_view_url_raises_if_version_not_found(self, mock_has_version, mock_gitrepo):
mock_has_version.return_value = False
bundle = GitDagBundle(
name="test",
refresh_interval=300,
repo_url="[email protected]:apache/airflow.git",
tracking_ref="main",
)
with pytest.raises(AirflowException, match="Version not_found not found in the repository"):
bundle.view_url("not_found")

0 comments on commit bcc7524

Please sign in to comment.