def setUp(self):
self.pd_service = ProcessDispatcherService()
self.pd_service.container = DotDict()
self.pd_service.container['spawn_process'] = Mock()
self.pd_service.container['id'] = 'mock_container_id'
self.pd_service.container['proc_manager'] = DotDict()
self.pd_service.container['resource_registry'] = Mock()
self.pd_service.container.proc_manager['terminate_process'] = Mock()
self.pd_service.container.proc_manager['procs'] = {}
pdcfg = dict(dashi_uri="amqp://hello",
dashi_exchange="123",
static_resources=True,
backend="native")
self.pd_service.CFG = DotDict()
self.pd_service.CFG['processdispatcher'] = pdcfg
self.mock_dashi = Mock()
with patch.multiple('ion.services.cei.process_dispatcher_service',
get_dashi=DEFAULT,
ProcessDispatcherCore=DEFAULT,
get_processdispatcher_store=DEFAULT,
EngineRegistry=DEFAULT,
PDMatchmaker=DEFAULT,
PDDoctor=DEFAULT) as mocks:
mocks['get_dashi'].return_value = self.mock_dashi
mocks[
'get_processdispatcher_store'].return_value = self.mock_store = Mock(
)
mocks[
'ProcessDispatcherCore'].return_value = self.mock_core = Mock(
)
mocks['PDMatchmaker'].return_value = self.mock_matchmaker = Mock()
mocks['PDDoctor'].return_value = self.mock_doctor = Mock()
mocks[
'EngineRegistry'].return_value = self.mock_engineregistry = Mock(
)
self.pd_service.init()
# replace the core and matchmaker with mocks
self.pd_service.backend.beat_subscriber = self.mock_beat_subscriber = Mock(
)
self.assertIsInstance(self.pd_service.backend, PDNativeBackend)
self.pd_service.backend.rr = self.mock_rr = Mock()
self.event_pub = Mock()
self.pd_service.backend.event_pub = self.event_pub
self.pd_service.start()
self.assertEqual(self.mock_dashi.handle.call_count, 1)
self.mock_matchmaker.start_election.assert_called_once_with()
self.mock_beat_subscriber.start.assert_called_once_with()
def setUp(self):
self.pd_service = ProcessDispatcherService()
self.pd_service.container = DotDict()
self.pd_service.container['spawn_process'] = Mock()
self.pd_service.container['id'] = 'mock_container_id'
self.pd_service.container['proc_manager'] = DotDict()
self.pd_service.container['resource_registry'] = Mock()
self.pd_service.container.proc_manager['terminate_process'] = Mock()
self.pd_service.container.proc_manager['procs'] = {}
self.mock_cc_spawn = self.pd_service.container.spawn_process
self.mock_cc_terminate = self.pd_service.container.proc_manager.terminate_process
self.mock_cc_procs = self.pd_service.container.proc_manager.procs
self.pd_service.init()
self.assertIsInstance(self.pd_service.backend, PDLocalBackend)
self.pd_service.backend.rr = self.mock_rr = Mock()
self.pd_service.backend.event_pub = self.mock_event_pub = Mock()
def setUp(self):
mock_clients = self._create_service_mock('process_dispatcher')
self.pd_service = ProcessDispatcherService()
self.pd_service.clients = mock_clients
self.pd_service.container = DotDict()
self.pd_service.container['spawn_process'] = Mock()
self.pd_service.container['id'] = 'mock_container_id'
self.pd_service.container['proc_manager'] = DotDict()
self.pd_service.container.proc_manager['terminate_process'] = Mock()
self.pd_service.container.proc_manager['procs'] = {}
# CRUD Shortcuts
self.mock_rr_create = self.pd_service.clients.resource_registry.create
self.mock_rr_read = self.pd_service.clients.resource_registry.read
self.mock_rr_update = self.pd_service.clients.resource_registry.update
self.mock_rr_delete = self.pd_service.clients.resource_registry.delete
self.mock_rr_find = self.pd_service.clients.resource_registry.find_objects
self.mock_rr_find_res = self.pd_service.clients.resource_registry.find_resources
self.mock_rr_assoc = self.pd_service.clients.resource_registry.find_associations
self.mock_rr_create_assoc = self.pd_service.clients.resource_registry.create_association
self.mock_rr_del_assoc = self.pd_service.clients.resource_registry.delete_association
self.mock_cc_spawn = self.pd_service.container.spawn_process
self.mock_cc_terminate = self.pd_service.container.proc_manager.terminate_process
self.mock_cc_procs = self.pd_service.container.proc_manager.procs
