def test_get_agent_definition(self):
loader = IONLoader()
ooi_loader = OOILoader(None, asset_path='res/preload/r2_ioc/ooi_assets', mapping_path=OOI_MAPPING_DOC)
loader.ooi_loader = ooi_loader
loader.ooi_loader.extract_ooi_assets()
inst_objs = ooi_loader.get_type_assets("instrument")
node_objs = ooi_loader.get_type_assets("node")
loader._get_resource_obj = Mock(return_value=IonObject(RT.ExternalDatasetAgent))
# for ooi_id in sorted(inst_objs):
# ooi_rd = OOIReferenceDesignator(ooi_id)
# agent_id, agent_obj = loader._get_agent_definition(ooi_rd)
# log.info("RD: %s, agent_id: %s", ooi_id, agent_id)
checks = [
# Check some mapping override cases
("CP01CNSM-MF004-03-DOSTAD999", "DART_DOSTA_D_CSTL"),
("CP01CNSM-RI003-05-FLORTD999", "DART_FLORT_D_CSTL"),
("CP02PMUO-RI001-01-ADCPSL999", "DART_ADCPS_L_CSTL"),
# Check some default cases
("GA03FLMB-RI001-03-DOSTAD999", "DART_DOSTA_D"),
("GA03FLMB-RI001-01-FLORTD999", "DART_FLORT_D"),
("GA03FLMB-RI001-04-ADCPSL999", "DART_ADCPS_L"),
# Check some cases without mapping (using default)
("GI05MOAS-GL001-01-FLORDM999", "DART_FLORD_M"),
]
for ooi_id, expected_agent_id in checks:
ooi_rd = OOIReferenceDesignator(ooi_id)
agent_id, agent_obj = loader._get_agent_definition(ooi_rd)
self.assertEquals(agent_id, expected_agent_id)
def test_ooi_loader(self):
self.ooi_loader = OOILoader(None,
asset_path='res/preload/r2_ioc/ooi_assets',
mapping_path=OOI_MAPPING_DOC)
self.ooi_loader.extract_ooi_assets()
ooiuntil = datetime.datetime.strptime("12/31/2013", "%m/%d/%Y")
self.ooi_loader.analyze_ooi_assets(ooiuntil)
# Test report
self.ooi_loader.report_ooi_assets()
def test_ooi_loader(self):
self.ooi_loader = OOILoader(None,
asset_path='res/preload/r2_ioc/ooi_assets',
mapping_path=OOI_MAPPING_DOC)
self.ooi_loader.extract_ooi_assets()
ooiuntil = datetime.datetime.strptime("12/31/2013", "%m/%d/%Y")
self.ooi_loader.analyze_ooi_assets(ooiuntil)
# Test report
self.ooi_loader.report_ooi_assets()
def test_get_agent_definition(self):
loader = IONLoader()
ooi_loader = OOILoader(None,
asset_path='res/preload/r2_ioc/ooi_assets',
mapping_path=OOI_MAPPING_DOC)
loader.ooi_loader = ooi_loader
loader.ooi_loader.extract_ooi_assets()
inst_objs = ooi_loader.get_type_assets("instrument")
node_objs = ooi_loader.get_type_assets("node")
loader._get_resource_obj = Mock(
return_value=IonObject(RT.ExternalDatasetAgent))
# for ooi_id in sorted(inst_objs):
# ooi_rd = OOIReferenceDesignator(ooi_id)
# agent_id, agent_obj = loader._get_agent_definition(ooi_rd)
# log.info("RD: %s, agent_id: %s", ooi_id, agent_id)
checks = [
# Check some mapping override cases
("CP01CNSM-MF004-03-DOSTAD999", "DART_DOSTA_D_CSTL"),
("CP01CNSM-RI003-05-FLORTD999", "DART_FLORT_D_CSTL"),
("CP02PMUO-RI001-01-ADCPSL999", "DART_ADCPS_L_CSTL"),
# Check some default cases
("GA03FLMB-RI001-03-DOSTAD999", "DART_DOSTA_D"),
("GA03FLMB-RI001-01-FLORTD999", "DART_FLORT_D"),
("GA03FLMB-RI001-04-ADCPSL999", "DART_ADCPS_L"),
# Check some cases without mapping (using default)
("GI05MOAS-GL001-01-FLORDM999", "DART_FLORD_M"),
]
for ooi_id, expected_agent_id in checks:
ooi_rd = OOIReferenceDesignator(ooi_id)
agent_id, agent_obj = loader._get_agent_definition(ooi_rd)
self.assertEquals(agent_id, expected_agent_id)
