aboutsummaryrefslogtreecommitdiffstats
path: root/tests/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/__init__.py')
-rw-r--r--tests/__init__.py43
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()])