def setUp(self):
mock_clients = self._create_service_mock('event_management')
self.event_management = EventManagementService()
self.event_management.clients = mock_clients
self.mock_rr_client = self.event_management.clients.resource_registry
self.mock_pd_client = self.event_management.clients.process_dispatcher
self.mock_pubsub_client = self.event_management.clients.pubsub_management
self.mock_dams_client = self.event_management.clients.data_acquisition_management
class EventManagementTest(PyonTestCase):
def setUp(self):
mock_clients = self._create_service_mock('event_management')
self.event_management = EventManagementService()
self.event_management.clients = mock_clients
self.mock_rr_client = self.event_management.clients.resource_registry
self.mock_pd_client = self.event_management.clients.process_dispatcher
self.mock_pubsub_client = self.event_management.clients.pubsub_management
self.mock_dams_client = self.event_management.clients.data_acquisition_management
def test_create_event_type(self):
"""
Test creating an event
"""
event_type = EventType(name="Test event type")
self.mock_rr_client.create = mocksignature(self.mock_rr_client.create)
self.mock_rr_client.create.return_value = ('event_type_id','rev_1')
self.mock_dams_client.register_process = mocksignature(self.mock_dams_client.register_process)
event_type_id = self.event_management.create_event_type(event_type)
self.assertEquals(event_type_id, 'event_type_id')
self.mock_rr_client.create.assert_called_once_with(event_type)
def test_update_event_type(self):
"""
Test updating an event
"""
event_type = EventType(name="Test event type")
self.mock_rr_client.update = mocksignature(self.mock_rr_client.update)
# self.mock_rr_client.update.return_value = ('event_type_id','rev_1')
self.event_management.update_event_type(event_type)
self.mock_rr_client.update.assert_called_once_with(event_type)
def test_read_event_type(self):
"""
Test reading an event
"""
event_type = Mock()
self.mock_rr_client.read = mocksignature(self.mock_rr_client.read)
self.mock_rr_client.read.return_value = event_type
result = self.event_management.read_event_type(event_type_id='event_type_id')
self.assertEquals(result, event_type)
self.mock_rr_client.read.assert_called_once_with('event_type_id', '')
def test_delete_event_type(self):
"""
Test updating an event
"""
self.mock_rr_client.delete = mocksignature(self.mock_rr_client.delete)
self.event_management.delete_event_type(event_type_id='event_type_id')
self.mock_rr_client.delete.assert_called_once_with('event_type_id')
def test_create_event_process_definition(self):
"""
Test creating an event process definition
"""
process_definition = Mock()
self.mock_pd_client.create_process_definition=mocksignature(self.mock_pd_client.create_process_definition)
self.mock_pd_client.create_process_definition.return_value='procdef_id'
self.mock_pd_client.schedule_process=mocksignature(self.mock_pd_client.schedule_process)
self.mock_pd_client.schedule_process.return_value='pid'
procdef_id = self.event_management.create_event_process_definition(version='version',
module ='module',
class_name='class_name',
uri='uri',
arguments='arguments',
event_types=['t1', 't2'],
sub_types=['t3'],
origin_types=['or1', 'or2'])
self.assertEquals(procdef_id, 'procdef_id')
def test_update_event_process_definition(self):
"""
Test updating an event process definition
"""
process_definition_id = 'an id'
process_def = ProcessDefinition()
self.mock_rr_client.read = mocksignature(self.mock_rr_client.read)
self.mock_rr_client.read.return_value = process_def
self.mock_rr_client.update = mocksignature(self.mock_rr_client.update)
self.event_management.update_event_process_definition( event_process_definition_id =process_definition_id,
version='version',
event_types=['t1', 't2'],
sub_types=['t3'],
origin_types=['or1', 'or2'])
self.mock_rr_client.update.assert_called_once_with(process_def)
def test_read_event_process_definition(self):
"""
Test reading an event process definition
"""
event_process_def = Mock()
self.mock_rr_client.read = mocksignature(self.mock_rr_client.read)
self.mock_rr_client.read.return_value = event_process_def
result = self.event_management.read_event_process_definition()
self.assertEquals(result, event_process_def)
def test_delete_event_process_definition(self):
"""
Test deleting an event process definition
"""
self.mock_rr_client.delete = mocksignature(self.mock_rr_client.delete)
self.event_management.delete_event_process_definition('an id')
self.mock_rr_client.delete.assert_called_once_with('an id')
def test_create_event_process(self):
"""
Test creating an event process
"""
# process_definition = ProcessDefinition(name='test')
mock = Mock()
mock.definition = 'a_definition'
mock.detail = 'event_process_detail'
mock_id = 'a_process_definition_id'
self.mock_rr_client.read = Mock()
self.mock_rr_client.read.return_value = mock
self.mock_rr_client.update = Mock()
self.mock_rr_client.update = mocksignature(self.mock_rr_client.update)
self.mock_rr_client.find_objects = Mock()
self.mock_rr_client.find_objects.return_value = ['stream_id_1'], 'obj_assoc_1'
self.mock_pd_client.schedule_process = Mock()
self.mock_pd_client.schedule_process.return_value = 'process_id'
self.mock_rr_client.create_association = mocksignature(self.mock_rr_client.create_association)
self.mock_dams_client.register_event_process = mocksignature(self.mock_dams_client.register_event_process)
pid = self.event_management.create_event_process(process_definition_id=mock_id,
event_types=['type_1', 'type_2'],
sub_types=['subtype_1', 'subtype_2'],
origins=['or_1', 'or_2'],
origin_types=['t1', 't2'],
out_data_products={'conductivity': 'id1'}
)
# self.assertEquals(pid, 'process_id')
def test_read_event_process(self):
"""
Test reading an event process
"""
event_process = Mock()
self.mock_rr_client.read = mocksignature(self.mock_rr_client.read)
self.mock_rr_client.read.return_value = event_process
result = self.event_management.read_event_process('event_process_id')
self.assertEquals(result, event_process)
def test_delete_event_process(self):
"""
Test deleting an event process
"""
self.mock_rr_client.delete = mocksignature(self.mock_rr_client.delete)
self.event_management.delete_event_process('event_process_id')
self.mock_rr_client.delete.assert_called_once_with('event_process_id')
