| // Copyright (c) 2020, the R8 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. |
| |
| package com.android.tools.r8.utils.codeinspector; |
| |
| import static com.android.tools.r8.TestBase.toDexType; |
| import static org.junit.Assert.assertTrue; |
| |
| import com.android.tools.r8.graph.DexItemFactory; |
| import com.android.tools.r8.graph.classmerging.VerticallyMergedClasses; |
| |
| public class VerticallyMergedClassesInspector { |
| |
| private final DexItemFactory dexItemFactory; |
| private final VerticallyMergedClasses verticallyMergedClasses; |
| |
| public VerticallyMergedClassesInspector( |
| DexItemFactory dexItemFactory, VerticallyMergedClasses verticallyMergedClasses) { |
| this.dexItemFactory = dexItemFactory; |
| this.verticallyMergedClasses = verticallyMergedClasses; |
| } |
| |
| public VerticallyMergedClassesInspector assertMergedIntoSubtype(Class<?> clazz) { |
| assertTrue(verticallyMergedClasses.hasBeenMergedIntoSubtype(toDexType(clazz, dexItemFactory))); |
| return this; |
| } |
| |
| public VerticallyMergedClassesInspector assertMergedIntoSubtype(Class<?>... classes) { |
| for (Class<?> clazz : classes) { |
| assertMergedIntoSubtype(clazz); |
| } |
| return this; |
| } |
| |
| public VerticallyMergedClassesInspector assertNoClassesMerged() { |
| assertTrue(verticallyMergedClasses.isEmpty()); |
| return this; |
| } |
| } |