class TestWOF(unittest.TestCase):
"""
Tests wof response generation methods for validation against
the WaterML 1.0 schema.
"""
def setUp(self):
dao = TestDao()
self.wof_inst = WOF(dao, TEST_CONFIG_FILE)
waterml_schema_path = os.path.join(os.path.dirname(__file__),
'cuahsiTimeSeries_v1_0.xsd')
waterml_schema_doc = etree.parse(waterml_schema_path)
self.waterml_schema = etree.XMLSchema(waterml_schema_doc)
def response_to_StringIO(self, response, root_name):
"""
Converts WOF method response to StringIO object.
"""
response_output = StringIO.StringIO()
response.export(response_output,
0,
name_=root_name,
namespacedef_=NSDEF)
response_output.seek(0)
return response_output
def save_response(self, response, filename, root_name):
f = open(filename, 'w')
response.export(f, 0, name_=root_name, namespacedef_=NSDEF)
f.close()
def compare_output_to_known_xml(self, response_stringIO,
known_xml_filename):
"""
Make sure response output is as expected. This tests that a
response contains the same XML as in the known XML file.
"""
known_file_path = os.path.join(os.path.dirname(__file__), TEST_XML_DIR,
known_xml_filename)
# Use objectify and etree to remove whitespace from XML for comparison
response_tree = objectify.parse(response_stringIO)
known_tree = objectify.parse(known_file_path)
response_string = etree.tostring(response_tree)
known_string = etree.tostring(known_tree)
assert response_string == known_string
def test_get_all_sites(self):
response = self.wof_inst.create_get_site_response()
response_string = self.response_to_StringIO(response, 'sitesResponse')
self.compare_output_to_known_xml(response_string, 'get_all_sites.xml')
def test_get_one_site(self):
response = self.wof_inst.create_get_site_response('TEST:SITE_A')
response_string = self.response_to_StringIO(response, 'sitesResponse')
self.compare_output_to_known_xml(response_string, 'get_one_site.xml')
def test_get_multiple_sites(self):
response = self.wof_inst.create_get_site_response(
'TEST:SITE_A,TEST:SITE_B')
response_string = self.response_to_StringIO(response, 'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_multiple_sites.xml')
def test_get_all_variables(self):
response = self.wof_inst.create_get_variable_info_response()
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_all_variables.xml')
def test_get_one_variable(self):
response = self.wof_inst.create_get_variable_info_response(
'TESTVocab:Temp')
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_one_variable.xml')
def test_get_multiple_variables(self):
response = self.wof_inst.create_get_variable_info_response(
'TESTVocab:Flow,TESTVocab:TP')
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_multiple_variables.xml')
def test_get_site_info_novar(self):
response = self.wof_inst.create_get_site_info_response('TEST:SITE_A')
response_string = self.response_to_StringIO(response, 'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_siteinfo_novar.xml')
def test_get_site_info_withvar(self):
response = self.wof_inst.create_get_site_info_response(
'TEST:SITE_B', 'TESTVocab:TP')
response_string = self.response_to_StringIO(response, 'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_siteinfo_withvar.xml')
def test_get_values_encloses_por(self):
response = self.wof_inst.create_get_values_response(
'TEST:SITE_A',
'TESTVocab:Temp',
startDateTime='2007-03-05 00:00',
endDateTime='2007-05-06 00:00')
response_string = self.response_to_StringIO(response,
'timeSeriesResponse')
self.compare_output_to_known_xml(response_string,
'get_values_encloses_por.xml')
class TestWOF(unittest.TestCase):
"""
Tests wof response generation methods for validation against
the WaterML 1.0 schema.
"""
def setUp(self):
dao = TestDao()
self.wof_inst = WOF(dao, TEST_CONFIG_FILE)
waterml_schema_path = os.path.join(os.path.dirname(__file__),
'cuahsiTimeSeries_v1_0.xsd')
waterml_schema_doc = etree.parse(waterml_schema_path)
self.waterml_schema = etree.XMLSchema(waterml_schema_doc)
def response_to_StringIO(self, response, root_name):
"""
Converts WOF method response to StringIO object.
"""
response_output = StringIO.StringIO()
response.export(response_output, 0, name_=root_name,
namespacedef_=NSDEF)
response_output.seek(0)
return response_output
def save_response(self, response, filename, root_name):
f = open(filename, 'w')
response.export(f, 0, name_=root_name,
namespacedef_=NSDEF)
f.close()
def compare_output_to_known_xml(self,
response_stringIO,
known_xml_filename):
"""
Make sure response output is as expected. This tests that a
response contains the same XML as in the known XML file.
"""
known_file_path = os.path.join(os.path.dirname(__file__),
TEST_XML_DIR, known_xml_filename)
# Use objectify and etree to remove whitespace from XML for comparison
response_tree = objectify.parse(response_stringIO)
known_tree = objectify.parse(known_file_path)
response_string = etree.tostring(response_tree)
known_string = etree.tostring(known_tree)
assert response_string == known_string
def test_get_all_sites(self):
response = self.wof_inst.create_get_site_response()
response_string = self.response_to_StringIO(response,
'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_all_sites.xml')
def test_get_one_site(self):
response = self.wof_inst.create_get_site_response(
'TEST:SITE_A')
response_string = self.response_to_StringIO(response,
'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_one_site.xml')
def test_get_multiple_sites(self):
response = self.wof_inst.create_get_site_response(
'TEST:SITE_A,TEST:SITE_B')
response_string = self.response_to_StringIO(response,
'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_multiple_sites.xml')
def test_get_all_variables(self):
response = self.wof_inst.create_get_variable_info_response()
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_all_variables.xml')
def test_get_one_variable(self):
response = self.wof_inst.create_get_variable_info_response(
'TESTVocab:Temp')
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_one_variable.xml')
def test_get_multiple_variables(self):
response = self.wof_inst.create_get_variable_info_response(
'TESTVocab:Flow,TESTVocab:TP')
response_string = self.response_to_StringIO(response,
'variablesResponse')
self.compare_output_to_known_xml(response_string,
'get_multiple_variables.xml')
def test_get_site_info_novar(self):
response = self.wof_inst.create_get_site_info_response('TEST:SITE_A')
response_string = self.response_to_StringIO(response,
'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_siteinfo_novar.xml')
def test_get_site_info_withvar(self):
response = self.wof_inst.create_get_site_info_response(
'TEST:SITE_B',
'TESTVocab:TP')
response_string = self.response_to_StringIO(response,
'sitesResponse')
self.compare_output_to_known_xml(response_string,
'get_siteinfo_withvar.xml')
def test_get_values_encloses_por(self):
response = self.wof_inst.create_get_values_response(
'TEST:SITE_A',
'TESTVocab:Temp',
startDateTime='2007-03-05 00:00',
endDateTime='2007-05-06 00:00')
response_string = self.response_to_StringIO(response,
'timeSeriesResponse')
self.compare_output_to_known_xml(response_string,
'get_values_encloses_por.xml')