Taragolis commented on code in PR #37937:
URL: https://github.com/apache/airflow/pull/37937#discussion_r1514470722
##########
tests/models/test_baseoperatormeta.py:
##########
@@ -0,0 +1,64 @@
+import os
+from typing import Any
+from unittest import TestCase
+from unittest.mock import Mock, patch, MagicMock
+
+import pendulum
+from sqlalchemy.orm import Session
+
+from airflow import AirflowException, DAG
+from airflow.models import TaskInstance, DagRun
+from airflow.models.baseoperator import BaseOperator
+from airflow.utils.context import Context
+from airflow.utils.state import DagRunState
+
+
+class HelloWorldOperator(BaseOperator):
+ called = False
+
+ def execute(self, context: Context) -> Any:
+ HelloWorldOperator.called = True
+ return f"Hello {self.owner}!"
+
+
+class IterableSession(Session):
+ def __next__(self):
+ pass
+
+
+class BaseOperatorMetaTestCase(TestCase):
+
+ @patch.dict(os.environ, {'AIRFLOW__CORE__UNIT_TEST_MODE': 'False'})
Review Comment:
We have mocker config helper in `tests.test_utils.config.conf_vars`
Some samples
https://github.com/apache/airflow/blob/8f1418f4de89b334f188d7d5a3b0f348ad4a0372/tests/cli/commands/test_config_command.py#L167-L173
https://github.com/apache/airflow/blob/1aa91a482630f389005a9a36b0f038e675163411/tests/www/views/test_views_robots.py#L31-L40
##########
tests/models/test_baseoperatormeta.py:
##########
@@ -0,0 +1,64 @@
+import os
+from typing import Any
+from unittest import TestCase
+from unittest.mock import Mock, patch, MagicMock
+
+import pendulum
+from sqlalchemy.orm import Session
+
+from airflow import AirflowException, DAG
+from airflow.models import TaskInstance, DagRun
+from airflow.models.baseoperator import BaseOperator
+from airflow.utils.context import Context
+from airflow.utils.state import DagRunState
+
+
+class HelloWorldOperator(BaseOperator):
+ called = False
+
+ def execute(self, context: Context) -> Any:
+ HelloWorldOperator.called = True
+ return f"Hello {self.owner}!"
+
+
+class IterableSession(Session):
+ def __next__(self):
+ pass
+
+
+class BaseOperatorMetaTestCase(TestCase):
Review Comment:
We do not use `unittest.TestCase` I guess it banned in multiple places.
##########
tests/models/test_baseoperatormeta.py:
##########
@@ -0,0 +1,64 @@
+import os
+from typing import Any
+from unittest import TestCase
+from unittest.mock import Mock, patch, MagicMock
+
+import pendulum
+from sqlalchemy.orm import Session
+
+from airflow import AirflowException, DAG
+from airflow.models import TaskInstance, DagRun
+from airflow.models.baseoperator import BaseOperator
+from airflow.utils.context import Context
+from airflow.utils.state import DagRunState
+
+
+class HelloWorldOperator(BaseOperator):
+ called = False
+
+ def execute(self, context: Context) -> Any:
+ HelloWorldOperator.called = True
+ return f"Hello {self.owner}!"
+
+
+class IterableSession(Session):
+ def __next__(self):
+ pass
+
+
+class BaseOperatorMetaTestCase(TestCase):
+
+ @patch.dict(os.environ, {'AIRFLOW__CORE__UNIT_TEST_MODE': 'False'})
+ def test_executor_safeguard_when_unauthorized(self):
+ with self.assertRaises(AirflowException):
+ dag = DAG(dag_id="hello_world")
+ context = MagicMock(spec=Context)
+
+ HelloWorldOperator(task_id="task_id",
dag=dag).execute(context=context)
+
+ @patch("sqlalchemy.orm.Session.__init__")
+ @patch.dict(os.environ, {'AIRFLOW__CORE__UNIT_TEST_MODE': 'False'})
+ def test_executor_safeguard_when_authorized(self, mock_session: MagicMock):
+ session = MagicMock(spec=IterableSession)
+ mock_session.return_value = session
+ session.__iter__.return_value = iter({})
+
+ dag = DAG(dag_id="hello_world")
+ context = {"params": {}}
Review Comment:
I'm not sure what you tried to achieve here but seems like it might be
replaced by
[`dag_maker`](https://github.com/apache/airflow/blob/0232ad031890c5f2b1bdbdcd3e28b4c6e16abfb3/tests/conftest.py#L612)
fixture.
##########
tests/models/test_baseoperatormeta.py:
##########
@@ -0,0 +1,64 @@
+import os
+from typing import Any
+from unittest import TestCase
+from unittest.mock import Mock, patch, MagicMock
+
+import pendulum
+from sqlalchemy.orm import Session
+
+from airflow import AirflowException, DAG
+from airflow.models import TaskInstance, DagRun
+from airflow.models.baseoperator import BaseOperator
+from airflow.utils.context import Context
+from airflow.utils.state import DagRunState
+
+
+class HelloWorldOperator(BaseOperator):
+ called = False
+
+ def execute(self, context: Context) -> Any:
+ HelloWorldOperator.called = True
+ return f"Hello {self.owner}!"
+
+
+class IterableSession(Session):
+ def __next__(self):
+ pass
+
+
+class BaseOperatorMetaTestCase(TestCase):
+
+ @patch.dict(os.environ, {'AIRFLOW__CORE__UNIT_TEST_MODE': 'False'})
+ def test_executor_safeguard_when_unauthorized(self):
+ with self.assertRaises(AirflowException):
Review Comment:
```suggestion
with pytest.raises(AirflowException, match="Some pattern of the
exception message here"):
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
