Files
redash/tests/query_runner/test_script.py
2018-06-21 14:45:12 +09:00

20 lines
620 B
Python

import os
from _pytest.monkeypatch import MonkeyPatch
from redash.query_runner.script import query_to_script_path
from tests import BaseTestCase
class TestQueryToScript(BaseTestCase):
monkeypatch = MonkeyPatch()
def test_unspecified(self):
self.assertEqual("/foo/bar/baz.sh", query_to_script_path("*", "/foo/bar/baz.sh"))
def test_specified(self):
self.assertRaises(IOError, lambda: query_to_script_path("/foo/bar", "baz.sh"))
self.monkeypatch.setattr(os.path, "exists", lambda x: True)
self.assertEqual(["/foo/bar/baz.sh"], query_to_script_path("/foo/bar", "baz.sh"))