Refactor PDFium python test utilities.
[pdfium.git] / testing / tools / run_corpus_tests.py
1 #!/usr/bin/env python
2 # Copyright 2015 The PDFium Authors. All rights reserved.
3 # Use of this source code is governed by a BSD-style license that can be
4 # found in the LICENSE file.
5
6 import optparse
7 import os
8 import re
9 import shutil
10 import subprocess
11 import sys
12
13 import common
14 import pngdiffer
15 import suppressor
16
17 # Nomenclature:
18 #   x_root - "x"
19 #   x_filename - "x.ext"
20 #   x_path - "path/to/a/b/c/x.ext"
21 #   c_dir - "path/to/a/b/c"
22
23 def test_one_file(input_filename, source_dir, working_dir,
24                   pdfium_test_path, image_differ):
25   input_path = os.path.join(source_dir, input_filename)
26   pdf_path = os.path.join(working_dir, input_filename)
27   try:
28     shutil.copyfile(input_path, pdf_path)
29     sys.stdout.flush()
30     subprocess.check_call([pdfium_test_path, '--png', pdf_path])
31   except subprocess.CalledProcessError as e:
32     print "FAILURE: " + input_filename + "; " + str(e)
33     return False
34   if image_differ.HasDifferences(input_filename, source_dir, working_dir):
35     return False
36   return True
37
38 def main():
39   parser = optparse.OptionParser()
40   parser.add_option('--build-dir', default=os.path.join('out', 'Debug'),
41                     help='relative path from the base source directory')
42   options, args = parser.parse_args()
43   finder = common.DirectoryFinder(options.build_dir)
44   pdfium_test_path = finder.ExecutablePath('pdfium_test')
45   working_dir = finder.WorkingDir(os.path.join('testing', 'corpus'))
46   if not os.path.exists(working_dir):
47     os.makedirs(working_dir)
48
49   test_suppressor = suppressor.Suppressor(finder)
50   image_differ = pngdiffer.PNGDiffer(finder)
51
52   # test files are under .../pdfium/testing/corpus.
53   failures = []
54   walk_from_dir = finder.TestingDir('corpus');
55   input_file_re = re.compile('^[a-zA-Z0-9_.]+[.]pdf$')
56   for source_dir, _, filename_list in os.walk(walk_from_dir):
57     for input_filename in filename_list:
58       if input_file_re.match(input_filename):
59          input_path = os.path.join(source_dir, input_filename)
60          if os.path.isfile(input_path):
61            if test_suppressor.IsSuppressed(input_filename):
62              continue
63          if not test_one_file(input_filename, source_dir, working_dir,
64                               pdfium_test_path, image_differ):
65              failures.append(input_path)
66
67   if failures:
68     print '\n\nSummary of Failures:'
69     for failure in failures:
70       print failure
71     return 1
72
73   return 0
74
75
76 if __name__ == '__main__':
77   sys.exit(main())