OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Tests for module write_json_summary.py""" | 6 """Tests for module write_json_summary.py""" |
7 | 7 |
8 import filecmp | 8 import filecmp |
9 import os | 9 import os |
10 import shutil | 10 import shutil |
11 import tempfile | 11 import tempfile |
12 import unittest | 12 import unittest |
13 | 13 |
14 import write_json_summary | 14 import write_json_summary |
15 | 15 |
16 | 16 |
17 class TestWriteJsonSummary(unittest.TestCase): | 17 class TestWriteJsonSummary(unittest.TestCase): |
18 | 18 |
19 def setUp(self): | 19 def setUp(self): |
20 self._test_json_dir = os.path.join( | 20 self._test_json_dir = os.path.join( |
21 os.path.dirname(os.path.realpath(__file__)), 'test_data') | 21 os.path.dirname(os.path.realpath(__file__)), 'test_data') |
22 self._output_file_name = 'summary.json' | 22 self._output_file_name = 'summary.json' |
23 self._gm_json_path = os.path.join(self._test_json_dir, 'gm_json_mock.py') | 23 self._gm_json_path = os.path.join(self._test_json_dir, 'gm_json_mock.py') |
24 self._imagediffdb_path = os.path.join(self._test_json_dir, | 24 self._imagediffdb_path = os.path.join(self._test_json_dir, |
25 'imagediffdb_mock.py') | 25 'imagediffdb_mock.py') |
26 self._skpdiff_output_csv = os.path.join(self._test_json_dir, 'output.csv') | |
27 self._actual_output_dir = tempfile.mkdtemp() | 26 self._actual_output_dir = tempfile.mkdtemp() |
28 self._actual_output_file_path = os.path.join(self._actual_output_dir, | 27 self._actual_output_file_path = os.path.join(self._actual_output_dir, |
29 self._output_file_name) | 28 self._output_file_name) |
30 self._gs_output_dir = 'gs://dummy-bucket/output-dir' | 29 self._gs_output_dir = 'gs://dummy-bucket/output-dir' |
31 self._gs_skp_dir = 'gs://dummy-bucket/skps' | 30 self._gs_skp_dir = 'gs://dummy-bucket/skps' |
32 self._img_root = '/tmp/' | 31 self._img_root = '/tmp/' |
33 self._nopatch_img_dir_name = 'nopatch' | 32 self._nopatch_images_base_url = 'file://fake/path/to/nopatch' |
34 self._withpatch_img_dir_name = 'withpatch' | 33 self._withpatch_images_base_url = 'file://fake/path/to/withpatch' |
35 self._slave_num = 1 | 34 self._slave_num = 1 |
36 | 35 |
37 def tearDown(self): | 36 def tearDown(self): |
38 shutil.rmtree(self._actual_output_dir) | 37 shutil.rmtree(self._actual_output_dir) |
39 | 38 |
40 def test_DifferentFiles(self): | 39 def test_DifferentFiles(self): |
41 write_json_summary.WriteJsonSummary( | 40 write_json_summary.WriteJsonSummary( |
42 img_root=self._img_root, | 41 img_root=self._img_root, |
43 nopatch_json=os.path.join(self._test_json_dir, 'output1.json'), | 42 nopatch_json=os.path.join(self._test_json_dir, 'output1.json'), |
44 nopatch_img_dir_name=self._nopatch_img_dir_name, | 43 nopatch_images_base_url=self._nopatch_images_base_url, |
45 withpatch_json=os.path.join(self._test_json_dir, 'output2.json'), | 44 withpatch_json=os.path.join(self._test_json_dir, 'output2.json'), |
46 withpatch_img_dir_name=self._withpatch_img_dir_name, | 45 withpatch_images_base_url=self._withpatch_images_base_url, |
47 output_file_path=self._actual_output_file_path, | 46 output_file_path=self._actual_output_file_path, |
48 gs_output_dir=self._gs_output_dir, | 47 gs_output_dir=self._gs_output_dir, |
49 gs_skp_dir=self._gs_skp_dir, | 48 gs_skp_dir=self._gs_skp_dir, |
50 slave_num=self._slave_num, | 49 slave_num=self._slave_num, |
51 gm_json_path=self._gm_json_path, | 50 gm_json_path=self._gm_json_path, |
52 imagediffdb_path=self._imagediffdb_path, | 51 imagediffdb_path=self._imagediffdb_path) |
53 skpdiff_output_csv=self._skpdiff_output_csv) | |
54 | 52 |
55 self.assertTrue( | 53 self.assertTrue( |
56 filecmp.cmp(os.path.join(self._test_json_dir, self._output_file_name), | 54 filecmp.cmp(os.path.join(self._test_json_dir, self._output_file_name), |
57 self._actual_output_file_path)) | 55 self._actual_output_file_path)) |
58 | 56 |
59 def test_NoDifferentFiles(self): | 57 def test_NoDifferentFiles(self): |
60 write_json_summary.WriteJsonSummary( | 58 write_json_summary.WriteJsonSummary( |
61 img_root=self._img_root, | 59 img_root=self._img_root, |
62 nopatch_json=os.path.join(self._test_json_dir, 'output1.json'), | 60 nopatch_json=os.path.join(self._test_json_dir, 'output1.json'), |
63 nopatch_img_dir_name=self._nopatch_img_dir_name, | 61 nopatch_images_base_url=self._nopatch_images_base_url, |
64 withpatch_json=os.path.join(self._test_json_dir, 'output1.json'), | 62 withpatch_json=os.path.join(self._test_json_dir, 'output1.json'), |
65 withpatch_img_dir_name=self._withpatch_img_dir_name, | 63 withpatch_images_base_url=self._withpatch_images_base_url, |
66 output_file_path=self._actual_output_file_path, | 64 output_file_path=self._actual_output_file_path, |
67 gs_output_dir=self._gs_output_dir, | 65 gs_output_dir=self._gs_output_dir, |
68 gs_skp_dir=self._gs_skp_dir, | 66 gs_skp_dir=self._gs_skp_dir, |
69 slave_num=self._slave_num, | 67 slave_num=self._slave_num, |
70 gm_json_path=self._gm_json_path, | 68 gm_json_path=self._gm_json_path, |
71 imagediffdb_path=self._imagediffdb_path, | 69 imagediffdb_path=self._imagediffdb_path) |
72 skpdiff_output_csv=self._skpdiff_output_csv) | |
73 | 70 |
74 self.assertFalse(os.path.isfile(self._actual_output_file_path)) | 71 self.assertFalse(os.path.isfile(self._actual_output_file_path)) |
75 | 72 |
76 | 73 |
77 if __name__ == '__main__': | 74 if __name__ == '__main__': |
78 unittest.main() | 75 unittest.main() |
79 | |
OLD | NEW |