def receive(self, context: AbstractContext) -> None: msg = context.message if isinstance(msg, RemoteTerminate): self.__watched.pop(msg.watcher.id, 'None') self.__watcher.pop(msg.watchee.id, 'None') t = Terminated(who=msg.watchee, address_terminated=True) msg.watcher.send_system_message(t) return None if isinstance(msg, EndpointTerminatedEvent): for id, pid in self.__watched.items(): t = Terminated(who=pid, address_terminated=True) watcher = PID(ProcessRegistry().Address, id, None) watcher.send_system_message(t) return None if isinstance(msg, RemoteUnwatch): self.__watched[msg.watcher.id] = None self.__watcher[msg.watchee.id] = None w = Unwatch(msg.watcher) msg.watchee.send_system_message(w) return None if isinstance(msg, RemoteWatch): self.__watched[msg.watcher.id] = msg.watchee self.__watcher[msg.watchee.id] = msg.watcher w = Watch(msg.watcher) msg.watchee.send_system_message(w) return None return None
def test_get_nonhost(nohost: ProcessRegistry, mock_process): _pid = ProcessRegistry().add('new_id', mock_process) assert ProcessRegistry().get(_pid) == mock_process
def test_get_deadletter(): _pid = mock.Mock() _pid.address = 'other_host' _pid.pid = '9999' assert isinstance(ProcessRegistry().get(_pid), DeadLettersProcess) is True
def test_remove(nohost: ProcessRegistry, mock_process): _pid = ProcessRegistry().add('new_id', mock_process) assert ProcessRegistry().get(_pid) == mock_process ProcessRegistry().remove(_pid) assert isinstance(ProcessRegistry().get(_pid), DeadLettersProcess) is True
def test_add(nohost: ProcessRegistry, mock_process): _pid = ProcessRegistry().add('new_id', mock_process) assert _pid.address == ProcessRegistry().address assert _pid.id == 'new_id' assert _pid.process == mock_process
def nohost(): return ProcessRegistry()