// 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.cf;

import com.android.tools.r8.NeverInline;
import com.android.tools.r8.R8TestRunResult;
import com.android.tools.r8.TestBase;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.TestParametersCollection;
import com.android.tools.r8.ToolHelper;
import com.android.tools.r8.utils.StringUtils;
import java.nio.file.Path;
import java.util.List;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

@RunWith(Parameterized.class)
public class MissingClassJoinsToObjectTest extends TestBase {

  static final String EXPECTED = StringUtils.lines("A::foo");

  private final TestParameters parameters;

  @Parameterized.Parameters(name = "{0}")
  public static TestParametersCollection data() {
    return getTestParameters().withAllRuntimes().withAllApiLevels().build();
  }

  public MissingClassJoinsToObjectTest(TestParameters parameters) {
    this.parameters = parameters;
  }

  private List<Path> getRuntimeClasspath() throws Exception {
    return buildOnDexRuntime(parameters, ToolHelper.getClassFileForTestClass(B.class));
  }

  @Test
  public void testReference() throws Exception {
    testForRuntime(parameters)
        .addProgramClasses(TestClass.class, A.class, B.class)
        .run(parameters.getRuntime(), TestClass.class)
        .assertSuccessWithOutput(EXPECTED);
  }

  @Test
  public void testR8() throws Exception {
    R8TestRunResult result =
        testForR8(parameters.getBackend())
            .enableInliningAnnotations()
            .addProgramClasses(TestClass.class, A.class)
            .addKeepMainRule(TestClass.class)
            .addDontWarn(B.class)
            .setMinApi(parameters.getApiLevel())
            .compile()
            .addRunClasspathFiles(getRuntimeClasspath())
            .run(parameters.getRuntime(), TestClass.class);
    if (parameters.isCfRuntime()) {
      // TODO(b/154792347): The analysis of types in the presence of undefined is incomplete.
      result.assertFailureWithErrorThatThrows(VerifyError.class);
    } else {
      result.assertSuccessWithOutput(EXPECTED);
    }
  }

  static class A {
    @NeverInline
    public void foo() {
      System.out.println("A::foo");
    }
  }

  // Missing at compile time.
  static class B extends A {
    // Intentionally empty.
  }

  static class TestClass {

    public static void main(String[] args) {
      // Due to the missing class B, the join is assigned Object.
      A join = args.length == 0 ? new A() : new B();
      // The call to Object::foo fails.
      join.foo();
    }
  }
}
