diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/__init__.py | 43 |
1 files changed, 38 insertions, 5 deletions
diff --git a/tests/__init__.py b/tests/__init__.py index 99eae44..26cc0d0 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -31,6 +31,15 @@ def relpath(path, start=test_dir): return test_dir return os.path.join(*rel_list) +def get_config(dir_name): + """ Get config for given directory name. """ + config = util.CustomConfigParser({'normalize': '0', + 'skip': '0', + 'input_ext': '.txt', + 'output_ext': '.html'}) + config.read(os.path.join(dir_name, 'test.cfg')) + return config + def get_section(file, config): """ Get name of config section for given file. """ filename = os.path.basename(file) @@ -99,11 +108,7 @@ class CheckSyntax(object): def TestSyntax(): for dir_name, sub_dirs, files in os.walk(test_dir): # Get dir specific config settings. - config = util.CustomConfigParser({'normalize': '0', - 'skip': '0', - 'input_ext': '.txt', - 'output_ext': '.html'}) - config.read(os.path.join(dir_name, 'test.cfg')) + config = get_config(dir_name) # Loop through files and generate tests. for file in files: root, ext = os.path.splitext(file) @@ -112,6 +117,34 @@ def TestSyntax(): check_syntax = CheckSyntax(description=relpath(path, test_dir)) yield check_syntax, path, config +def generate(file, config): + """ Write expected output file for given input. """ + cfg_section = get_section(file, config) + if config.get(cfg_section, 'skip'): + print 'Skipping:', file + return None + input_file = file + config.get(cfg_section, 'input_ext') + output_file = file + config.get(cfg_section, 'output_ext') + if not os.path.isfile(output_file) or \ + os.path.getmtime(output_file) < os.path.getmtime(input_file): + print 'Generating:', file + markdown.markdownFromFile(input=input_file, output=output_file, + encoding='utf-8', **get_args(file, config)) + else: + print 'Already up-to-date:', file + +def generate_all(): + """ Generate expected output for all outdated tests. """ + for dir_name, sub_dirs, files in os.walk(test_dir): + # Get dir specific config settings. + config = get_config(dir_name) + # Loop through files and generate tests. + for file in files: + root, ext = os.path.splitext(file) + if ext == config.get(get_section(file, config), 'input_ext'): + generate(os.path.join(dir_name, root), config) + + def run(): nose.main(addplugins=[HtmlOutput(), Markdown()]) |