#!/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) || BUILDFLAG(IS_DESKTOP_ANDROID) '
      '|| BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_MAC) || 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)', _getPlatformIfdefs(cpp_bundle_generator, model)
    )


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