def setUp(self): self.mock_ionobj = self._create_IonObject_mock( 'ion.services.sa.acquisition.data_acquisition_management_service.IonObject' ) mock_clients = self._create_service_mock('data_acquisition_management') self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = 'dsname' self.data_source.type = 'foo' self.data_source.description = 'data source desc' self.data_source.connection_params = {'param1': '111'} self.data_producer = Mock() self.data_producer.name = 'dproducer' self.data_producer.type = 'instrument' self.data_process = Mock() self.data_process.name = 'dprocess' self.data_process.description = 'process desc' self.instrument = Mock() self.instrument.name = 'inst' self.instrument.description = 'inst desc'
def setUp(self): self.mock_ionobj = self._create_IonObject_mock('ion.services.sa.acquisition.data_acquisition_management_service.IonObject') mock_clients = self._create_service_mock('data_acquisition_management') self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # must call this manually self.data_acquisition_mgmt_service.on_init() # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = 'dsname' self.data_source.type = 'foo' self.data_source.description = 'data source desc' self.data_source.connection_params = {'param1':'111'} self.data_producer = Mock() self.data_producer.name = 'dproducer' self.data_producer.type = 'instrument' self.data_process = Mock() self.data_process.name = 'dprocess' self.data_process.description = 'process desc' self.instrument = Mock() self.instrument.name = 'inst' self.instrument.description = 'inst desc'
def setUp(self): self.mock_ionobj = self._create_IonObject_mock( "ion.services.sa.acquisition.data_acquisition_management_service.IonObject" ) mock_clients = self._create_service_mock("data_acquisition_management") self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = "dsname" self.data_source.type = "foo" self.data_source.description = "data source desc" self.data_source.connection_params = {"param1": "111"} self.data_producer = Mock() self.data_producer.name = "dproducer" self.data_producer.type = "instrument" self.data_process = Mock() self.data_process.name = "dprocess" self.data_process.description = "process desc" self.instrument = Mock() self.instrument.name = "inst" self.instrument.description = "inst desc"
class TestDataAcquisitionManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock('ion.services.sa.acquisition.data_acquisition_management_service.IonObject') mock_clients = self._create_service_mock('data_acquisition_management') self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # must call this manually self.data_acquisition_mgmt_service.on_init() # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = 'dsname' self.data_source.type = 'foo' self.data_source.description = 'data source desc' self.data_source.connection_params = {'param1':'111'} self.data_producer = Mock() self.data_producer.name = 'dproducer' self.data_producer.type = 'instrument' self.data_process = Mock() self.data_process.name = 'dprocess' self.data_process.description = 'process desc' self.instrument = Mock() self.instrument.name = 'inst' self.instrument.description = 'inst desc' # def test_register_process(self): # self.mock_read.return_value = self.data_process # 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_process('111') #self.mock_read.assert_called_once_with('111', '') #self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_process_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_process('bad') ex = cm.exception self.assertEqual(ex.message, 'Data Process bad does not exist') self.mock_read.assert_called_once_with('bad', '') # def test_register_data_set(self): # self.mock_read.return_value = self.data_process # 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_external_data_set('111') #self.mock_read.assert_called_once_with('111', '') #self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call 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', '')
class TestDataAcquisitionManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock( "ion.services.sa.acquisition.data_acquisition_management_service.IonObject" ) mock_clients = self._create_service_mock("data_acquisition_management") self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = "dsname" self.data_source.type = "foo" self.data_source.description = "data source desc" self.data_source.connection_params = {"param1": "111"} self.data_producer = Mock() self.data_producer.name = "dproducer" self.data_producer.type = "instrument" self.data_process = Mock() self.data_process.name = "dprocess" self.data_process.description = "process desc" self.instrument = Mock() self.instrument.name = "inst" self.instrument.description = "inst desc" ############################################################################################## # # DataSource # ############################################################################################## def test_create_data_source(self): self.mock_create.return_value = ("111", "bla") data_source_id = self.data_acquisition_mgmt_service.create_data_source(self.data_source) self.mock_create.assert_called_once_with(self.data_source) self.assertEqual(data_source_id, "111") def test_read_and_update_data_source(self): self.mock_read.return_value = self.data_source dsrc = self.data_acquisition_mgmt_service.read_data_source("111") assert dsrc is self.mock_read.return_value self.mock_read.assert_called_once_with("111", "") dsrc.type = "Bar" self.mock_update.return_value = ["111", 2] self.data_acquisition_mgmt_service.update_data_source(dsrc) self.mock_update.assert_called_once_with(dsrc) # def test_register_process(self): # self.mock_read.return_value = self.data_process # 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_process('111') # self.mock_read.assert_called_once_with('111', '') # self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_process_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_process("bad") ex = cm.exception self.assertEqual(ex.message, "Data Process bad does not exist") self.mock_read.assert_called_once_with("bad", "") # def test_register_data_set(self): # self.mock_read.return_value = self.data_process # 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_external_data_set('111') # self.mock_read.assert_called_once_with('111', '') # self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call 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", "")
class TestDataAcquisitionManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock( 'ion.services.sa.acquisition.data_acquisition_management_service.IonObject' ) mock_clients = self._create_service_mock('data_acquisition_management') self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = 'dsname' self.data_source.type = 'foo' self.data_source.description = 'data source desc' self.data_source.connection_params = {'param1': '111'} self.data_producer = Mock() self.data_producer.name = 'dproducer' self.data_producer.type = 'instrument' self.data_process = Mock() self.data_process.name = 'dprocess' self.data_process.description = 'process desc' self.instrument = Mock() self.instrument.name = 'inst' self.instrument.description = 'inst desc' ############################################################################################## # # DataSource # ############################################################################################## def test_create_data_source(self): self.mock_create.return_value = ('111', 'bla') data_source_id = self.data_acquisition_mgmt_service.create_data_source( self.data_source) self.mock_create.assert_called_once_with(self.data_source) self.assertEqual(data_source_id, '111') def test_read_and_update_data_source(self): self.mock_read.return_value = self.data_source dsrc = self.data_acquisition_mgmt_service.read_data_source('111') assert dsrc is self.mock_read.return_value self.mock_read.assert_called_once_with('111', '') dsrc.type = 'Bar' self.mock_update.return_value = ['111', 2] self.data_acquisition_mgmt_service.update_data_source(dsrc) self.mock_update.assert_called_once_with(dsrc) def test_delete_data_source(self): self.mock_read.return_value = self.data_source self.data_acquisition_mgmt_service.delete_data_source('111') self.mock_delete.assert_called_once_with('111') def test_register_process(self): self.mock_read.return_value = self.data_process 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_process('111') #self.mock_read.assert_called_once_with('111', '') #self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_process_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_process('bad') ex = cm.exception self.assertEqual(ex.message, 'Data Process bad does not exist') self.mock_read.assert_called_once_with('bad', '') def test_register_data_set(self): self.mock_read.return_value = self.data_process 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_external_data_set('111') #self.mock_read.assert_called_once_with('111', '') #self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call 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') # self.mock_read.assert_called_once_with('111', '') # self.mock_create_association.assert_called_once_with('111', PRED.hasDataProducer, '222', None) def test_register_instrument_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_instrument('bad') ex = cm.exception self.assertEqual(ex.message, 'Instrument bad does not exist') self.mock_read.assert_called_once_with('bad', '')
class TestDataAcquisitionManagement(PyonTestCase): def setUp(self): self.mock_ionobj = self._create_IonObject_mock('ion.services.sa.acquisition.data_acquisition_management_service.IonObject') mock_clients = self._create_service_mock('data_acquisition_management') self.data_acquisition_mgmt_service = DataAcquisitionManagementService() self.data_acquisition_mgmt_service.clients = mock_clients # save some typing self.mock_create = mock_clients.resource_registry.create self.mock_update = mock_clients.resource_registry.update self.mock_delete = mock_clients.resource_registry.delete self.mock_read = mock_clients.resource_registry.read self.mock_create_association = mock_clients.resource_registry.create_association self.data_source = Mock() self.data_source.name = 'dsname' self.data_source.type = 'foo' self.data_source.description = 'data source desc' self.data_source.connection_params = {'param1':'111'} self.data_producer = Mock() self.data_producer.name = 'dproducer' self.data_producer.type = 'instrument' self.data_producer.stream_id = '123' self.data_process = Mock() self.data_process.name = 'dprocess' self.data_process.description = 'process desc' self.instrument = Mock() self.instrument.name = 'inst' self.instrument.description = 'inst desc' ############################################################################################## # # DataSource # ############################################################################################## def test_create_data_source(self): self.mock_create.return_value = ('111', 'bla') data_source_id = self.data_acquisition_mgmt_service.create_data_source(self.data_source) self.mock_create.assert_called_once_with(self.data_source) self.assertEqual(data_source_id, '111') def test_read_and_update_data_source(self): self.mock_read.return_value = self.data_source dsrc = self.data_acquisition_mgmt_service.read_data_source('111') assert dsrc is self.mock_read.return_value self.mock_read.assert_called_once_with('111', '') dsrc.type = 'Bar' self.mock_update.return_value = ['111', 2] self.data_acquisition_mgmt_service.update_data_source(dsrc) self.mock_update.assert_called_once_with(dsrc) def test_delete_data_source(self): self.mock_read.return_value = self.data_source self.data_acquisition_mgmt_service.delete_data_source('111') self.mock_read.assert_called_once_with('111', '') self.mock_delete.assert_called_once_with(self.data_source) def test_read_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.read_data_source('bad') ex = cm.exception self.assertEqual(ex.message, 'DataSource bad does not exist') self.mock_read.assert_called_once_with('bad', '') def test_delete_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.delete_data_source('bad') ex = cm.exception self.assertEqual(ex.message, 'DataSource bad does not exist') self.mock_read.assert_called_once_with('bad', '') ############################################################################################## # # DataProducer # ############################################################################################## def test_create_data_producer(self): self.mock_create.return_value = ('111', 'bla') data_prod_id = self.data_acquisition_mgmt_service.create_data_producer(self.data_producer) self.mock_create.assert_called_once_with(self.data_producer) self.assertEqual(data_prod_id, '111') def test_read_and_update_data_producer(self): self.mock_read.return_value = self.data_producer dp = self.data_acquisition_mgmt_service.read_data_producer('111') assert dp is self.mock_read.return_value self.mock_read.assert_called_once_with('111', '') dp.type = 'transform' self.mock_update.return_value = ['111', 2] self.data_acquisition_mgmt_service.update_data_producer(dp) self.mock_update.assert_called_once_with(dp) def test_delete_data_producer(self): self.mock_read.return_value = self.data_producer self.data_acquisition_mgmt_service.delete_data_producer('111') self.mock_read.assert_called_once_with('111', '') self.mock_delete.assert_called_once_with(self.data_producer) def test_read_data_producer_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.read_data_producer('bad') ex = cm.exception self.assertEqual(ex.message, 'Data producer bad does not exist') self.mock_read.assert_called_once_with('bad', '') def test_delete_data_producer_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.delete_data_producer('bad') ex = cm.exception self.assertEqual(ex.message, 'Data producer bad does not exist') self.mock_read.assert_called_once_with('bad', '') def test_register_process(self): self.mock_read.return_value = self.data_process 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_process('111') self.mock_read.assert_called_once_with('111', '') self.mock_create_association.assert_called_once_with('111', AT.hasDataProducer, '222', None) def test_register_process_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_process('bad') ex = cm.exception self.assertEqual(ex.message, 'Data Process bad does not exist') self.mock_read.assert_called_once_with('bad', '') def test_register_data_source(self): self.mock_read.return_value = self.data_source 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_data_source('111') self.mock_read.assert_called_once_with('111', '') self.mock_create_association.assert_called_once_with('111', AT.hasDataProducer, '222', None) def test_register_data_source_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_data_source('bad') ex = cm.exception self.assertEqual(ex.message, 'Data Source 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') self.mock_read.assert_called_once_with('111', '') self.mock_create_association.assert_called_once_with('111', AT.hasDataProducer, '222', None) def test_register_instrument_not_found(self): self.mock_read.return_value = None # TEST: Execute the service operation call with self.assertRaises(NotFound) as cm: self.data_acquisition_mgmt_service.register_instrument('bad') ex = cm.exception self.assertEqual(ex.message, 'Instrument bad does not exist') self.mock_read.assert_called_once_with('bad', '')