class GISTEMPTest(TestCase):
"""
Tests the dataqs.gistemp 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 = GISTEMPProcessor()
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that a file is downloaded
"""
httpretty.register_uri(httpretty.GET,
self.processor.base_url,
body=get_mock_image())
imgfile = self.processor.download(
self.processor.base_url, '{}.nc'.format(self.processor.layer_name))
self.assertTrue(
os.path.exists(os.path.join(self.processor.tmp_dir, imgfile)))
def test_cleanup(self):
httpretty.register_uri(httpretty.GET,
self.processor.base_url,
body=get_mock_image())
self.processor.download(self.processor.base_url,
'{}.nc'.format(self.processor.layer_name))
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 GISTEMPTest(TestCase):
"""
Tests the dataqs.gistemp 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 = GISTEMPProcessor()
httpretty.enable()
def tearDown(self):
httpretty.disable()
self.processor.cleanup()
def test_download(self):
"""
Verify that a file is downloaded
"""
httpretty.register_uri(httpretty.GET,
self.processor.base_url,
body=get_mock_image())
imgfile = self.processor.download(
self.processor.base_url,
'{}.nc'.format(self.processor.layer_name))
self.assertTrue(os.path.exists(
os.path.join(self.processor.tmp_dir, imgfile)))
def test_cleanup(self):
httpretty.register_uri(httpretty.GET,
self.processor.base_url,
body=get_mock_image())
self.processor.download(self.processor.base_url,
'{}.nc'.format(self.processor.layer_name))
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 gistemp_task():
processor = GISTEMPProcessor()
processor.run()
def setUp(self):
self.processor = GISTEMPProcessor()
httpretty.enable()
def setUp(self):
self.processor = GISTEMPProcessor()
httpretty.enable()