updated test

apache · Feb 4, 2025 · 3a35817 · 3a35817
1 parent d5ddaf3
commit 3a35817
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/dag_processing/test_dag_bundles.py b/tests/dag_processing/test_dag_bundles.py
@@ -147,7 +147,6 @@ def setup_class(cls) -> None:
                 conn_id=CONN_ONLY_PATH,
                 host="path/to/repo",
                 conn_type="git",
-                extra={"key_file": "path/to/key"},
             )
         )
         db.merge_conn(
@@ -212,11 +211,11 @@ def test_given_both_private_key_and_key_file(self):
             GitHook(git_conn_id=CONN_BOTH_PATH_INLINE)
 
     def test_key_file_git_hook_has_env(self):
-        hook = GitHook(git_conn_id=CONN_ONLY_PATH)
+        hook = GitHook(git_conn_id=CONN_DEFAULT)
 
         assert hasattr(hook, "env")
         assert hook.env == {
-            "GIT_SSH_COMMAND": "ssh -i path/to/key -o IdentitiesOnly=yes -o StrictHostKeyChecking=no"
+            "GIT_SSH_COMMAND": "ssh -i /files/pkey.pem -o IdentitiesOnly=yes -o StrictHostKeyChecking=no"
         }
 
     def test_private_key_lazy_env_var(self):