#!/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 compile_java.py"""

import collections
import unittest

import compile_java


def _CreateData(class_annotation='',
                class_prefix='public ',
                nested_annotation='',
                suffix=''):
  return f"""\
package pkg;

import foo;
import foo.Bar;
import foo.Bar.Baz;

{class_annotation}
@SomeThing
{class_prefix}class Foo {{
  {nested_annotation}
  public static class Nested {{ }}
}}
""" + suffix


class CompileJavaTests(unittest.TestCase):

  def do_service_loader_test(self, **kwargs):
    data = _CreateData(**kwargs)
    services_map = collections.defaultdict(list)
    compile_java.ParseJavaSource(data, services_map)
    return dict(services_map)

  def do_classes_test(self, **kwargs):
    data = _CreateData(**kwargs)
    services_map = collections.defaultdict(list)
    _, class_names = compile_java.ParseJavaSource(data, services_map)
    return class_names

  def testServiceImpl_NoUses(self):
    services_map = self.do_service_loader_test()
    self.assertEqual({}, services_map)

  def testServiceImpl_LocalType(self):
    services_map = self.do_service_loader_test(
        class_annotation='@ServiceImpl(Local.class)')
    self.assertEqual({'pkg.Local': ['pkg.Foo']}, services_map)

  def testServiceImpl_ImportedTopType(self):
    services_map = self.do_service_loader_test(
        class_annotation='@ServiceImpl(Bar.class)')
    self.assertEqual({'foo.Bar': ['pkg.Foo']}, services_map)

  def testServiceImpl_ImportedNestedType1(self):
    services_map = self.do_service_loader_test(
        class_annotation='@ServiceImpl(Baz.class)')
    self.assertEqual({'foo.Bar$Baz': ['pkg.Foo']}, services_map)

  def testServiceImpl_ImportedNestedType2(self):
    services_map = self.do_service_loader_test(
        class_annotation='@ServiceImpl(Bar.Baz.class)')
    self.assertEqual({'foo.Bar$Baz': ['pkg.Foo']}, services_map)

  def testServiceImpl_NestedImpl(self):
    services_map = self.do_service_loader_test(
        class_annotation='@ServiceImpl(Baz.class)',
        nested_annotation='@ServiceImpl(Baz.class)')
    self.assertEqual({'foo.Bar$Baz': ['pkg.Foo', 'pkg.Foo$Nested']},
                     services_map)

  def testParseClasses(self):
    classes = self.do_classes_test(class_prefix='public final ',
                                   suffix='\nprivate class Extra {}')
    self.assertEqual(['Foo', 'Extra'], classes)

  def testErrorOnNonPublic(self):

    def inner():
      self.do_classes_test(class_annotation='@ServiceImpl(Local.class)',
                           class_prefix='')

    self.assertRaises(Exception, inner)

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