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()