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