8080old_create = None
8181
8282template_coverage_collection = None
83+ code_coverage_collection = None
8384url_coverage_collection = None
8485
8586
@@ -223,15 +224,18 @@ def save_test_results(failures, test_labels):
223224
224225def set_coverage_checking (flag = True ):
225226 global template_coverage_collection
227+ global code_coverage_collection
226228 global url_coverage_collection
227229 if settings .SERVER_MODE == 'test' :
228230 if flag :
229231 settings .TEST_CODE_COVERAGE_CHECKER .collector .resume ()
230232 template_coverage_collection = True
233+ code_coverage_collection = True
231234 url_coverage_collection = True
232235 else :
233236 settings .TEST_CODE_COVERAGE_CHECKER .collector .pause ()
234237 template_coverage_collection = False
238+ code_coverage_collection = False
235239 url_coverage_collection = False
236240
237241class CoverageReporter (Reporter ):
@@ -521,7 +525,6 @@ def setup_test_environment(self, **kwargs):
521525 settings .TEMPLATES [0 ]['OPTIONS' ]['loaders' ] = ('ietf.utils.test_runner.TemplateCoverageLoader' ,) + settings .TEMPLATES [0 ]['OPTIONS' ]['loaders' ]
522526
523527 settings .MIDDLEWARE = ('ietf.utils.test_runner.record_urls_middleware' ,) + tuple (settings .MIDDLEWARE )
524- url_coverage_collection = True
525528
526529 self .code_coverage_checker = settings .TEST_CODE_COVERAGE_CHECKER
527530 if not self .code_coverage_checker ._started :
@@ -628,7 +631,7 @@ def get_test_paths(self, test_labels):
628631 return test_apps , test_paths
629632
630633 def run_tests (self , test_labels , extra_tests = [], ** kwargs ):
631- global old_destroy , old_create , test_database_name , template_coverage_collection
634+ global old_destroy , old_create , test_database_name , template_coverage_collection , code_coverage_collection , url_coverage_collection
632635 from django .db import connection
633636 from ietf .doc .tests import TemplateTagTest
634637
@@ -652,6 +655,8 @@ def run_tests(self, test_labels, extra_tests=[], **kwargs):
652655
653656 if self .check_coverage :
654657 template_coverage_collection = True
658+ code_coverage_collection = True
659+ url_coverage_collection = True
655660 extra_tests += [
656661 CoverageTest (test_runner = self , methodName = 'interleaved_migrations_test' ),
657662 CoverageTest (test_runner = self , methodName = 'url_coverage_test' ),
0 commit comments