class NasaGpmTest(TestCase):
"""
Tests the dataqs.nasa_gpm 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 = GPMProcessor()
def tearDown(self):
self.processor.cleanup()
@patch('ftplib.FTP', autospec=True)
@patch('ftplib.FTP.retrbinary', mock_retrbinary)
@patch('ftplib.FTP.nlst', mock_nlst)
@patch('ftplib.FTP.connect', mock_none)
@patch('ftplib.FTP.login', mock_none)
@patch('ftplib.FTP.cwd', mock_none)
def test_download(self, mock_ftp):
"""
Verify that files are downloaded.
"""
today = datetime.datetime.utcnow()
imgfile = self.processor.download()[0]
self.assertTrue(os.path.exists(os.path.join(
self.processor.tmp_dir, imgfile)))
self.assertTrue('3B-HHR-E.MS.MRG.3IMERG.{}'.format(
datetime.datetime.strftime(today, '%Y%m%d')) in imgfile)
def test_parse_name(self):
"""
Layer title should contain date of image
:return:
"""
f = '3B-HHR-E.MS.MRG.3IMERG.20151027-S133000-E135959.0810.V03E.1day.tif'
title = self.processor.parse_name(f)[0]
self.assertTrue('NASA Global Precipitation Estimate (1day) - 2015-10-27'
in title)
@patch('ftplib.FTP', autospec=True)
@patch('ftplib.FTP.retrbinary', mock_retrbinary)
@patch('ftplib.FTP.nlst', mock_nlst)
@patch('ftplib.FTP.connect', mock_none)
@patch('ftplib.FTP.login', mock_none)
@patch('ftplib.FTP.cwd', mock_none)
def test_convert_image(self, mock_ftp):
"""
Converted image should be create in temp directory
:return:
"""
dl_tif = self.processor.download()[0]
convert_tif = self.processor.convert(dl_tif)
self.assertTrue(os.path.exists(os.path.join(
self.processor.tmp_dir, convert_tif)))
@patch('ftplib.FTP', autospec=True)
@patch('ftplib.FTP.retrbinary', mock_retrbinary)
@patch('ftplib.FTP.nlst', mock_nlst)
@patch('ftplib.FTP.connect', mock_none)
@patch('ftplib.FTP.login', mock_none)
@patch('ftplib.FTP.cwd', mock_none)
def test_cleanup(self, mock_ftp):
"""
Temporary files should be gone after cleanup
:return:
"""
dl_tif = self.processor.download()[0]
self.processor.convert(dl_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 = GPMProcessor()
def nasa_gpm_task():
processor = GPMProcessor()
processor.run()