with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_external_data_set('bad') ex = cm.exception self.assertEqual(ex.message, 'External Data Set bad does not exist') self.mock_read.assert_called_once_with('bad', '') # def test_register_instrument(self): # self.mock_read.return_value = self.instrument # self.mock_create.return_value = ('111', 'bla') # self.mock_create.return_value = ('222', 'bla') # self.mock_create_association.return_value = ['333', 1] # # self.data_acquisition_mgmt_service.register_instrument('111') utg = UnitTestGenerator(TestDataAcquisitionManagement, DataAcquisitionManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.ExternalDataProvider, "external_data_provider", {}) utg.add_resource_unittests(RT.DataSource, "data_source", {}) utg.add_resource_unittests(RT.DataSourceModel, "data_source_model", {}) utg.add_resource_unittests(RT.DataSourceAgent, "data_source_agent", {}) utg.add_resource_unittests(RT.DataSourceAgentInstance, "data_source_agent_instance", {}) utg.add_resource_unittests(RT.ExternalDataset, "external_dataset", {}) utg.add_resource_unittests(RT.ExternalDatasetAgent, "external_dataset_agent", {}) utg.add_resource_unittests(RT.ExternalDatasetAgentInstance, "external_dataset_agent_instance", {})
@unittest.skip("not working") def test_read_data_process(self): # setup self.resource_registry.find_associations.return_value = ([self.transform_id], "don't care") self.transform_object.data_process_definition_id = self.data_proc_def_id self.transform_object.in_subscription_id = self.in_product_A self.transform_object.out_data_product_id = self.out_product_A self.transform_management_service.read_transform.return_value = self.transform_object # test call dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(self.data_process_id) # verify results self.assertEqual(dpd_id, self.data_proc_def_id) self.assertEqual(in_id, self.in_product_A) self.assertEqual(out_id, self.out_product_A) self.resource_registry.find_associations.assert_called_once_with( self.data_process_id, PRED.hasTransform, "", None, False ) self.transform_management_service.read_transform.assert_called_once_with(self.transform_id) utg = UnitTestGenerator(Test_DataProcessManagementService_Unit, DataProcessManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.TransformFunction, "transform_function", {}) utg.add_resource_unittests(RT.DataProcessDefinition, "data_process_definition", {}) # utg.add_resource_unittests(RT.DataProcess, "data_process", {}) # utg.add_resource_unittests(RT.DataProcess, "data_process2", {})
dp_obj = IonObject(RT.DataProduct, name='DP_X', description='some existing dp') self.clients.resource_registry.find_resources.return_value = ([dp_obj], []) # test call result = self.data_product_management_service.find_data_products() # check results self.assertEqual(result, [dp_obj]) self.clients.resource_registry.find_resources.assert_called_once_with( RT.DataProduct, None, None, False) utg = UnitTestGenerator(TestDataProductManagementServiceUnit, DataProductManagementService) #utg.test_all_in_one(True) utg.add_resource_unittests(RT.DataProduct, "data_product", {}) utg.add_resource_unittests(RT.DataProductCollection, "data_product_collection", {}) #remove some tests that don't work delattr(TestDataProductManagementServiceUnit, "test_data_product_create_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_noname_d14a028") delattr(TestDataProductManagementServiceUnit,
self.instrument_mgmt_service = InstrumentManagementService() self.instrument_mgmt_service.clients = mock_clients # must call this manually self.instrument_mgmt_service.on_init() self.addCleanup(delattr, self, "instrument_mgmt_service") self.addCleanup(delattr, self, "mock_ionobj") #self.resource_impl_cleanup() log.debug("setUp complete") #def resource_impl_cleanup(self): #pass utg = UnitTestGenerator(TestInstrumentManagement, InstrumentManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.InstrumentAgentInstance, "instrument_agent_instance", {}) utg.add_resource_unittests(RT.InstrumentAgent, "instrument_agent", {"driver_module": "potato"}) utg.add_resource_unittests(RT.InstrumentDevice, "instrument_device", {"serial_number": "123", "firmware_version": "x"}) utg.add_resource_unittests(RT.InstrumentModel, "instrument_model") utg.add_resource_unittests(RT.PlatformAgentInstance, "platform_agent_instance", {}) utg.add_resource_unittests(RT.PlatformAgent, "platform_agent", {"description": "the big donut"}) utg.add_resource_unittests(RT.PlatformDevice, "platform_device", {"serial_number": "2345"}) utg.add_resource_unittests(RT.PlatformModel, "platform_model", {"description": "desc"}) utg.add_resource_unittests(RT.SensorDevice, "sensor_device", {"serial_number": "123"}) utg.add_resource_unittests(RT.SensorModel, "sensor_model")
#from pyon.core.exception import BadRequest, Conflict, Inconsistent, NotFound #import unittest from pyon.util.unit_test import PyonTestCase @attr('UNIT', group='sa') class TestObservatoryManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock('ion.services.sa.observatory.observatory_management_service.IonObject') #self.mock_ionobj = IonObject mock_clients = self._create_service_mock('observatory_management') self.observatory_mgmt_service = ObservatoryManagementService() self.observatory_mgmt_service.clients = mock_clients # must call this manually self.observatory_mgmt_service.on_init() utg = UnitTestGenerator(TestObservatoryManagement, ObservatoryManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.Deployment, "deployment") utg.add_resource_unittests(RT.Observatory, "observatory") utg.add_resource_unittests(RT.Subsite, "subsite") utg.add_resource_unittests(RT.PlatformSite, "platform_site") utg.add_resource_unittests(RT.InstrumentSite, "instrument_site")
# Data Product dp_obj = IonObject(RT.DataProduct, name='DP_X', description='some existing dp') self.clients.resource_registry.find_resources.return_value = ([dp_obj], []) # test call result = self.data_product_management_service.find_data_products() # check results self.assertEqual(result, [dp_obj]) self.clients.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, None, False) utg = UnitTestGenerator(TestDataProductManagementServiceUnit, DataProductManagementService) #utg.test_all_in_one(True) utg.add_resource_unittests(RT.DataProduct, "data_product", {}) utg.add_resource_unittests(RT.DataProductCollection, "data_product_collection", {}) #remove some tests that don't work delattr(TestDataProductManagementServiceUnit, "test_data_product_create_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_dupname_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_noname_d14a028") delattr(TestDataProductManagementServiceUnit, "test_data_product_collection_create_bad_wrongtype_d14a028")
self.transform_id ], "don't care") self.transform_object.data_process_definition_id = self.data_proc_def_id self.transform_object.in_subscription_id = self.in_product_A self.transform_object.out_data_product_id = self.out_product_A self.transform_management_service.read_transform.return_value = self.transform_object # test call dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process( self.data_process_id) # verify results self.assertEqual(dpd_id, self.data_proc_def_id) self.assertEqual(in_id, self.in_product_A) self.assertEqual(out_id, self.out_product_A) self.resource_registry.find_associations.assert_called_once_with( self.data_process_id, PRED.hasTransform, '', None, False) self.transform_management_service.read_transform.assert_called_once_with( self.transform_id) utg = UnitTestGenerator(Test_DataProcessManagementService_Unit, DataProcessManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.TransformFunction, "transform_function", {}) utg.add_resource_unittests(RT.DataProcessDefinition, "data_process_definition", {}) #utg.add_resource_unittests(RT.DataProcess, "data_process", {})
PRED.hasInputProduct, self.in_product_A, None) @unittest.skip('not working') def test_read_data_process(self): # setup self.resource_registry.find_associations.return_value = ([self.transform_id], "don't care") self.transform_object.data_process_definition_id = self.data_proc_def_id self.transform_object.in_subscription_id = self.in_product_A self.transform_object.out_data_product_id = self.out_product_A self.transform_management_service.read_transform.return_value = self.transform_object # test call dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process(self.data_process_id) # verify results self.assertEqual(dpd_id, self.data_proc_def_id) self.assertEqual(in_id, self.in_product_A) self.assertEqual(out_id, self.out_product_A) self.resource_registry.find_associations.assert_called_once_with(self.data_process_id, PRED.hasTransform, '', None, False) self.transform_management_service.read_transform.assert_called_once_with(self.transform_id) utg = UnitTestGenerator(Test_DataProcessManagementService_Unit, DataProcessManagementService) utg.test_all_in_one(True)
'bad') ex = cm.exception self.assertEqual(ex.message, 'External Data Set bad does not exist') self.mock_read.assert_called_once_with('bad', '') # def test_register_instrument(self): # self.mock_read.return_value = self.instrument # self.mock_create.return_value = ('111', 'bla') # self.mock_create.return_value = ('222', 'bla') # self.mock_create_association.return_value = ['333', 1] # # self.data_acquisition_mgmt_service.register_instrument('111') utg = UnitTestGenerator(TestDataAcquisitionManagement, DataAcquisitionManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.ExternalDataProvider, "external_data_provider", {}) utg.add_resource_unittests(RT.DataSource, "data_source", {}) utg.add_resource_unittests(RT.DataSourceModel, "data_source_model", {}) utg.add_resource_unittests(RT.DataSourceAgent, "data_source_agent", {}) utg.add_resource_unittests(RT.DataSourceAgentInstance, "data_source_agent_instance", {}) utg.add_resource_unittests(RT.ExternalDataset, "external_dataset", {}) utg.add_resource_unittests(RT.ExternalDatasetAgent, "external_dataset_agent", {}) utg.add_resource_unittests(RT.ExternalDatasetAgentInstance, "external_dataset_agent_instance", {})
@unittest.skip('not working') def test_read_data_process(self): # setup self.resource_registry.find_associations.return_value = ([ self.transform_id ], "don't care") self.transform_object.data_process_definition_id = self.data_proc_def_id self.transform_object.in_subscription_id = self.in_product_A self.transform_object.out_data_product_id = self.out_product_A self.transform_management_service.read_transform.return_value = self.transform_object # test call dpd_id, in_id, out_id = self.data_process_mgmt_service.read_data_process( self.data_process_id) # verify results self.assertEqual(dpd_id, self.data_proc_def_id) self.assertEqual(in_id, self.in_product_A) self.assertEqual(out_id, self.out_product_A) self.resource_registry.find_associations.assert_called_once_with( self.data_process_id, PRED.hasTransform, '', None, False) self.transform_management_service.read_transform.assert_called_once_with( self.transform_id) utg = UnitTestGenerator(Test_DataProcessManagementService_Unit, DataProcessManagementService) utg.test_all_in_one(True)
self.instrument_mgmt_service.clients = mock_clients # must call this manually self.instrument_mgmt_service.on_init() self.addCleanup(delattr, self, "instrument_mgmt_service") self.addCleanup(delattr, self, "mock_ionobj") #self.resource_impl_cleanup() log.debug("setUp complete") #def resource_impl_cleanup(self): #pass utg = UnitTestGenerator(TestInstrumentManagement, InstrumentManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.InstrumentAgentInstance, "instrument_agent_instance", {}) utg.add_resource_unittests(RT.InstrumentAgent, "instrument_agent", {"driver_module": "potato"}) utg.add_resource_unittests(RT.InstrumentDevice, "instrument_device", { "serial_number": "123", "firmware_version": "x" }) utg.add_resource_unittests(RT.InstrumentModel, "instrument_model") utg.add_resource_unittests(RT.PlatformAgentInstance, "platform_agent_instance", {}) utg.add_resource_unittests(RT.PlatformAgent, "platform_agent",
from pyon.util.unit_test import PyonTestCase @attr('UNIT', group='sa') class TestObservatoryManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock( 'ion.services.sa.observatory.observatory_management_service.IonObject' ) #self.mock_ionobj = IonObject mock_clients = self._create_service_mock('observatory_management') self.observatory_mgmt_service = ObservatoryManagementService() self.observatory_mgmt_service.clients = mock_clients # must call this manually self.observatory_mgmt_service.on_init() utg = UnitTestGenerator(TestObservatoryManagement, ObservatoryManagementService) utg.test_all_in_one(True) utg.add_resource_unittests(RT.Deployment, "deployment") utg.add_resource_unittests(RT.Observatory, "observatory") utg.add_resource_unittests(RT.Subsite, "subsite") utg.add_resource_unittests(RT.PlatformSite, "platform_site") utg.add_resource_unittests(RT.InstrumentSite, "instrument_site")