|
| 1 | +import os |
| 2 | +import re |
| 3 | +import sublime |
| 4 | +import sublime_plugin |
| 5 | + |
| 6 | +from collections import ChainMap |
| 7 | +from fnmatch import fnmatch |
| 8 | +from glob import glob |
| 9 | + |
| 10 | +from .utils import OutputPanel |
| 11 | + |
| 12 | +DEFAULT_SETTINGS = { |
| 13 | + # input |
| 14 | + "tests_dir": "tests", |
| 15 | + "pattern": "test*.py", |
| 16 | + # runner |
| 17 | + "async": False, # deprecated |
| 18 | + "deferred": True, |
| 19 | + "condition_timeout": 4000, |
| 20 | + "failfast": False, |
| 21 | + # output |
| 22 | + "output": None, |
| 23 | + "verbosity": 2, |
| 24 | + "capture_console": False, |
| 25 | + # reloader |
| 26 | + "reload_package_on_testing": True, |
| 27 | + "show_reload_progress": False, |
| 28 | + # coverage |
| 29 | + "coverage": False, |
| 30 | + "start_coverage_after_reload": False, |
| 31 | + "coverage_on_worker_thread": False, # experimental |
| 32 | + "generate_html_report": False, |
| 33 | + "generate_xml_report": False, |
| 34 | +} |
| 35 | + |
| 36 | +DONE_MESSAGE = "UnitTesting: Done.\n" |
| 37 | + |
| 38 | + |
| 39 | +def casedpath(path): |
| 40 | + # path on Windows may not be properly cased |
| 41 | + r = glob(re.sub(r"([^:/\\])(?=[/\\]|$)", r"[\1]", path)) |
| 42 | + return r and r[0] or path |
| 43 | + |
| 44 | + |
| 45 | +def relative_to_spp(path): |
| 46 | + spp = sublime.packages_path() |
| 47 | + spp_real = casedpath(os.path.realpath(spp)) |
| 48 | + for p in [path, casedpath(os.path.realpath(path))]: |
| 49 | + for sp in [spp, spp_real]: |
| 50 | + if p.startswith(sp + os.sep): |
| 51 | + return p[len(sp) :] |
| 52 | + return None |
| 53 | + |
| 54 | + |
| 55 | +class BaseUnittestingCommand(sublime_plugin.WindowCommand): |
| 56 | + def current_package_name(self): |
| 57 | + view = self.window.active_view() |
| 58 | + if view and view.file_name(): |
| 59 | + file_path = relative_to_spp(view.file_name()) |
| 60 | + if file_path and file_path.endswith(".py"): |
| 61 | + return file_path.split(os.sep)[1] |
| 62 | + |
| 63 | + folders = self.window.folders() |
| 64 | + if folders and len(folders) > 0: |
| 65 | + first_folder = relative_to_spp(folders[0]) |
| 66 | + if first_folder: |
| 67 | + return os.path.basename(first_folder) |
| 68 | + |
| 69 | + return None |
| 70 | + |
| 71 | + def current_test_file(self, pattern): |
| 72 | + view = self.window.active_view() |
| 73 | + if view: |
| 74 | + current_file = os.path.basename(view.file_name() or "") |
| 75 | + if current_file and fnmatch(current_file, pattern): |
| 76 | + self.window.settings().set("UnitTesting.last_test_file", current_file) |
| 77 | + return current_file |
| 78 | + |
| 79 | + return self.window.settings().get("UnitTesting.last_test_file") |
| 80 | + |
| 81 | + def load_stream(self, package, settings): |
| 82 | + output = settings["output"] |
| 83 | + if not output or output == "<panel>": |
| 84 | + output_panel = OutputPanel( |
| 85 | + self.window, "exec", file_regex=r'File "([^"]*)", line (\d+)' |
| 86 | + ) |
| 87 | + output_panel.show() |
| 88 | + return output_panel |
| 89 | + |
| 90 | + if not os.path.isabs(output): |
| 91 | + output = os.path.join(sublime.packages_path(), package, output) |
| 92 | + os.makedirs(os.path.dirname(output), exist_ok=True) |
| 93 | + return open(output, "w", encoding="utf-8") |
| 94 | + |
| 95 | + def load_unittesting_settings(self, package, options): |
| 96 | + file_name = os.path.join(sublime.packages_path(), package, "unittesting.json") |
| 97 | + |
| 98 | + try: |
| 99 | + with open(file_name, "r", encoding="utf-8") as fp: |
| 100 | + json_data = sublime.decode_value(fp.read()) |
| 101 | + if not isinstance(json_data, dict): |
| 102 | + raise ValueError("unittesting.json content must be an object!") |
| 103 | + except FileNotFoundError: |
| 104 | + json_data = {} |
| 105 | + except Exception as e: |
| 106 | + json_data = {} |
| 107 | + print("ERROR: Unable to load 'unittesting.json'\n ", str(e)) |
| 108 | + |
| 109 | + return ChainMap(options, json_data, DEFAULT_SETTINGS) |
0 commit comments