tools/json_comment_eater/json_comment_eater_test.py

#!/usr/bin/env python3
# Copyright 2013 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from json_comment_eater import Nom
import os
import unittest

class JsonCommentEaterTest(unittest.TestCase):
  def _Load(self, test_name):
    '''Loads the input and expected output for |test_name| as given by reading
    in |test_name|.json and |test_name|_expected.json, and returns the string
    contents as a tuple in that order.
    '''
    def read(file_name):
      file_path = os.path.join(os.path.dirname(__file__), file_name)
      with open(file_path, 'r') as f:
        return f.read()
    return [read(pattern % test_name)
            for pattern in ('%s.json', '%s_expected.json')]

  def testEverything(self):
    json, expected_json = self._Load('everything')
    self.assertEqual(expected_json, Nom(json))

if __name__ == '__main__':
  unittest.main()