class UsgsQuakesTest(TestCase):
"""
Tests the dataqs.usgs_quakes module. Since each processor is highly
dependent on a running GeoNode instance for most functions, only
independent functions are tested here.
"""
def setUp(self):
today = datetime.datetime.utcnow()
self.edate = today.strftime("%Y-%m-%d")
self.sdate = (today - datetime.timedelta(days=7)).strftime("%Y-%m-%d")
self.processor = USGSQuakeProcessor(edate=self.edate, sdate=self.sdate)
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that files are downloaded.
"""
with open(os.path.join(script_dir,
'resources/test_quakes.json')) as inf:
response = inf.read()
dl_url = self.processor.base_url.format(self.processor.params['sdate'],
self.processor.params['edate'])
httpretty.register_uri(httpretty.GET, dl_url, body=response)
jsonfile = self.processor.download(dl_url,
filename=self.processor.prefix +
'.rss')
jsonpath = os.path.join(self.processor.tmp_dir, jsonfile)
with open(jsonpath) as json_in:
quakejson = json.load(json_in)
self.assertTrue("features" in quakejson)
def test_cleanup(self):
"""
Temporary files should be gone after cleanup
:return:
"""
with open(os.path.join(script_dir,
'resources/test_quakes.json')) as inf:
response = inf.read()
dl_url = self.processor.base_url.format(self.processor.params['sdate'],
self.processor.params['edate'])
httpretty.register_uri(httpretty.GET, dl_url, body=response)
self.processor.download(dl_url,
filename=self.processor.prefix + '.rss')
self.assertNotEqual([],
glob.glob(
os.path.join(self.processor.tmp_dir,
self.processor.prefix + '*')))
self.processor.cleanup()
self.assertEquals([],
glob.glob(
os.path.join(self.processor.tmp_dir,
self.processor.prefix + '*')))
class UsgsQuakesTest(TestCase):
"""
Tests the dataqs.usgs_quakes module. Since each processor is highly
dependent on a running GeoNode instance for most functions, only
independent functions are tested here.
"""
def setUp(self):
today = datetime.datetime.utcnow()
self.edate = today.strftime("%Y-%m-%d")
self.sdate = (today - datetime.timedelta(days=7)).strftime("%Y-%m-%d")
self.processor = USGSQuakeProcessor(edate=self.edate, sdate=self.sdate)
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that files are downloaded.
"""
with open(os.path.join(
script_dir, 'resources/test_quakes.json')) as inf:
response = inf.read()
dl_url = self.processor.base_url.format(
self.processor.params['sdate'], self.processor.params['edate'])
httpretty.register_uri(httpretty.GET, dl_url,
body=response)
jsonfile = self.processor.download(
dl_url, filename=self.processor.prefix + '.rss')
jsonpath = os.path.join(
self.processor.tmp_dir, jsonfile)
with open(jsonpath) as json_in:
quakejson = json.load(json_in)
self.assertTrue("features" in quakejson)
def test_cleanup(self):
"""
Temporary files should be gone after cleanup
:return:
"""
with open(os.path.join(
script_dir, 'resources/test_quakes.json')) as inf:
response = inf.read()
dl_url = self.processor.base_url.format(
self.processor.params['sdate'], self.processor.params['edate'])
httpretty.register_uri(httpretty.GET, dl_url,
body=response)
self.processor.download(dl_url, filename=self.processor.prefix + '.rss')
self.assertNotEqual([], glob.glob(os.path.join(
self.processor.tmp_dir, self.processor.prefix + '*')))
self.processor.cleanup()
self.assertEquals([], glob.glob(os.path.join(
self.processor.tmp_dir, self.processor.prefix + '*')))
def setUp(self):
today = datetime.datetime.utcnow()
self.edate = today.strftime("%Y-%m-%d")
self.sdate = (today - datetime.timedelta(days=7)).strftime("%Y-%m-%d")
self.processor = USGSQuakeProcessor(edate=self.edate, sdate=self.sdate)
httpretty.enable()
def usgs_quake_task():
processor = USGSQuakeProcessor()
processor.run()
def setUp(self):
today = datetime.datetime.utcnow()
self.edate = today.strftime("%Y-%m-%d")
self.sdate = (today - datetime.timedelta(days=7)).strftime("%Y-%m-%d")
self.processor = USGSQuakeProcessor(edate=self.edate, sdate=self.sdate)
httpretty.enable()