mirror of
https://github.com/getredash/redash.git
synced 2026-03-22 19:00:09 -04:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
from redash import extensions
|
|
from redash.tasks import periodic_job_definitions
|
|
from tests import BaseTestCase
|
|
|
|
logger = logging.getLogger(__name__)
|
|
dummy_extension = "redash-dummy"
|
|
dummy_path = os.path.join(os.path.dirname(__file__), dummy_extension)
|
|
|
|
|
|
class TestExtensions(BaseTestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
sys.path.insert(0, dummy_path)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
sys.path.remove(dummy_path)
|
|
|
|
def test_working_extension(self):
|
|
self.assertIn("working_extension", extensions.extensions.keys())
|
|
self.assertEqual(
|
|
extensions.extensions.get("working_extension"), "extension loaded"
|
|
)
|
|
|
|
def test_assertive_extension(self):
|
|
self.assertNotIn("assertive_extension", extensions.extensions.keys())
|
|
|
|
def test_not_findable_extension(self):
|
|
self.assertNotIn("not_findable_extension", extensions.extensions.keys())
|
|
|
|
def test_not_importable_extension(self):
|
|
self.assertNotIn("not_importable_extension", extensions.extensions.keys())
|
|
|
|
def test_non_callable_extension(self):
|
|
self.assertNotIn("non_callable_extension", extensions.extensions.keys())
|
|
|
|
def test_dummy_periodic_task(self):
|
|
# need to load the periodic tasks manually since this isn't
|
|
# done automatically on test suite start but only part of
|
|
# the worker configuration
|
|
extensions.load_periodic_jobs(logger)
|
|
self.assertIn("dummy_periodic_job", extensions.periodic_jobs.keys())
|
|
|
|
def test_dummy_periodic_task_definitions(self):
|
|
jobs = periodic_job_definitions()
|
|
from redash_dummy import job_callback
|
|
self.assertIn(job_callback, [job.get("func", None) for job in jobs])
|