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
|
#!/usr/bin/env python
# subunit: extensions to python unittest to get test results from subprocesses.
# Copyright (C) 2013 Thomi Richards <thomi.richards@canonical.com>
#
# Licensed under either the Apache License, Version 2.0 or the BSD 3-clause
# license at the users choice. A copy of both licenses are available in the
# project source as Apache-2.0 and BSD. You may not use this file except in
# compliance with one of these two licences.
#
# Unless required by applicable law or agreed to in writing, software
# distributed under these licenses is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# license you chose for the specific language governing permissions and
# limitations under that license.
from argparse import ArgumentParser
import datetime
from sys import stdout
from subunit.v2 import StreamResultToBytes
def output_main():
args = parse_arguments()
output = get_output_stream_writer()
generate_bytestream(args, output)
return 0
def parse_arguments(args=None):
"""Parse arguments from the command line.
If specified, args must be a list of strings, similar to sys.argv[1:].
"""
parser = ArgumentParser(
prog='subunit-output',
description="A tool to generate a subunit result byte-stream",
)
common_args = ArgumentParser(add_help=False)
common_args.add_argument("test_id", help="""A string that uniquely
identifies this test.""")
sub_parsers = parser.add_subparsers(dest="action")
parser_start = sub_parsers.add_parser("start", help="Start a test.",
parents=[common_args])
parser_pass = sub_parsers.add_parser("pass", help="Pass a test.",
parents=[common_args])
parser_fail = sub_parsers.add_parser("fail", help="Fail a test.",
parents=[common_args])
parser_skip = sub_parsers.add_parser("skip", help="Skip a test.",
parents=[common_args])
return parser.parse_args(args)
def translate_command_name(command_name):
"""Turn the friendly command names we show users on the command line into
something subunit understands.
"""
return {
'start': 'inprogress',
'pass': 'success',
}.get(command_name, command_name)
def get_output_stream_writer():
return StreamResultToBytes(stdout)
def generate_bytestream(args, output_writer):
output_writer.startTestRun()
output_writer.status(
test_id=args.test_id,
test_status=translate_command_name(args.action),
timestamp=create_timestamp()
)
output_writer.stopTestRun()
_ZERO = datetime.timedelta(0)
class UTC(datetime.tzinfo):
"""UTC"""
def utcoffset(self, dt):
return _ZERO
def tzname(self, dt):
return "UTC"
def dst(self, dt):
return _ZERO
utc = UTC()
def create_timestamp():
return datetime.datetime.now(utc)
|
