forked from mgedmin/imgdiff
/
tests.py
135 lines (106 loc) · 4.19 KB
/
tests.py
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
import doctest
import os
import shutil
import sys
import tempfile
import unittest
try:
from cStringIO import StringIO
except ImportError:
from io import StringIO
import mock
import imgdiff
@mock.patch('sys.stderr', StringIO())
class TestMain(unittest.TestCase):
def setUp(self):
self.tmpdir = None
def tearDown(self):
if self.tmpdir:
shutil.rmtree(self.tmpdir)
def mkdtemp(self):
if self.tmpdir is None:
self.tmpdir = tempfile.mkdtemp(prefix='imgdiff-tests-')
return self.tmpdir
def main(self, *args):
try:
imgdiff.main(['imgdiff'] + list(args))
except SystemExit:
pass
def assertIn(self, member, container): # Python 2.6 compat
if member not in container:
self.fail('%s not found in %s' % (repr(member), repr(container)))
def test_color_parsing_in_options(self):
self.main('--bgcolor', 'invalid')
self.assertIn("error: option --bgcolor: invalid color value: 'invalid'",
sys.stderr.getvalue())
def test_wrong_number_of_arguments(self):
self.main('foo.png')
self.assertIn("error: expecting two arguments, got 1",
sys.stderr.getvalue())
self.main('foo.png', 'bar.png', 'baz.png')
self.assertIn("error: expecting two arguments, got 3",
sys.stderr.getvalue())
def test_two_directories(self):
self.main('set1', 'set2')
self.assertIn("error: at least one argument must be a file, not a directory",
sys.stderr.getvalue())
def test_all_ok(self):
self.main('example1.png', 'example2.png', '--viewer=true')
def test_highlight(self):
self.main('example1.png', 'example2.png', '-H', '--viewer=true')
def test_smart_highlight(self):
self.main('example1.png', 'example2.png', '-S', '--viewer=true')
def test_outfile(self):
fn = os.path.join(self.mkdtemp(), 'diff.png')
self.main('example1.png', 'example2.png', '-o', fn)
self.assertTrue(os.path.exists(fn))
@mock.patch('imgdiff.Image.Image.show')
def test_builtin_viewer(self, mock_show):
self.main('example1.png', 'example2.png')
self.assertTrue(mock_show.called)
def test_one_directory(self):
self.main('set1/canary.png', 'set2', '--viewer', 'true')
self.main('set1', 'set2/canary.png', '--viewer', 'true', '--tb')
def test_different_size_images(self):
# tickle the unexplored branches in best_diff()
self.main('set1/extra-info.png', 'set1/sample-graph.png', '--viewer=true', '-H', '--timeout=1')
def test_different_size_images_sloow(self):
# tickle the unexplored branches in slow_highlight()
self.main('set1/extra-info.png', 'set1/sample-graph.png', '--viewer=true', '-S', '--timeout=10')
class TestProgress(unittest.TestCase):
def test_terminal_output(self):
p = imgdiff.Progress(3, delay=0)
p.stream = StringIO()
p.isatty = True
for n in range(3):
p.next()
p.done()
self.assertEqual(p.stream.getvalue(),
'\r33% (1 out of 3 possible alignments)'
'\r66% (2 out of 3 possible alignments)'
'\r100% (3 out of 3 possible alignments)'
'\r\n')
def test_not_a_terminal(self):
p = imgdiff.Progress(3, delay=0)
p.stream = StringIO()
p.isatty = False
for n in range(3):
p.next()
p.done()
self.assertEqual(p.stream.getvalue(), '')
def test_timeout(self):
p = imgdiff.Progress(3, timeout=1)
p.stream = StringIO()
p.isatty = False
p.started -= 2 # so it seems we started a second ago
self.assertRaises(imgdiff.Timeout, p.next)
p.done()
self.assertEqual(p.stream.getvalue(),
'Highlighting takes too long: timed out after 1 seconds\n')
def test_suite():
return unittest.TestSuite([
doctest.DocTestSuite(imgdiff),
unittest.defaultTestLoader.loadTestsFromName(__name__),
])
if __name__ == '__main__':
unittest.main(defaultTest='test_suite')