From de27d211deec7ba35e26d33b4330d671690b39aa Mon Sep 17 00:00:00 2001 From: Ephraim Anierobi Date: Wed, 15 Jan 2025 13:10:01 +0100 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Jed Cunningham <66968678+jedcunningham@users.noreply.github.com> --- tests/dag_processing/test_dag_bundles.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/dag_processing/test_dag_bundles.py b/tests/dag_processing/test_dag_bundles.py index b91f0455ee1be..6032136610b27 100644 --- a/tests/dag_processing/test_dag_bundles.py +++ b/tests/dag_processing/test_dag_bundles.py @@ -205,7 +205,6 @@ def test_uses_dag_bundle_root_storage_path(self, git_repo): @mock.patch("airflow.dag_processing.bundles.git.GitHook") def test_get_current_version(self, mock_githook, git_repo): - mock_githook.get_conn.return_value = mock.MagicMock() repo_path, repo = git_repo mock_githook.return_value.repo_url = repo_path bundle = GitDagBundle(name="test", refresh_interval=300, tracking_ref=GIT_DEFAULT_BRANCH) @@ -423,7 +422,7 @@ def test_refresh_with_git_connection(self, mock_gitRepo, mock_hook): ], ) @mock.patch("airflow.dag_processing.bundles.git.GitHook") - def test_repo_url_starts_with_git_when_using_ssh_conn_id(self, mock_hook, repo_url, session): + def test_repo_url_validation_for_ssh(self, mock_hook, repo_url, session): mock_hook.get_conn.return_value = mock.MagicMock() mock_hook.return_value.repo_url = repo_url bundle = GitDagBundle(