def setUp(self):
test_db_path = os.path.join(os.path.dirname(__file__), "test_swis2.db")
test_config_path = os.path.join(os.path.dirname(__file__), "test_swis_config.cfg")
self.dao = SwisDao(test_config_path, database_uri="sqlite:///" + test_db_path)
self.known_site_codes = [
"ARA",
"ARROYD",
"ARROYS",
"BAFF",
"BAYT",
"BIRD",
"BLB",
"BOBH",
"BOLI",
"BRAZOSD",
"BRAZOSS",
"BZ1U",
"BZ1UX",
"BZ2L",
"BZ2U",
"BZ3L",
"BZ3U",
"BZ4L",
"BZ4U",
"BZ5L",
"BZ5U",
"BZ6U",
"CANEY",
"CED1",
"CED2",
"CHKN",
"CONT",
"COP",
"COWT",
"DELT",
"DOLLAR",
"EAST",
"EEMAT",
"ELTORO",
"EMATC",
"EMATH",
"EMATT",
"FISH",
"FRES",
"FRPT",
"GIW1",
"GWSB1",
"GWSB2",
"HANN",
"ICFR",
"INGL",
"ISABEL",
"JARD",
"JDM1",
"JDM2",
"JDM3",
"JDM4",
"JFK",
"JOB",
"LAVC",
"LM-ARR",
"MANS",
"MATA",
"MCF1",
"MCF2",
"MES",
"MIDG",
"MIDSAB",
"MOSQ",
"NUE",
"NUECWN",
"NUECWS",
"NUELOW",
"NUEPP",
"NUERIV",
"NUEUP",
"OLDR",
"OSO",
"RED",
"RIOA",
"RIOF",
"SAB1",
"SAB2",
"SANT",
"SB1S",
"SB1W",
"SB2S",
"SB2W",
"SB3S",
"SB3W",
"SB5S",
"SB5W",
"SB6W",
"SBBP",
"SBR1",
"SBR2",
"SBR3",
"SBR4",
"SBR5",
"SBWS",
"SLNDCUT",
"SWBR",
"TRIN",
"UPBAFF",
"USAB",
]
self.fake_codes = ["junk", "trash", "fake"]
self.known_var_codes = [
"air_pressure",
"instrument_battery_voltage",
"water_specific_conductance",
"water_electrical_conductivity",
"water_dissolved_oxygen_concentration",
"water_dissolved_oxygen_percent_saturation",
"water_ph",
"seawater_salinity",
"water_temperature",
"air_temperature",
"water_total_dissolved_salts",
"water_turbidity",
"water_depth_non_vented",
"water_depth_vented",
"northward_water_velocity",
"eastward_water_velocity",
"upward_water_velocity",
"water_x_velocity",
"water_y_velocity",
]
# Need more test data, only JOB and BAYT have datavalues
# associated with them
self.known_series = dict(
BAYT=[
"instrument_battery_voltage",
"water_electrical_conductivity",
"water_dissolved_oxygen_percent_saturation",
"seawater_salinity",
"water_temperature",
"water_depth_non_vented",
],
JOB=["water_dissolved_oxygen_percent_saturation"],
)
# TODO: Need more instruments in test_swis.db
self.known_method_ids = [1]
# Only have one Source in SWIS DAO, whose info is in dao.contact_info
self.known_source_id = 1
self.known_source_info = {
"Name": "Andrew Wilson",
"Phone": "512-555-5555",
"Email": "*****@*****.**",
"Organization": "TWDB",
"Link": "http://www.twdb.state.tx.us/",
"Description": "Texas Water Development Board Surface Water Information System (TWDB SWIS)",
"Address": "1700 North Congress, Suite #670",
"City": "Austin",
"State": "TX",
"ZipCode": "78701",
}
self.known_qualifier_ids = ()
self.known_quality_control_ids = ()
self.known_offsettype_ids = ()
class TestSwisDao(unittest.TestCase):
def setUp(self):
test_db_path = os.path.join(os.path.dirname(__file__), "test_swis2.db")
test_config_path = os.path.join(os.path.dirname(__file__), "test_swis_config.cfg")
self.dao = SwisDao(test_config_path, database_uri="sqlite:///" + test_db_path)
self.known_site_codes = [
"ARA",
"ARROYD",
"ARROYS",
"BAFF",
"BAYT",
"BIRD",
"BLB",
"BOBH",
"BOLI",
"BRAZOSD",
"BRAZOSS",
"BZ1U",
"BZ1UX",
"BZ2L",
"BZ2U",
"BZ3L",
"BZ3U",
"BZ4L",
"BZ4U",
"BZ5L",
"BZ5U",
"BZ6U",
"CANEY",
"CED1",
"CED2",
"CHKN",
"CONT",
"COP",
"COWT",
"DELT",
"DOLLAR",
"EAST",
"EEMAT",
"ELTORO",
"EMATC",
"EMATH",
"EMATT",
"FISH",
"FRES",
"FRPT",
"GIW1",
"GWSB1",
"GWSB2",
"HANN",
"ICFR",
"INGL",
"ISABEL",
"JARD",
"JDM1",
"JDM2",
"JDM3",
"JDM4",
"JFK",
"JOB",
"LAVC",
"LM-ARR",
"MANS",
"MATA",
"MCF1",
"MCF2",
"MES",
"MIDG",
"MIDSAB",
"MOSQ",
"NUE",
"NUECWN",
"NUECWS",
"NUELOW",
"NUEPP",
"NUERIV",
"NUEUP",
"OLDR",
"OSO",
"RED",
"RIOA",
"RIOF",
"SAB1",
"SAB2",
"SANT",
"SB1S",
"SB1W",
"SB2S",
"SB2W",
"SB3S",
"SB3W",
"SB5S",
"SB5W",
"SB6W",
"SBBP",
"SBR1",
"SBR2",
"SBR3",
"SBR4",
"SBR5",
"SBWS",
"SLNDCUT",
"SWBR",
"TRIN",
"UPBAFF",
"USAB",
]
self.fake_codes = ["junk", "trash", "fake"]
self.known_var_codes = [
"air_pressure",
"instrument_battery_voltage",
"water_specific_conductance",
"water_electrical_conductivity",
"water_dissolved_oxygen_concentration",
"water_dissolved_oxygen_percent_saturation",
"water_ph",
"seawater_salinity",
"water_temperature",
"air_temperature",
"water_total_dissolved_salts",
"water_turbidity",
"water_depth_non_vented",
"water_depth_vented",
"northward_water_velocity",
"eastward_water_velocity",
"upward_water_velocity",
"water_x_velocity",
"water_y_velocity",
]
# Need more test data, only JOB and BAYT have datavalues
# associated with them
self.known_series = dict(
BAYT=[
"instrument_battery_voltage",
"water_electrical_conductivity",
"water_dissolved_oxygen_percent_saturation",
"seawater_salinity",
"water_temperature",
"water_depth_non_vented",
],
JOB=["water_dissolved_oxygen_percent_saturation"],
)
# TODO: Need more instruments in test_swis.db
self.known_method_ids = [1]
# Only have one Source in SWIS DAO, whose info is in dao.contact_info
self.known_source_id = 1
self.known_source_info = {
"Name": "Andrew Wilson",
"Phone": "512-555-5555",
"Email": "*****@*****.**",
"Organization": "TWDB",
"Link": "http://www.twdb.state.tx.us/",
"Description": "Texas Water Development Board Surface Water Information System (TWDB SWIS)",
"Address": "1700 North Congress, Suite #670",
"City": "Austin",
"State": "TX",
"ZipCode": "78701",
}
self.known_qualifier_ids = ()
self.known_quality_control_ids = ()
self.known_offsettype_ids = ()
def test_get_all_sites(self):
siteResultList = self.dao.get_all_sites()
resultSiteCodes = [s.SiteCode for s in siteResultList]
for known_code in self.known_site_codes:
self.assertTrue(known_code in resultSiteCodes)
def test_get_site_by_code(self):
for known_code in self.known_site_codes:
siteResult = self.dao.get_site_by_code(known_code)
self.assertEqual(known_code, siteResult.SiteCode)
def test_get_sites_by_codes(self):
siteResultList = self.dao.get_sites_by_codes(self.known_site_codes)
resultSiteCodes = [s.SiteCode for s in siteResultList]
for known_code in self.known_site_codes:
self.assertTrue(known_code in resultSiteCodes)
def test_get_all_variables(self):
varResultList = self.dao.get_all_variables()
resultVarCodes = [v.VariableCode for v in varResultList]
for known_code in self.known_var_codes:
self.assertTrue(known_code in resultVarCodes)
def test_get_var_by_code(self):
for known_code in self.known_var_codes:
varResult = self.dao.get_variable_by_code(known_code)
self.assertEqual(known_code, varResult.VariableCode)
def test_get_vars_by_codes(self):
varResultList = self.dao.get_variables_by_codes(self.known_var_codes)
resultVarCodes = [v.VariableCode for v in varResultList]
for known_code in self.known_var_codes:
self.assertTrue(known_code in resultVarCodes)
def test_get_series_by_sitecode(self):
for site_code in self.known_series:
seriesResultArr = self.dao.get_series_by_sitecode(site_code)
for series_cat in seriesResultArr:
self.assertTrue(series_cat.Site)
self.assertTrue(series_cat.Variable)
self.assertTrue(series_cat.Source)
self.assertEqual(site_code, series_cat.Site.SiteCode)
self.assertTrue(series_cat.ValueCount > 0)
def test_get_series_by_sitecode_and_varcode(self):
for site_code in self.known_series:
for var_code in self.known_series[site_code]:
seriesResultArr = self.dao.get_series_by_sitecode_and_varcode(site_code, var_code)
for series_cat in seriesResultArr:
self.assertTrue(series_cat.Site)
self.assertTrue(series_cat.Variable)
self.assertTrue(series_cat.Source)
self.assertEqual(site_code, series_cat.Site.SiteCode)
self.assertTrue(series_cat.ValueCount > 0)
self.assertEqual(var_code, series_cat.Variable.VariableCode)
def test_get_datavalues(self):
for site_code in self.known_series:
for var_code in self.known_series[site_code]:
dv = self.dao.get_datavalues(site_code, var_code)
self.assertNotEqual(dv, None)
self.assertNotEqual(len(dv), 0)
def test_get_method_by_id(self):
for method_id in self.known_method_ids:
methodResult = self.dao.get_method_by_id(method_id)
self.assertNotEqual(methodResult, None)
self.assertTrue(methodResult.MethodID in self.known_method_ids)
def test_get_methods_by_ids(self):
methodResultArr = self.dao.get_methods_by_ids(self.known_method_ids)
self.assertNotEqual(methodResultArr, None)
self.assertNotEqual(len(methodResultArr), 0)
for methodResult in methodResultArr:
self.assertTrue(methodResult.MethodID in self.known_method_ids)
def test_get_source_by_id(self):
# Only have one source in SWIS
sourceResult = self.dao.get_source_by_id(1)
self.assertNotEqual(sourceResult, None)
self.assertEqual(sourceResult.ContactName, self.known_source_info["Name"])
self.assertEqual(sourceResult.Phone, self.known_source_info["Phone"])
self.assertEqual(sourceResult.Email, self.known_source_info["Email"])
self.assertEqual(sourceResult.Organization, self.known_source_info["Organization"])
self.assertEqual(sourceResult.SourceLink, self.known_source_info["Link"])
self.assertEqual(sourceResult.SourceDescription, self.known_source_info["Description"])
self.assertEqual(sourceResult.Address, self.known_source_info["Address"])
self.assertEqual(sourceResult.City, self.known_source_info["City"])
self.assertEqual(sourceResult.State, self.known_source_info["State"])
self.assertEqual(sourceResult.ZipCode, self.known_source_info["ZipCode"])
def test_get_sources_by_ids(self):
sourceResultArr = self.dao.get_sources_by_ids([1])
# should only have one Source for SWIS
self.assertEqual(len(sourceResultArr), 1)
sourceResult = sourceResultArr[0]
self.assertEqual(sourceResult.ContactName, self.known_source_info["Name"])
self.assertEqual(sourceResult.Phone, self.known_source_info["Phone"])
self.assertEqual(sourceResult.Email, self.known_source_info["Email"])
self.assertEqual(sourceResult.Organization, self.known_source_info["Organization"])
self.assertEqual(sourceResult.SourceLink, self.known_source_info["Link"])
self.assertEqual(sourceResult.SourceDescription, self.known_source_info["Description"])
self.assertEqual(sourceResult.Address, self.known_source_info["Address"])
self.assertEqual(sourceResult.City, self.known_source_info["City"])
self.assertEqual(sourceResult.State, self.known_source_info["State"])
self.assertEqual(sourceResult.ZipCode, self.known_source_info["ZipCode"])
# TODO
def test_get_qualifier_by_id(self):
pass
# TODO
def test_get_qualifiers_by_ids(self):
pass
# TODO
def test_get_qualcontrol_by_id(self):
pass
# TODO
def test_get_qualcontrols_by_ids(self):
pass
# TODO
def test_get_offsettype_by_id(self):
pass
# TODO
def test_get_offsettypes_by_ids(self):
pass