sdk/pkg/dart2bytecode/test/object_table_test.dart

// Copyright (c) 2024, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:dart2bytecode/object_table.dart';
import 'package:kernel/ast.dart';
import 'package:kernel/core_types.dart' show CoreTypes;
import 'package:kernel/testing/mock_sdk_component.dart';
import 'package:test/test.dart';
import 'package:expect/expect.dart';

main() {
  late Library lib1;
  late Library lib2;
  late CoreTypes coreTypes;
  late Supertype objectSupertype;
  late ObjectTable objectTable;

  Class addClass(
    Library lib,
    String name, [
    List<TypeParameter> typeParameters = const [],
  ]) {
    final cls = Class(
      name: name,
      supertype: objectSupertype,
      typeParameters: typeParameters,
      fileUri: lib.fileUri,
    );
    cls.parent = lib;
    lib.addClass(cls);
    return cls;
  }

  setUp(() {
    // Start with mock SDK libraries.
    final component = createMockSdkComponent();
    coreTypes = CoreTypes(component);
    objectSupertype = coreTypes.objectClass.asThisSupertype;

    // Add test libraries.
    lib1 = Library(
      Uri.parse('org-dartlang:///test1.dart'),
      name: 'lib1',
      fileUri: Uri.parse('file:///test1.dart'),
    );
    lib1.parent = component;
    component.libraries.add(lib1);

    lib2 = Library(
      Uri.parse('org-dartlang:///test2.dart'),
      name: 'lib2',
      fileUri: Uri.parse('file:///test2.dart'),
    );
    lib2.parent = component;
    component.libraries.add(lib2);

    objectTable = ObjectTable(coreTypes);
  });

  tearDown(() {});

  test('libraries', () {
    final h1 = objectTable.getHandle(lib1)!;
    final h2 = objectTable.getHandle(lib2)!;
    Expect.notEquals(h1, h2);
    Expect.identical(h1, objectTable.getHandle(lib1));
    Expect.identical(h2, objectTable.getHandle(lib2));
    Expect.equals(true, h1.isCacheable);
    Expect.equals(true, h2.isCacheable);
    Expect.equals(false, h1.shouldBeIncludedIntoIndexTable); // 2 uses
    Expect.identical(h1, objectTable.getHandle(lib1));
    Expect.equals(true, h1.shouldBeIncludedIntoIndexTable); // 3 uses
  });

  test('classes', () {
    final Class c1 = addClass(lib1, 'A');
    final Class c2 = addClass(lib1, 'B');
    final Class c3 = addClass(lib2, 'A');
    final h1 = objectTable.getHandle(c1)!;
    final h2 = objectTable.getHandle(c2)!;
    final h3 = objectTable.getHandle(c3)!;
    Expect.notEquals(h1, h2);
    Expect.notEquals(h1, h3);
    Expect.notEquals(h2, h3);
    Expect.identical(h1, objectTable.getHandle(c1));
    Expect.identical(h2, objectTable.getHandle(c2));
    Expect.identical(h3, objectTable.getHandle(c3));
    Expect.equals(true, h1.isCacheable);
    Expect.equals(true, h2.isCacheable);
    Expect.equals(true, h3.isCacheable);
    final lib1h = objectTable.getHandle(lib1)!;
    final lib2h = objectTable.getHandle(lib2)!;
    Expect.equals(true, lib1h.shouldBeIncludedIntoIndexTable); // 3 uses
    Expect.equals(false, lib2h.shouldBeIncludedIntoIndexTable); // 2 uses
  });

  test('simple-types', () {
    final h1a = objectTable.getHandle(
      InterfaceType(coreTypes.intClass, Nullability.nonNullable),
    )!;
    final h1b = objectTable.getHandle(
      InterfaceType(coreTypes.intClass, Nullability.nonNullable),
    )!;
    final h2a = objectTable.getHandle(const DynamicType())!;
    final h2b = objectTable.getHandle(DynamicType())!;
    Expect.identical(h1a, h1b);
    Expect.identical(h2a, h2b);
    Expect.notEquals(h1a, h2a);
    Expect.equals(true, h1a.isCacheable);
    Expect.equals(true, h2a.isCacheable);
    Expect.equals(false, h1a.shouldBeIncludedIntoIndexTable); // 2 uses
    objectTable.getHandle(
      InterfaceType(coreTypes.listClass, Nullability.nonNullable, [
        InterfaceType(coreTypes.intClass, Nullability.nonNullable),
      ]),
    );
    Expect.equals(true, h1a.shouldBeIncludedIntoIndexTable); // 3 uses
  });
}