Skip to content

Commit 9186e76

Browse files
committed
Update test_worker.py
1 parent 9d5c657 commit 9186e76

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_serverless/test_worker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,14 +488,14 @@ def mock_is_alive():
488488
res = scale_behavior['counter'] <= 10
489489
scale_behavior['counter'] += 1
490490

491-
# Let's oscillate between upscaling, downscaling, upscaling, downscaling, ...
491+
# Mock get_job to return a job every other call.
492492
if scale_behavior['counter'] % 2 == 0:
493493
mock_get_job.return_value = {
494494
"id": "123", "input": {"number": 1}}
495495
else:
496496
mock_get_job.return_value = None
497497

498-
return bool(res)
498+
return bool(mock_run_job.call_count > 5)
499499

500500
# Define the mock behaviors
501501
mock_run_job.return_value = {"result": "odd"}

0 commit comments

Comments
 (0)