def test_constructor(self):
"""Test create shake data."""
local_path = os.path.join(temp_dir('realtime-test'), 'shakemaps')
try:
event_one = ShakeData(working_dir=local_path)
event_two = ShakeData(working_dir=temp_dir('realtime-test'),
event=SHAKE_ID)
self.assertEqual(event_one.event_id, SHAKE_ID)
self.assertEqual(event_two.event_id, SHAKE_ID)
except:
raise
def test_get_list_event_ids(self):
"""Test get_list_event_ids."""
local_path = os.path.join(temp_dir('realtime-test'), 'shakemaps')
shake_data = ShakeData(working_dir=local_path)
list_id = shake_data.get_list_event_ids()
expected_list_id = [SHAKE_ID]
message = 'I got %s for the event ID in the server, Expectation %s' % (
list_id, expected_list_id)
self.assertEqual(list_id, expected_list_id, message)
def test_extract(self):
"""Test extracting data to be used in earth quake realtime."""
local_path = os.path.join(temp_dir('realtime-test'), 'shakemaps')
shake_data = ShakeData(working_dir=local_path)
shake_data.extract()
final_grid_xml_file = os.path.join(
shake_data.extract_dir(), 'grid.xml')
self.assertTrue(
os.path.exists(final_grid_xml_file), 'grid.xml not found')
def test_get_latest_event_id(self):
"""Test get latest event id."""
local_path = os.path.join(temp_dir('realtime-test'), 'shakemaps')
shake_data = ShakeData(working_dir=local_path)
latest_id = shake_data.get_latest_event_id()
# The latest event ID should be = SHAKE_ID since there's only one
expected_event_id = SHAKE_ID
message = 'I got %s for this latest event id, Expectation %s' % (
latest_id, expected_event_id)
self.assertEqual(expected_event_id, latest_id, message)
def test_validate_event(self):
"""Test validate_event works."""
local_path = os.path.join(temp_dir('realtime-test'), 'shakemaps')
event = ShakeData(working_dir=local_path)
self.assertTrue(event.validate_event())
