Django单元测试案例代码覆盖率统计 - 自定义test runner

如何在Django中编写单元测试案例，以及使用测试用的test_setting和test runner，见：django单元测试历险记

代码很好懂，不做什么解释了。如果需要拷过去，根据自己的需要修改一下，不是什么难事。

test_settings.py

COVERAGE_MODULES  =  [ ' testapp.models ' ,  ' testapp.lib ' ,  ' testapp.common ' ]

TEST_RUNNER   =   ' testapp.testrunner.test_runner_with_coverage '

testrunner.py

代码

import  os, shutil, sys, unittest

#  Look for coverage.py in __file__/lib as well as sys.path
sys.path  =  [os.path.join(os.path.dirname( __file__ ),  " lib " )]  +  sys.path

from  coverage  import  coverage
from  inspect  import  getmembers, ismodule

from  django.test.simple  import  run_tests as django_test_runner

from  django.conf  import  settings

def  get_all_coverage_modules(module_path):
     """
    Returns all possible modules to report coverage on, even if they
    aren't loaded.
     """
    app_path  =  module_path.split( ' . ' )
    app_package  =   __import__ (module_path, {}, {}, app_path[ - 1 ])
    app_dirpath  =  app_package. __path__ [ - 1 ]

    mod_list  =  []
     for  root, dirs, files  in  os.walk(app_dirpath):
        root_path  =  app_path  +  root[len(app_dirpath):].split(os.path.sep)[ 1 :]
         for  file  in  files:
             if  file.lower().endswith( ' .py ' ):
                mod_name  =  file[: - 3 ].lower()
                 try :
                    mod  =   __import__ ( ' . ' .join(root_path  +  [mod_name]), {}, {},
                        mod_name)
                 except  ImportError:
                     pass
                 else :
                    mod_list.append(mod)

     return  mod_list

def  test_runner_with_coverage(test_labels, verbosity = 1 , interactive = True, extra_tests = []):
     """ Custom test runner.  Follows the django.test.simple.run_tests() interface. """
     #  Start code coverage before anything else if necessary
    cov  =  None
     if  hasattr(settings,  ' COVERAGE_MODULES ' ):
        cov  =  coverage()
        cov.use_cache( 1 )  #  Do not cache any of the coverage.py stuff
         # cov.exclude('if __name__ == .__main__.:')
        cov.start()

    test_results  =  django_test_runner(test_labels, verbosity, interactive, extra_tests)

     #  Stop code coverage after tests have completed
     if  hasattr(settings,  ' COVERAGE_MODULES ' ):
        cov.stop()

         #  Print code metrics header
         print   ''
         print   ' ---------------------------------------------------------------------- '
         print   '  Unit Test Code Coverage Results '
         print   ' ---------------------------------------------------------------------- '

     #  Report code coverage metrics
     if  hasattr(settings,  ' COVERAGE_MODULES ' ):
        coverage_modules  =  []
         for  module  in  settings.COVERAGE_MODULES:
            coverage_modules.extend(get_all_coverage_modules(module))

        cov.report(coverage_modules, show_missing = 1 )
         # cov.html_report(directory='covhtml')
         # cov.combine()
        cov.save()

         #  Print code metrics footer
         print   ' ---------------------------------------------------------------------- '

     return  test_results