[Coverage] Add sancov_formatter unittest for split.

tools/sanitizers/sancov_formatter_test.py

 # Copyright 2016 the V8 project authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 # Requires python-coverage. Native python coverage version >= 3.7.1 should
 # be installed to get the best speed.
 
 import copy
 import coverage
 import logging
+import json
 import os
+import shutil
 import sys
+import tempfile
 import unittest
 
 
 # Directory of this file.
 LOCATION = os.path.dirname(os.path.abspath(__file__))
 
 # V8 checkout directory.
 BASE_DIR = os.path.dirname(os.path.dirname(LOCATION))
 
 # Executable location.
@@ skipping 98 matching lines @@
   'version': 1,
   'tests': ['cctest', 'd8', 'unittests'],
   'files': {
     'src/baz/bar.h': [[0, 0b0], [3, 0b11], [7, 0b11], [8, 0b0]],
     'src/baz.cc': [[1, 0b0], [2, 0b101]],
     'src/foo.cc': [[1, 0b1], [11, 0b10], [92, 0b100], [93, 0b0]],
   },
 }
 
 
+#------------------------------------------------------------------------------
+
+# Data for test_split.
+
+EXPECTED_SPLIT_FILES = [
+  (
+    os.path.join('src', 'baz', 'bar.h.json'),
+    {
+      'version': 1,
+      'tests': ['cctest', 'd8', 'unittests'],
+      'files': {
+        'src/baz/bar.h': [[0, 0b0], [3, 0b11], [7, 0b11], [8, 0b0]],
+      },
+    },
+  ),
+  (
+    os.path.join('src', 'baz.cc.json'),
+    {
+      'version': 1,
+      'tests': ['cctest', 'd8', 'unittests'],
+      'files': {
+        'src/baz.cc': [[1, 0b0], [2, 0b101]],
+      },
+    },
+  ),
+  (
+    os.path.join('src', 'foo.cc.json'),
+    {
+      'version': 1,
+      'tests': ['cctest', 'd8', 'unittests'],
+      'files': {
+        'src/foo.cc': [[1, 0b1], [11, 0b10], [92, 0b100], [93, 0b0]],
+      },
+    },
+  ),
+]
+
+
 class FormatterTests(unittest.TestCase):
   @classmethod
   def setUpClass(cls):
     sys.path.append(LOCATION)
     cls._cov = coverage.coverage(
         include=([os.path.join(LOCATION, 'sancov_formatter.py')]))
     cls._cov.start()
     import sancov_formatter
     global sancov_formatter
 
   @classmethod
   def tearDownClass(cls):
     cls._cov.stop()
     cls._cov.report()
 
   def test_process_symbolizer_output(self):
     result = sancov_formatter.process_symbolizer_output(SYMBOLIZER_OUTPUT)
     self.assertEquals(EXPECTED_PROCESSED_OUTPUT, result)
 
   def test_merge_instrumented_line_results(self):
     result = sancov_formatter.merge_instrumented_line_results(
       EXE_LIST, INSTRUMENTED_LINE_RESULTS)
     self.assertEquals(EXPECTED_INSTRUMENTED_LINES_DATA, result)
 
   def test_merge_covered_line_results(self):
     data = copy.deepcopy(EXPECTED_INSTRUMENTED_LINES_DATA)
     sancov_formatter.merge_covered_line_results(
       data, COVERED_LINE_RESULTS)
     self.assertEquals(EXPECTED_COVERED_LINES_DATA, data)
+
+  def test_split(self):
+    _, json_input = tempfile.mkstemp(prefix='tmp_coverage_test_split')
+    with open(json_input, 'w') as f:
+      json.dump(EXPECTED_COVERED_LINES_DATA, f)
+    output_dir = tempfile.mkdtemp(prefix='tmp_coverage_test_split')
+
+    try:
+      sancov_formatter.main([
+        'split',
+        '--json-input', json_input,
+        '--output-dir', output_dir,
+      ])
+
+      for file_name, expected_data in EXPECTED_SPLIT_FILES:
+        full_path = os.path.join(output_dir, file_name)
+        self.assertTrue(os.path.exists(full_path))
+        with open(full_path) as f:
+          self.assertEquals(expected_data, json.load(f))
+    finally:
+      os.remove(json_input)
+      shutil.rmtree(output_dir)