compile_java_tests.py

Enable keyboard shortcuts

#!/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()