src/test/java/com/android/tools/r8/diagnosticinspector/FoundMissingDefinitionContextSubject.java

// Copyright (c) 2021, 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.diagnosticinspector;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import com.android.tools.r8.diagnostic.DefinitionClassContext;
import com.android.tools.r8.diagnostic.DefinitionContext;
import com.android.tools.r8.diagnostic.DefinitionFieldContext;
import com.android.tools.r8.diagnostic.DefinitionMethodContext;

public class FoundMissingDefinitionContextSubject {

  private final DefinitionContext context;

  public FoundMissingDefinitionContextSubject(DefinitionContext context) {
    this.context = context;
  }

  public FoundMissingDefinitionContextSubject assertEqualTo(
      DefinitionClassContext expectedContext) {
    assertTrue(context.isClassContext());
    assertEquals(expectedContext.getClassReference(), context.asClassContext().getClassReference());
    return this;
  }

  public FoundMissingDefinitionContextSubject assertEqualTo(
      DefinitionFieldContext expectedContext) {
    assertTrue(context.isFieldContext());
    assertEquals(expectedContext.getFieldReference(), context.asFieldContext().getFieldReference());
    return this;
  }

  public FoundMissingDefinitionContextSubject assertEqualTo(
      DefinitionMethodContext expectedContext) {
    assertTrue(context.isMethodContext());
    assertEquals(
        expectedContext.getMethodReference(), context.asMethodContext().getMethodReference());
    return this;
  }
}