This is an automated email from the ASF dual-hosted git repository.
jscheffl pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 33363d54185 fix test in main (#67861)
33363d54185 is described below
commit 33363d541851f3017ee96f94ceb328d684ed8360
Author: Anish Giri <[email protected]>
AuthorDate: Mon Jun 1 15:49:18 2026 -0500
fix test in main (#67861)
---
.../unit/edge3/executors/test_edge_executor.py | 23 ++++++++++++++--------
1 file changed, 15 insertions(+), 8 deletions(-)
diff --git a/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py
b/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py
index 3846826c0a6..3554689b794 100644
--- a/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py
+++ b/providers/edge3/tests/unit/edge3/executors/test_edge_executor.py
@@ -602,7 +602,8 @@ class TestQueueWorkload:
executor = EdgeExecutor()
workload = self._make_execute_task()
- executor.queue_workload(workload)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
with create_session() as session:
job = session.scalar(select(EdgeJobModel))
@@ -617,8 +618,10 @@ class TestQueueWorkload:
executor = EdgeExecutor()
workload = self._make_execute_task()
- executor.queue_workload(workload)
- executor.queue_workload(workload)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
with create_session() as session:
jobs = session.scalars(select(EdgeJobModel)).all()
@@ -644,7 +647,8 @@ class TestQueueWorkload:
log_path="test.log",
)
- executor.queue_workload(workload)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
with create_session() as session:
job = session.scalar(select(EdgeJobModel))
@@ -673,8 +677,10 @@ class TestQueueWorkload:
log_path="test.log",
)
- executor.queue_workload(workload)
- executor.queue_workload(workload)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
+ with create_session() as session:
+ executor.queue_workload(workload, session=session)
with create_session() as session:
jobs = session.scalars(select(EdgeJobModel)).all()
@@ -683,5 +689,6 @@ class TestQueueWorkload:
def test_queue_workload_unknown_type_raises(self):
executor = EdgeExecutor()
- with pytest.raises(TypeError, match="Don't know how to queue
workload"):
- executor.queue_workload(MagicMock(spec=[]))
+ with create_session() as session:
+ with pytest.raises(TypeError, match="Don't know how to queue
workload"):
+ executor.queue_workload(MagicMock(spec=[]), session=session)
