示例#1
0
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 + '*')))
示例#2
0
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 + '*')))
示例#3
0
def gistemp_task():
    processor = GISTEMPProcessor()
    processor.run()
示例#4
0
 def setUp(self):
     self.processor = GISTEMPProcessor()
     httpretty.enable()
示例#5
0
 def setUp(self):
     self.processor = GISTEMPProcessor()
     httpretty.enable()