class SpieTest(TestCase):
"""
Tests the dataqs.spie module. Since each processor is highly
dependent on a running GeoNode instance for most functions, only
independent functions are tested here.
"""
def setUp(self):
self.processor = SPEIProcessor()
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that files are downloaded.
"""
imgurl = "{}spei03.nc".format(self.processor.base_url)
httpretty.register_uri(httpretty.GET, imgurl, body=get_mock_image())
imgfile = self.processor.download(imgurl, 'spei03.tif')
self.assertTrue(
os.path.exists(os.path.join(self.processor.tmp_dir, imgfile)))
def test_cleanup(self):
"""
Temporary files should be gone after cleanup
:return:
"""
imgurl = "{}spei03.nc".format(self.processor.base_url)
httpretty.register_uri(httpretty.GET, imgurl, body=get_mock_image())
self.processor.download(imgurl, 'spei03.tif')
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 SpieTest(TestCase):
"""
Tests the dataqs.spie module. Since each processor is highly
dependent on a running GeoNode instance for most functions, only
independent functions are tested here.
"""
def setUp(self):
self.processor = SPEIProcessor()
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that files are downloaded.
"""
imgurl = "{}spei03.nc".format(self.processor.base_url)
httpretty.register_uri(httpretty.GET, imgurl,
body=get_mock_image())
imgfile = self.processor.download(imgurl, 'spei03.tif')
self.assertTrue(os.path.exists(os.path.join(
self.processor.tmp_dir, imgfile)))
def test_cleanup(self):
"""
Temporary files should be gone after cleanup
:return:
"""
imgurl = "{}spei03.nc".format(self.processor.base_url)
httpretty.register_uri(httpretty.GET, imgurl,
body=get_mock_image())
self.processor.download(imgurl, 'spei03.tif')
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):
self.processor = SPEIProcessor()
httpretty.enable()
def setUp(self):
self.processor = SPEIProcessor()
httpretty.enable()
def spei_task():
processor = SPEIProcessor()
processor.run()
