ephraimbuddy commented on pull request #19487:
URL: https://github.com/apache/airflow/pull/19487#issuecomment-968367900
> I wrote the following test cases and it works:
>
> ```python
> @provide_session
> def test_should_respond_200_for_none_state_filter(self, session):
> dag_id = 'example_python_operator'
> dag = self.dagbag.get_dag(dag_id)
> run_id = "TEST_DAG_RUN_ID"
> dr = DagRun(
> run_id=run_id,
> dag_id=dag_id,
> run_type=DagRunType.MANUAL,
> )
> session.add(dr)
> session.commit()
> ti = TaskInstance(task=dag.tasks[0], run_id=run_id, state=None)
> ti.dag_run = dr
> session.add(ti)
> session.commit()
> response = self.client.get(
> f"/api/v1/dags/{dag_id}/dagRuns/~/taskInstances?state=none",
> environ_overrides={"REMOTE_USER": "test"},
> )
> assert 1 == session.query(TaskInstance).count()
> assert 1 == session.query(DagRun).count()
> assert response.status_code == 200
> assert [('example_python_operator', 'TEST_DAG_RUN_ID',
'print_the_context', None)] == [
> (d['dag_id'], d['dag_run_id'], d['task_id'], d['state']) for d
in response.json['task_instances']
> ]
> response = self.client.get(
> f"/api/v1/dags/{dag_id}/dagRuns/~/taskInstances?state=running",
> environ_overrides={"REMOTE_USER": "test"},
> )
> assert 200 == response.status_code
> assert 0 == len(response.json['task_instances'])
> ```
>
> However, I ran into another problem. I could not run the tests as long as
the `execution_date` field was defined in TaskInstanceSchema. Have you
encountered this problem?
>
> ```
> $ pytest tests/api_connexion/endpoints/test_task_instance_endpoint.py -k
test_should_respond_200_for_none_state_filter -s
> ....
> E sqlalchemy.exc.InvalidRequestError: Mapper 'mapped class
TaskInstance->task_instance' has no property 'execution_date'
>
> /usr/local/lib/python3.6/site-packages/sqlalchemy/util/compat.py:182:
InvalidRequestError
> ```
Yeah.
You have an old version of marshmallow-sqlalchemy. Probably 0.23. You need
to get the latest image or update the package.
--
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]
