Refactor PDFium python test utilities.

testing/tools/suppressor.py

+#!/usr/bin/env python
+# Copyright 2015 The PDFium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import os
+import sys
+
+class Suppressor:
+  def __init__(self, os_name, path_to_files):
+    self.suppressions_filename = 'SUPPRESSIONS'

[Lei Zhang, 2015/04/02 23:16:20]

class constant

[Tom Sepez, 2015/04/03 18:13:15]

Done.

+    self.platform_suppressions_filename = 'SUPPRESSIONS_%s' % os_name
+    self.suppression_list = self._ExtractSuppressions(
+      os.path.join(path_to_files, self.suppressions_filename))
+    self.platform_suppression_list = self._ExtractSuppressions(
+      os.path.join(path_to_files, self.platform_suppressions_filename))
+
+  def _ExtractSuppressions(self, suppressions_filename):
+    with open(suppressions_filename) as f:
+      return [y for y in [x.split('#')[0].strip() for x in f.readlines()] if y]
+
+  def IsSuppressed(self, input_filename):
+    if input_filename in self.suppression_list:
+      print ("Not running %s, found in %s file" %
+             (input_filename, self.suppressions_filename))
+      return True
+    if input_filename in self.platform_suppression_list:
+      print ("Not running %s, found in %s file" %
+             (input_filename, self.platform_suppressions_filename))
+      return True
+    return False
+
+
+# Testing.
+def main(argv):
+  if len(argv) != 3:
+    print "usaage: %s <test-file-name> <path-to-files>" % argv[0]
+    return 1
+  if sys.platform.startswith('linux'):
+    os_name = 'linux'
+  elif sys.platform.startswith('win'):
+    os_name = 'win'
+  elif sys.platform.startswith('darwin'):
+    os_name = 'mac'
+  else:
+    print 'Confused, can not determine OS, aborting.'
+    return 1
+  suppressor = Suppressor(os_name, argv[2])
+  if suppressor.IsSuppressed(argv[1]):
+    print "Result was true"
+  return 0
+
+
+if __name__ == '__main__':
+  sys.exit(main(sys.argv))