Skip to content

Commit 14a516a

Browse files
fix failing tests
1 parent b32da6b commit 14a516a

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

airflow-core/tests/unit/jobs/test_scheduler_job.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def set_instance_attrs(self) -> Generator:
234234
@pytest.fixture
235235
def mock_executors(self):
236236
mock_jwt_generator = MagicMock(spec=JWTGenerator)
237-
mock_jwt_generator.generate.return_value = "mock-token"
237+
mock_jwt_generator.generate_queue_token.return_value = "mock-token"
238238

239239
default_executor = mock.MagicMock(name="DefaultExecutor", slots_available=8, slots_occupied=0)
240240
default_executor.name = ExecutorName(alias="default_exec", module_path="default.exec.module.path")

devel-common/src/tests_common/test_utils/mock_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def __init__(self, do_update=True, *args, **kwargs):
5858

5959
# Mock JWT generator for token generation
6060
mock_jwt_generator = MagicMock()
61-
mock_jwt_generator.generate.return_value = "mock-token"
61+
mock_jwt_generator.generate_queue_token.return_value = "mock-token"
6262

6363
self.jwt_generator = mock_jwt_generator
6464

0 commit comments

Comments
 (0)