mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-28 08:03:36 +00:00
105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
#!/usr/bin/python3
|
|
# -*- mode: python; mode: auto-fill; fill-column: 80 -*-
|
|
#
|
|
# This file is part of Plinth.
|
|
#
|
|
# Dervied from code sample at:
|
|
# http://jeetworks.org/adding-test-code-coverage-analysis-to-a-python-projects-setup-command/
|
|
#
|
|
# Copyright 2009 Jeet Sukumaran and Mark T. Holder.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
"""
|
|
Support for integration of code test coverage analysis with setuptools.
|
|
"""
|
|
|
|
import glob
|
|
import setuptools
|
|
import shutil
|
|
import time
|
|
import unittest
|
|
|
|
from plinth import tests
|
|
|
|
|
|
# Project directories with testable source code
|
|
SOURCE_DIRS = ['plinth'] + glob.glob('plinth/modules/*')
|
|
|
|
# Files to exclude from coverage analysis and reporting
|
|
FILES_TO_OMIT = ['plinth/tests/*.py']
|
|
|
|
# Location of coverage HTML report files
|
|
COVERAGE_REPORT_DIR = 'plinth/tests/coverage/report'
|
|
|
|
|
|
class TestCoverageCommand(setuptools.Command):
|
|
"""
|
|
Subclass of setuptools Command to perform code test coverage analysis.
|
|
"""
|
|
|
|
description = 'Run test coverage analysis'
|
|
user_options = [
|
|
('test-module=', 't', 'Explicitly specify a single module to test')
|
|
]
|
|
|
|
def initialize_options(self):
|
|
"""Initialize options to default values."""
|
|
self.test_module = None
|
|
|
|
def finalize_options(self):
|
|
pass
|
|
|
|
def run(self):
|
|
"""Main command implementation."""
|
|
if self.distribution.install_requires:
|
|
self.distribution.fetch_build_eggs(
|
|
self.distribution.install_requires)
|
|
|
|
if self.distribution.tests_require:
|
|
self.distribution.fetch_build_eggs(self.distribution.tests_require)
|
|
|
|
# Erase any existing HTML report files
|
|
try:
|
|
shutil.rmtree(COVERAGE_REPORT_DIR, True)
|
|
except:
|
|
pass
|
|
|
|
# Initialize a test suite for one or all modules
|
|
if self.test_module is None:
|
|
test_suite = tests.TEST_SUITE
|
|
else:
|
|
test = unittest.defaultTestLoader.loadTestsFromNames(
|
|
[self.test_module])
|
|
test_suite = unittest.TestSuite(test)
|
|
|
|
# Run the coverage analysis
|
|
runner = unittest.TextTestRunner()
|
|
import coverage
|
|
cov = coverage.coverage(auto_data=True, branch=True,
|
|
source=SOURCE_DIRS, omit=FILES_TO_OMIT)
|
|
cov.erase() # Erase existing coverage data file
|
|
cov.start()
|
|
runner.run(test_suite)
|
|
cov.stop()
|
|
|
|
# Generate an HTML report and print overall coverage
|
|
html_report_title = 'FreedomBox:Plinth -- Test Coverage as of ' + \
|
|
time.strftime('%x %X %Z')
|
|
_coverage = cov.html_report(directory=COVERAGE_REPORT_DIR,
|
|
omit=FILES_TO_OMIT,
|
|
title=html_report_title)
|
|
print('\nOverall test coverage: {0:.2f} %\n'.format(_coverage))
|