Exemplo n.º 1
0
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 + '*')))
Exemplo n.º 2
0
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 + '*')))
Exemplo n.º 3
0
 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()
Exemplo n.º 4
0
def usgs_quake_task():
    processor = USGSQuakeProcessor()
    processor.run()
Exemplo n.º 5
0
 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()