oid

dags/b2shareoperator.py

@@ -44,7 +44,7 @@ class B2ShareOperator(BaseOperator):
 
     def execute(self, **kwargs):
         hook = HttpHook(http_conn_id=self.conn_id, method='GET')
-
+        print(kwargs)
         params = kwargs['context']['params']
         oid = params['oid']
         hrespo = hook.run(endpoint=f"/api/records/{oid}")

tests/test_b2shareoperator.py

@@ -20,7 +20,7 @@ class B2ShareOperatorTest(unittest.TestCase):
 
    def test_execute_no_trigger(self):
        
-       self.ti.run(ignore_ti_state=False)
+       self.ti.run(ignore_ti_state=False, context={'params': {'oid': 111}})
        print(self.ti.state)
        self.assertEqual(State.SUCCESS, self.ti.state)
        # Assert something related to tasks results
\ No newline at end of file