openaq_ingestion/test.py

import os
import shutil
import tempfile
import unittest

from openaq_ingestion.openaq_ingestion import process_queue, \
    settings, _initialize_db


class TestOpenaqIngestion(unittest.TestCase):
    """
    Test openaq_ingestion
    """
    def setUp(self):
        self.test_dir = tempfile.mkdtemp()
        settings['database_uri'] = 'sqlite:///' + \
            os.path.join(self.test_dir, 'db.sqlite')
        _initialize_db()

    def tearDown(self):
        shutil.rmtree(self.test_dir)

    def test_process_queue(self):
        process_queue()