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
|
import re
try:
import json
except ImportError:
import simplejson as json
import pytest
import docopt
def pytest_collect_file(file_path, path, parent):
if path.ext == ".docopt" and path.basename.startswith("test"):
if hasattr(DocoptTestFile, "from_parent"):
return DocoptTestFile.from_parent(parent, path=file_path)
return DocoptTestFile(path, parent)
def parse_test(raw):
raw = re.compile('#.*$', re.M).sub('', raw).strip()
if raw.startswith('"""'):
raw = raw[3:]
for fixture in raw.split('r"""'):
name = ''
doc, _, body = fixture.partition('"""')
cases = []
for case in body.split('$')[1:]:
argv, _, expect = case.strip().partition('\n')
expect = json.loads(expect)
prog, _, argv = argv.strip().partition(' ')
cases.append((prog, argv, expect))
yield name, doc, cases
class DocoptTestFile(pytest.File):
def collect(self):
raw = self.path.open().read()
index = 1
for name, doc, cases in parse_test(raw):
name = self.path.name
for case in cases:
if hasattr(DocoptTestItem, "from_parent"):
yield DocoptTestItem.from_parent(self, name="%s(%d)" % (name, index), doc=doc, case=case)
else:
yield DocoptTestItem("%s(%d)" % (name, index), self, doc, case)
index += 1
class DocoptTestItem(pytest.Item):
def __init__(self, name, parent, doc, case):
super(DocoptTestItem, self).__init__(name, parent)
self.doc = doc
self.prog, self.argv, self.expect = case
def runtest(self):
try:
result = docopt.docopt(self.doc, argv=self.argv)
except docopt.DocoptExit:
result = 'user-error'
if self.expect != result:
raise DocoptTestException(self, result)
def repr_failure(self, excinfo):
"""Called when self.runtest() raises an exception."""
if isinstance(excinfo.value, DocoptTestException):
return "\n".join((
"usecase execution failed:",
self.doc.rstrip(),
"$ %s %s" % (self.prog, self.argv),
"result> %s" % json.dumps(excinfo.value.args[1]),
"expect> %s" % json.dumps(self.expect),
))
def reportinfo(self):
return self.path, 0, "usecase: %s" % self.name
class DocoptTestException(Exception):
pass
|
