def test_wait_sync():
mock_worker1, mock_worker2 = Mock(), Mock()
manager = SyncManager()
manager.workers.append(mock_worker1)
manager.workers.append(mock_worker2)
manager.wait_sync()
mock_worker1.join.assert_called_once()
mock_worker2.join.assert_called_once()
def _sync(
manager: SyncManager,
status: Status,
workers: int,
sync: Callable,
sync_options: Dict[str, Any],
wait: bool = False,
) -> Tuple[Optional[SyncManager], Status]:
for i in range(1, workers + 1):
sync_options["worker_id"] = i
worker = threading.Thread(target=sync, kwargs=sync_options)
worker.start()
manager.workers.append(worker)
if wait:
manager.wait_sync()
return None, status
return manager, status