runner.py 790 B

1234567891011121314151617181920
  1. from django.test.runner import DiscoverRunner
  2. class CustomOptionsTestRunner(DiscoverRunner):
  3. def __init__(self, verbosity=1, interactive=True, failfast=True,
  4. option_a=None, option_b=None, option_c=None, **kwargs):
  5. super().__init__(verbosity=verbosity, interactive=interactive, failfast=failfast)
  6. self.option_a = option_a
  7. self.option_b = option_b
  8. self.option_c = option_c
  9. @classmethod
  10. def add_arguments(cls, parser):
  11. parser.add_argument('--option_a', '-a', default='1'),
  12. parser.add_argument('--option_b', '-b', default='2'),
  13. parser.add_argument('--option_c', '-c', default='3'),
  14. def run_tests(self, test_labels, **kwargs):
  15. print("%s:%s:%s" % (self.option_a, self.option_b, self.option_c))