1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
#!/usr/bin/env python
# Copyright (c) 2016 Cable Television Laboratories, Inc. and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
"""Define the parent class to run unittest.TestSuite as TestCase."""
from __future__ import division
import logging
import os
import shutil
import subprocess
import time
import unittest
from subunit.run import SubunitTestRunner
import six
from xtesting.core import testcase
__author__ = ("Steven Pisarski <s.pisarski@cablelabs.com>, "
"Cedric Ollivier <cedric.ollivier@orange.com>")
class Suite(testcase.TestCase):
"""Base model for running unittest.TestSuite."""
__logger = logging.getLogger(__name__)
def __init__(self, **kwargs):
super(Suite, self).__init__(**kwargs)
self.suite = None
@classmethod
def generate_stats(cls, stream):
"""Generate stats from subunit stream
Raises:
Exception
"""
stream.seek(0)
stats = subprocess.Popen(
['subunit-stats'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
output, _ = stats.communicate(stream.read())
cls.__logger.info("\n\n%s", output.decode("utf-8"))
def generate_xunit(self, stream):
"""Generate junit report from subunit stream
Raises:
Exception
"""
stream.seek(0)
with open("{}/results.xml".format(self.res_dir), "w") as xml:
stats = subprocess.Popen(
['subunit2junitxml'], stdin=subprocess.PIPE,
stdout=subprocess.PIPE)
output, _ = stats.communicate(stream.read())
xml.write(output.decode("utf-8"))
def generate_html(self, stream):
"""Generate html report from subunit stream
Raises:
CalledProcessError
"""
cmd = ['subunit2html', stream, '{}/results.html'.format(self.res_dir)]
output = subprocess.check_output(cmd)
self.__logger.debug("\n%s\n\n%s", ' '.join(cmd), output)
def run(self, **kwargs):
"""Run the test suite.
It allows running any unittest.TestSuite and getting its
execution status.
By default, it runs the suite defined as instance attribute.
It can be overriden by passing name as arg. It must
conform with TestLoader.loadTestsFromName().
It sets the following attributes required to push the results
to DB:
* result,
* start_time,
* stop_time,
* details.
Args:
kwargs: Arbitrary keyword arguments.
Return:
TestCase.EX_OK if any TestSuite has been run
TestCase.EX_RUN_ERROR otherwise.
"""
try:
name = kwargs["name"]
try:
self.suite = unittest.TestLoader().loadTestsFromName(name)
except ImportError:
self.__logger.error("Can not import %s", name)
return testcase.TestCase.EX_RUN_ERROR
except KeyError:
pass
try:
assert self.suite
self.start_time = time.time()
if not os.path.isdir(self.res_dir):
os.makedirs(self.res_dir)
stream = six.BytesIO()
result = SubunitTestRunner(
stream=stream, verbosity=2).run(self.suite).decorated
self.generate_stats(stream)
self.generate_xunit(stream)
with open('{}/subunit_stream'.format(self.res_dir), 'wb') as subfd:
stream.seek(0)
shutil.copyfileobj(stream, subfd)
self.generate_html('{}/subunit_stream'.format(self.res_dir))
self.stop_time = time.time()
self.details = {
"testsRun": result.testsRun,
"failures": len(result.failures),
"errors": len(result.errors)}
self.result = 100 * (
(result.testsRun - (len(result.failures) +
len(result.errors))) /
result.testsRun)
return testcase.TestCase.EX_OK
except AssertionError:
self.__logger.error("No suite is defined")
return testcase.TestCase.EX_RUN_ERROR
except ZeroDivisionError:
self.__logger.error("No test has been run")
return testcase.TestCase.EX_RUN_ERROR
except Exception: # pylint: disable=broad-except
self.__logger.exception("something wrong occurs")
return testcase.TestCase.EX_RUN_ERROR
|
