Add a restart mechanism to all workers in the multiplexer (#2622)
This commit is contained in:
parent
8e720365c2
commit
92e7463721
|
@ -21,9 +21,14 @@ class WorkerMultiplexer:
|
|||
"state": ProcessState.ACKED.name,
|
||||
}
|
||||
|
||||
def restart(self, name: str = ""):
|
||||
def restart(self, name: str = "", all_workers: bool = False):
|
||||
if name and all_workers:
|
||||
raise ValueError(
|
||||
"Ambiguous restart with both a named process and"
|
||||
" all_workers=True"
|
||||
)
|
||||
if not name:
|
||||
name = self.name
|
||||
name = "__ALL_PROCESSES__:" if all_workers else self.name
|
||||
self._monitor_publisher.send(name)
|
||||
|
||||
reload = restart # no cov
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
from multiprocessing import Event
|
||||
from os import environ, getpid
|
||||
from typing import Any, Dict
|
||||
from typing import Any, Dict, Type, Union
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
|
@ -117,3 +117,26 @@ def test_properties(
|
|||
assert m.workers == worker_state
|
||||
assert m.state == worker_state["Test"]
|
||||
assert isinstance(m.state, WorkerState)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"params,expected",
|
||||
(
|
||||
({}, "Test"),
|
||||
({"name": "foo"}, "foo"),
|
||||
({"all_workers": True}, "__ALL_PROCESSES__:"),
|
||||
({"name": "foo", "all_workers": True}, ValueError),
|
||||
),
|
||||
)
|
||||
def test_restart_params(
|
||||
monitor_publisher: Mock,
|
||||
m: WorkerMultiplexer,
|
||||
params: Dict[str, Any],
|
||||
expected: Union[str, Type[Exception]],
|
||||
):
|
||||
if isinstance(expected, str):
|
||||
m.restart(**params)
|
||||
monitor_publisher.send.assert_called_once_with(expected)
|
||||
else:
|
||||
with pytest.raises(expected):
|
||||
m.restart(**params)
|
||||
|
|
Loading…
Reference in New Issue
Block a user