X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/6642c1c71c5c6ce6ef3401c8c9da84cf076b018b..c6d792e73002970964923c0105b1f0c366d744fc:/tests/utils.py diff --git a/tests/utils.py b/tests/utils.py index 1870a07..fc87532 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,62 +1,23 @@ +# -*- coding: utf-8 -*- +# +# This file is part of Librarian, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# from __future__ import with_statement +from os.path import realpath, join, dirname +import glob -import os -from distutils.core import Command -from unittest import TextTestRunner, TestLoader -from glob import glob -from os.path import dirname, join, realpath, splitext, basename, walk -from os import listdir -import codecs -class AutoTestMetaclass(type): +def get_fixture_dir(dir_name): + """Returns path to fixtures directory dir_name.""" + return realpath(join(dirname(__file__), 'files', dir_name)) - def __new__(cls, name, bases, class_dict): - test_dir = class_dict.pop('TEST_DIR') - path = realpath( join(dirname(__file__), 'files', test_dir) ) - for file in listdir(path): - base, ext = splitext(file) - if ext != '.xml': - continue +def get_fixture(dir_name, file_name): + """Returns path to fixture file_name in directory dir_name.""" + return join(get_fixture_dir(dir_name), file_name) - class_dict['test_'+base] = cls.make_test_runner(base, \ - join(path, base +'.xml'), join(path, base + '.out') ) - - return type.__new__(cls, name, bases, class_dict) - - @staticmethod - def make_test_runner(name, inputf, outputf): - def runner(self): - with open(inputf, 'rb') as ifd: - with codecs.open(outputf, 'rb', encoding='utf-8') as ofd: - self.run_auto_test(ifd.read(), ofd.read()) - return runner - - -def get_file_path(dir_name, file_name): - return realpath(join(dirname(__file__), 'files', dir_name, file_name)) - -class TestCommand(Command): - user_options = [] - - def initialize_options(self): - self._dir = os.getcwd() - - def finalize_options(self): - pass - - def run(self): - ''' - Finds all the tests modules in tests/, and runs them. - ''' - testfiles = [] - for t in glob(join(self._dir, 'tests', '*.py')): - module_name = splitext(basename(t))[0] - if module_name.startswith('test'): - testfiles.append('.'.join(['tests', module_name]) - ) - - tests = TestLoader().loadTestsFromNames(testfiles) - t = TextTestRunner(verbosity=2) - t.run(tests) +def get_all_fixtures(dir_name, glob_pattern='*'): + """Returns list of paths for fixtures in directory dir_name matching the glob_pattern.""" + return [get_fixture(dir_name, file_name) for file_name in glob.glob(join(get_fixture_dir(dir_name), glob_pattern))]