BackendTests.py 1.16 KB
import sys
import unittest
from datetime import datetime as ddatetime
from pprint import pprint

from web.run import retrieve_amda_netcdf


class BaseTestCase(unittest.TestCase):
    def setUp(self):
        # initialise app
        # self.app = create_app(TestConfig)

        # update flask context
        # self.app_context = self.app.app_context()
        # self.app_context.push()
        pass

    def tearDown(self):
        # if os.path.isfile(self.db_path):
        #     os.remove(self.db_path)
        # self.app_context.pop()
        pass

    def test_always_true(self):
        self.assertTrue(True)


class AmdaTestCase(BaseTestCase):

    def test_amda_retrieve(self):
        FILE_DATE_FMT = "%Y-%m-%d %H:%M:%S"
        started_at = ddatetime.strptime("2021-12-17 00:00:00", FILE_DATE_FMT)
        stopped_at = ddatetime.strptime("2021-12-18 00:00:00", FILE_DATE_FMT)
        amda_list = retrieve_amda_netcdf("earth", "omni_hour_all", started_at, stopped_at)
        self.assertEqual(1, len(amda_list))
        self.assertIn('amda-irap-omp-eu-ddservice-base-data-omni-hour-omni202107010-nc', amda_list[0])

    def test_projects(self):
        self.assertTrue(False)