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

This test suite contains various tests for the 'java_cpp_template' build rule,
which uses the gcc preprocessor to turn a template into Java code.
"""

import unittest
import tempfile

import gcc_preprocess


class TestPreprocess(unittest.TestCase):

  def testParsePackageName(self):
    with tempfile.NamedTemporaryFile(mode='w') as f:
      template = f.name
      f.file.write("""
package org.chromium.fake;
public class Empty {
}
""")
      f.file.flush()
      package_name, data = gcc_preprocess.ProcessJavaFile(template, [], [])
      self.assertEqual('org.chromium.fake', package_name)
      self.assertEqual(
          """
package org.chromium.fake;
public class Empty {
}
""".strip(), data.strip())

  def testMissingPackageName(self):
    with tempfile.NamedTemporaryFile(mode='w') as f:
      template = f.name
      f.file.write("""
public class Empty {
}
""")
      f.file.flush()
      with self.assertRaisesRegex(Exception,
                                  r'Could not find java package of.*'):
        gcc_preprocess.ProcessJavaFile(template, [], [])

  def testSinglePreprocessorEvaluation(self):
    with tempfile.NamedTemporaryFile(mode='w') as f:
      template = f.name
      f.file.write("""
package org.chromium.fake;
public class Sample {
#if defined(_ENABLE_ASSERTS)
    public boolean ENABLE_ASSERTS = true;
#else
    public boolean ENABLE_ASSERTS = false;
#endif
}
""")
      f.file.flush()
      defines = [
          '_ENABLE_ASSERTS',
      ]
      package_name, data = gcc_preprocess.ProcessJavaFile(template, defines, [])
      self.assertEqual('org.chromium.fake', package_name)
      self.assertEqual(
          """
package org.chromium.fake;
public class Sample {
    public boolean ENABLE_ASSERTS = true;
}
""".strip(), data.strip())

  def testNestedPreprocessorEvaluation(self):
    with tempfile.NamedTemporaryFile(mode='w') as f:
      template = f.name
      f.file.write("""
package org.chromium.fake;
#if defined(USE_FINAL)
#define MAYBE_FINAL final
#else
#define MAYBE_FINAL
#endif
public class Sample {
#if defined(_ENABLE_ASSERTS)
    public MAYBE_FINAL boolean ENABLE_ASSERTS = true;
#else
    public MAYBE_FINAL boolean ENABLE_ASSERTS = false;
#endif
}
""")
      f.file.flush()
      defines = [
          '_ENABLE_ASSERTS',
          'USE_FINAL',
      ]
      package_name, data = gcc_preprocess.ProcessJavaFile(template, defines, [])
      self.assertEqual('org.chromium.fake', package_name)
      self.assertEqual(
          """
package org.chromium.fake;
public class Sample {
    public final boolean ENABLE_ASSERTS = true;
}
""".strip(), data.strip())

  def testPreserveComments(self):
    with tempfile.NamedTemporaryFile(mode='w') as f:
      template = f.name
      f.file.write("""
// Copyright header ...
package org.chromium.fake;
/**
 * Some javadoc.
 */
public class Sample {
    // This is a comment outside the #if block.
#if defined(_ENABLE_ASSERTS)
    // Inside the #if block.
    public boolean ENABLE_ASSERTS = true;
#else
    // Inside the #else block.
    public boolean ENABLE_ASSERTS = false;
#endif
}
""")
      f.file.flush()
      defines = [
          '_ENABLE_ASSERTS',
      ]
      package_name, data = gcc_preprocess.ProcessJavaFile(template, defines, [])
      self.assertEqual('org.chromium.fake', package_name)
      self.assertEqual(
          """
// Copyright header ...
package org.chromium.fake;
/**
 * Some javadoc.
 */
public class Sample {
    // This is a comment outside the #if block.
    // Inside the #if block.
    public boolean ENABLE_ASSERTS = true;
}
""".strip(), data.strip())


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