Skip to content

Commit c6c6691

Browse files
committed
Fix tests
1 parent 6a2ccce commit c6c6691

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/tests/_internal/server/background/pipeline_tasks/test_submitted_jobs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import asyncio
22
import uuid
33
from datetime import timedelta
4+
from typing import cast
45
from unittest.mock import Mock, patch
56

67
import pytest
@@ -538,7 +539,8 @@ async def test_defers_new_capacity_provisioning_until_fleet_master_is_elected(
538539
assert job.lock_owner is None
539540
assert job.lock_token is None
540541
assert job.lock_expires_at is None
541-
worker._pipeline_hinter.hint_fetch.assert_called_once_with(FleetModel.__name__)
542+
hint_fetch = cast(Mock, worker._pipeline_hinter.hint_fetch)
543+
hint_fetch.assert_called_once_with(FleetModel.__name__)
542544

543545
async def test_provisioning_non_master_job_ignores_cluster_master_fleet_lock(
544546
self, test_db, session: AsyncSession, worker: JobSubmittedWorker

0 commit comments

Comments
 (0)