From 5355e9d49fb04aeab631ff6ce04904475c97178d Mon Sep 17 00:00:00 2001
From: Christian Boettcher <c.boettcher@fz-juelich.de>
Date: Fri, 24 Jun 2022 08:13:57 +0200
Subject: [PATCH] allow testing different oid for secrets backend test
---
dags/test_secrets_backend.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/dags/test_secrets_backend.py b/dags/test_secrets_backend.py
index f840966..1cfe574 100644
--- a/dags/test_secrets_backend.py
+++ b/dags/test_secrets_backend.py
@@ -11,8 +11,12 @@ default_args = {
@dag(default_args=default_args, schedule_interval=None, start_date=days_ago(2), tags=['example'])
def test_secrets_backend():
@task()
- def get_print_and_return_conenction():
- conn = BaseHook.get_connection('860355e9-975f-4253-9421-1815e20c879b')
+ def get_print_and_return_conenction(**kwargs):
+ oid = '860355e9-975f-4253-9421-1815e20c879b'
+ params = kwargs['params']
+ if 'oid' in params:
+ oid = params['oid']
+ conn = BaseHook.get_connection(oid)
print(conn.get_extra())
get_print_and_return_conenction()
--
GitLab