chromium/tools/json_schema_compiler/cpp_bundle_generator_test.py

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

from cpp_bundle_generator import CppBundleGenerator
from model import Model

import json_schema
import os
import unittest


def _createCppBundleGenerator(file_path):
  json_object = json_schema.Load(file_path)
  model = Model()
  model.AddNamespace(json_object[0], file_path)
  cpp_bundle_generator = CppBundleGenerator(None, model, None, None,
                                            'generated_api_schemas', None, None,
                                            None)
  return (cpp_bundle_generator, model)


def _getPlatformIfdefs(cpp_bundle_generator, model):
  return cpp_bundle_generator._GetPlatformIfdefs(
      list(list(model.namespaces.values())[0].functions.values())[0])


class CppBundleGeneratorTest(unittest.TestCase):

  def testIfDefsForWinLinux(self):
    cpp_bundle_generator, model = _createCppBundleGenerator(
        'test/function_platform_win_linux.json')
    self.assertEqual('BUILDFLAG(IS_WIN) || BUILDFLAG(IS_LINUX)',
                     _getPlatformIfdefs(cpp_bundle_generator, model))

  def testIfDefsForAll(self):
    cpp_bundle_generator, model = _createCppBundleGenerator(
        'test/function_platform_all.json')
    self.assertEqual(
        'BUILDFLAG(IS_CHROMEOS_ASH) || '
        'BUILDFLAG(IS_FUCHSIA) || BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_WIN)',
        _getPlatformIfdefs(cpp_bundle_generator, model))

  def testIfDefsForChromeOS(self):
    cpp_bundle_generator, model = _createCppBundleGenerator(
        'test/function_platform_chromeos.json')
    self.assertEqual('BUILDFLAG(IS_CHROMEOS_ASH)',
                     _getPlatformIfdefs(cpp_bundle_generator, model))


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