Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions tests/integration/test_heartbeat_sender.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
# =================================================================================================
# ↓ BOOTCAMPERS MODIFY BELOW THIS COMMENT ↓
# =================================================================================================
# Add your own constants here
controller = worker_controller.WorkerController()

# =================================================================================================
# ↑ BOOTCAMPERS MODIFY ABOVE THIS COMMENT ↑
Expand All @@ -44,13 +44,11 @@ def start_drone() -> None:
# =================================================================================================
# ↓ BOOTCAMPERS MODIFY BELOW THIS COMMENT ↓
# =================================================================================================
def stop(
args, # Add any necessary arguments
) -> None:
def stop() -> None:
"""
Stop the workers.
"""
pass # Add logic to stop your worker
controller.request_exit() # Add logic to stop your worker


# =================================================================================================
Expand Down Expand Up @@ -95,10 +93,12 @@ def main() -> int:
# Create a worker controller for your worker

# Just set a timer to stop the worker after a while, since the worker infinite loops
threading.Timer(HEARTBEAT_PERIOD * NUM_TRIALS, stop, (args,)).start()
threading.Timer(HEARTBEAT_PERIOD * NUM_TRIALS, stop).start()

heartbeat_sender_worker.heartbeat_sender_worker(
# Place your own arguments here
connection,
controller,
)
# =============================================================================================
# ↑ BOOTCAMPERS MODIFY ABOVE THIS COMMENT ↑
Expand Down
Loading