mirror of
https://github.com/getredash/redash.git
synced 2025-12-21 18:35:48 -05:00
2dff8b9a00
* Apply black formatting * Add auto formatting when committing to master * Update CONTRIBUTING.md re. Black & Prettier
32 lines
934 B
Python
32 lines
934 B
Python
from mock import patch
|
|
from tests import BaseTestCase
|
|
|
|
from redash.tasks import refresh_schemas
|
|
|
|
|
|
class TestRefreshSchemas(BaseTestCase):
|
|
def test_calls_refresh_of_all_data_sources(self):
|
|
self.factory.data_source # trigger creation
|
|
with patch(
|
|
"redash.tasks.queries.maintenance.refresh_schema.delay"
|
|
) as refresh_job:
|
|
refresh_schemas()
|
|
refresh_job.assert_called()
|
|
|
|
def test_skips_paused_data_sources(self):
|
|
self.factory.data_source.pause()
|
|
|
|
with patch(
|
|
"redash.tasks.queries.maintenance.refresh_schema.delay"
|
|
) as refresh_job:
|
|
refresh_schemas()
|
|
refresh_job.assert_not_called()
|
|
|
|
self.factory.data_source.resume()
|
|
|
|
with patch(
|
|
"redash.tasks.queries.maintenance.refresh_schema.delay"
|
|
) as refresh_job:
|
|
refresh_schemas()
|
|
refresh_job.assert_called()
|