class TestDataProductManagementServiceUnit(PyonTestCase):
def setUp(self):
self.clients = self._create_service_mock('data_product_management')
self.data_product_management_service = DataProductManagementService()
self.data_product_management_service.clients = self.clients
# must call this manually
self.data_product_management_service.on_init()
self.data_source = Mock()
self.data_source.name = 'data_source_name'
self.data_source.description = 'data source desc'
#@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_success(self):
# setup
self.resource_registry.find_resources.return_value = ([], 'do not care')
self.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.data_acquisition_management.assign_data_product.return_value = None
# Data Product
dpt_obj = IonObject(RT.DataProduct,
name='DPT_Y',
description='some new data product')
# test call
dp_id = self.data_product_management_service.create_data_product(dpt_obj, 'stream_def_id')
# check results
self.assertEqual(dp_id, 'SOME_RR_ID1')
self.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.pubsub_management.create_stream.assert_called_once_with('', True, 'stream_def_id', 'DPT_Y', 'some new data product', '')
self.resource_registry.create.assert_called_once_with(dpt_obj)
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_with_id_NotFound(self):
# setup
self.resource_registry.find_resources.return_value = ([], 'do not care')
self.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.pubsub_management.create_stream.return_value = 'stream1'
# Data Product
dpt_obj = IonObject(RT.DataProduct, name='DPT_X', description='some new data product')
# test call
with self.assertRaises(NotFound) as cm:
dp_id = self.data_product_management_service.create_data_product(dpt_obj, 'stream_def_id')
# check results
self.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.resource_registry.create.assert_called_once_with(dpt_obj)
#todo: what are errors to check in create stream?
def test_findDataProduct_success(self):
# setup
# Data Product
dp_obj = IonObject(RT.DataProduct,
name='DP_X',
description='some existing dp')
self.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.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, None, False)
class TestDataProductManagementServiceUnit(PyonTestCase):
def setUp(self):
self.clients = self._create_service_mock("data_product_management")
self.data_product_management_service = DataProductManagementService()
self.data_product_management_service.clients = self.clients
# must call this manually
self.data_product_management_service.on_init()
self.data_source = Mock()
self.data_source.name = "data_source_name"
self.data_source.description = "data source desc"
def test_createDataProduct_and_DataProducer_success(self):
# setup
self.resource_registry.find_resources.return_value = ([], "do not care")
self.resource_registry.create.return_value = ("SOME_RR_ID1", "Version_1")
self.data_acquisition_management.assign_data_product.return_value = None
# Data Product
dpt_obj = IonObject(RT.DataProduct, name="DPT_Y", description="some new data product")
# test call
dp_id = self.data_product_management_service.create_data_product(dpt_obj, "source_resource_id")
# check results
self.assertEqual(dp_id, "SOME_RR_ID1")
self.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.resource_registry.create.assert_called_once_with(dpt_obj)
# self.data_acquisition_management.assign_data_product.assert_called_once_with('source_resource_id', 'SOME_RR_ID1', True)
def test_createDataProduct_and_DataProducer_with_id_NotFound(self):
# setup
self.resource_registry.find_resources.return_value = ([], "do not care")
self.resource_registry.create.return_value = ("SOME_RR_ID1", "Version_1")
self.data_acquisition_management.assign_data_product.return_value = None
self.data_acquisition_management.assign_data_product.side_effect = NotFound(
"Object with id SOME_RR_ID1 does not exist."
)
# Data Product
dpt_obj = IonObject(RT.DataProduct, name="DPT_X", description="some new data product")
# test call
with self.assertRaises(NotFound) as cm:
dp_id = self.data_product_management_service.create_data_product(dpt_obj, "source_resource_id")
# check results
self.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.resource_registry.create.assert_called_once_with(dpt_obj)
# self.data_acquisition_management.assign_data_product.assert_called_once_with('source_resource_id', 'SOME_RR_ID1', True)
ex = cm.exception
self.assertEqual(ex.message, "Object with id SOME_RR_ID1 does not exist.")
def test_findDataProduct_success(self):
# setup
# Data Product
dp_obj = IonObject(RT.DataProduct, name="DP_X", description="some existing dp")
self.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.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, None, False)
class TestDataProductManagementServiceUnit(PyonTestCase):
def setUp(self):
self.clients = self._create_service_mock('data_product_management')
self.data_product_management_service = DataProductManagementService()
self.data_product_management_service.clients = self.clients
# must call this manually
self.data_product_management_service.on_init()
self.data_source = Mock()
self.data_source.name = 'data_source_name'
self.data_source.description = 'data source desc'
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_success(self):
# setup
self.clients.resource_registry.find_resources.return_value = ([], 'do not care')
self.clients.resource_registry.find_associations.return_value = []
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.clients.data_acquisition_management.assign_data_product.return_value = None
self.clients.pubsub_management.create_stream.return_value = "stream_id"
craft = CoverageCraft
sdom, tdom = craft.create_domains()
sdom = sdom.dump()
tdom = tdom.dump()
parameter_dictionary = craft.create_parameters()
parameter_dictionary = parameter_dictionary.dump()
dp_obj = IonObject(RT.DataProduct,
name='DP1',
description='some new dp',
temporal_domain = tdom,
spatial_domain = sdom)
# test call
dp_id = self.data_product_management_service.create_data_product(data_product=dp_obj,
stream_definition_id='a stream def id',
parameter_dictionary=parameter_dictionary)
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_with_id_NotFound(self):
# setup
self.clients.resource_registry.find_resources.return_value = ([], 'do not care')
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.clients.pubsub_management.create_stream.return_value = 'stream1'
# Data Product
dpt_obj = IonObject(RT.DataProduct, name='DPT_X', description='some new data product')
# test call
with self.assertRaises(NotFound) as cm:
dp_id = self.data_product_management_service.create_data_product(dpt_obj, 'stream_def_id')
# check results
self.clients.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.clients.resource_registry.create.assert_called_once_with(dpt_obj)
#todo: what are errors to check in create stream?
def test_findDataProduct_success(self):
# setup
# 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)
class TestDataProductManagementServiceUnit(PyonTestCase):
def setUp(self):
self.clients = self._create_service_mock('data_product_management')
self.data_product_management_service = DataProductManagementService()
self.data_product_management_service.clients = self.clients
# must call this manually
self.data_product_management_service.on_init()
self.data_source = Mock()
self.data_source.name = 'data_source_name'
self.data_source.description = 'data source desc'
def test_createDataProduct_and_DataProducer_success(self):
# setup
self.clients.resource_registry.find_resources.return_value = (
[], 'do not care')
self.clients.resource_registry.find_associations.return_value = []
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1',
'Version_1')
self.clients.data_acquisition_management.assign_data_product.return_value = None
self.clients.pubsub_management.create_stream.return_value = "stream_id", "route_id"
# Construct temporal and spatial Coordinate Reference System objects
tcrs = CRS([AxisTypeEnum.TIME])
scrs = CRS([AxisTypeEnum.LON, AxisTypeEnum.LAT])
# Construct temporal and spatial Domain objects
tdom = GridDomain(GridShape('temporal', [0]), tcrs,
MutabilityEnum.EXTENSIBLE) # 1d (timeline)
sdom = GridDomain(GridShape('spatial',
[0]), scrs, MutabilityEnum.IMMUTABLE
) # 1d spatial topology (station/trajectory)
sdom = sdom.dump()
tdom = tdom.dump()
dp_obj = IonObject(RT.DataProduct,
name='DP1',
description='some new dp',
temporal_domain=tdom,
spatial_domain=sdom)
# test call
dp_id = self.data_product_management_service.create_data_product(
data_product=dp_obj, stream_definition_id='a stream def id')
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_with_id_NotFound(self):
# setup
self.clients.resource_registry.find_resources.return_value = (
[], 'do not care')
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1',
'Version_1')
self.clients.pubsub_management.create_stream.return_value = 'stream1'
# Data Product
dpt_obj = IonObject(RT.DataProduct,
name='DPT_X',
description='some new data product')
# test call
with self.assertRaises(NotFound) as cm:
dp_id = self.data_product_management_service.create_data_product(
dpt_obj, 'stream_def_id')
# check results
self.clients.resource_registry.find_resources.assert_called_once_with(
RT.DataProduct, None, dpt_obj.name, True)
self.clients.resource_registry.create.assert_called_once_with(dpt_obj)
#todo: what are errors to check in create stream?
def test_findDataProduct_success(self):
# setup
# 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)
class TestDataProductManagementServiceUnit(PyonTestCase):
def setUp(self):
self.clients = self._create_service_mock('data_product_management')
self.data_product_management_service = DataProductManagementService()
self.data_product_management_service.clients = self.clients
# must call this manually
self.data_product_management_service.on_init()
self.data_source = Mock()
self.data_source.name = 'data_source_name'
self.data_source.description = 'data source desc'
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_success(self):
# setup
self.clients.resource_registry.find_resources.return_value = ([], 'do not care')
self.clients.resource_registry.find_associations.return_value = []
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.clients.data_acquisition_management.assign_data_product.return_value = None
self.clients.pubsub_management.create_stream.return_value = "stream_id"
# Construct temporal and spatial Coordinate Reference System objects
tcrs = CRS([AxisTypeEnum.TIME])
scrs = CRS([AxisTypeEnum.LON, AxisTypeEnum.LAT])
# Construct temporal and spatial Domain objects
tdom = GridDomain(GridShape('temporal', [0]), tcrs, MutabilityEnum.EXTENSIBLE) # 1d (timeline)
sdom = GridDomain(GridShape('spatial', [0]), scrs, MutabilityEnum.IMMUTABLE) # 1d spatial topology (station/trajectory)
sdom = sdom.dump()
tdom = tdom.dump()
#@TODO: DO NOT DO THIS, WHEN THIS TEST IS REWRITTEN GET RID OF THIS, IT WILL FAIL, thanks -Luke
parameter_dictionary = get_param_dict('ctd_parsed_param_dict')
parameter_dictionary = parameter_dictionary.dump()
dp_obj = IonObject(RT.DataProduct,
name='DP1',
description='some new dp',
temporal_domain = tdom,
spatial_domain = sdom)
# test call
dp_id = self.data_product_management_service.create_data_product(data_product=dp_obj,
stream_definition_id='a stream def id')
@unittest.skip('not working')
def test_createDataProduct_and_DataProducer_with_id_NotFound(self):
# setup
self.clients.resource_registry.find_resources.return_value = ([], 'do not care')
self.clients.resource_registry.create.return_value = ('SOME_RR_ID1', 'Version_1')
self.clients.pubsub_management.create_stream.return_value = 'stream1'
# Data Product
dpt_obj = IonObject(RT.DataProduct, name='DPT_X', description='some new data product')
# test call
with self.assertRaises(NotFound) as cm:
dp_id = self.data_product_management_service.create_data_product(dpt_obj, 'stream_def_id')
# check results
self.clients.resource_registry.find_resources.assert_called_once_with(RT.DataProduct, None, dpt_obj.name, True)
self.clients.resource_registry.create.assert_called_once_with(dpt_obj)
#todo: what are errors to check in create stream?
def test_findDataProduct_success(self):
# setup
# 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)
