mirror of
https://github.com/borgbackup/borg.git
synced 2024-12-26 01:37:20 +00:00
35 lines
961 B
Python
35 lines
961 B
Python
import unittest
|
|
|
|
|
|
class AtticTestCase(unittest.TestCase):
|
|
"""
|
|
"""
|
|
assert_equal = unittest.TestCase.assertEqual
|
|
assert_not_equal = unittest.TestCase.assertNotEqual
|
|
assert_raises = unittest.TestCase.assertRaises
|
|
|
|
|
|
def get_tests(suite):
|
|
"""Generates a sequence of tests from a test suite
|
|
"""
|
|
for item in suite:
|
|
try:
|
|
# TODO: This could be "yield from..." with Python 3.3+
|
|
for i in get_tests(item):
|
|
yield i
|
|
except TypeError:
|
|
yield item
|
|
|
|
|
|
class TestLoader(unittest.TestLoader):
|
|
"""A customzied test loader that properly detects and filters our test cases
|
|
"""
|
|
def loadTestsFromName(self, pattern, module=None):
|
|
suite = self.discover('attic.testsuite', '*.py')
|
|
tests = unittest.TestSuite()
|
|
for test in get_tests(suite):
|
|
if pattern.lower() in test.id().lower():
|
|
tests.addTest(test)
|
|
return tests
|
|
|
|
|