// Copyright 2024 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package java

import (
 "runtime"
 "testing"

 "android/soong/android"
)

var prepareRobolectricRuntime = android.GroupFixturePreparers(
 android.FixtureRegisterWithContext(func(ctx android.RegistrationContext) {
  RegisterRobolectricBuildComponents(ctx)
 }),
 android.FixtureAddTextFile("robolectric/Android.bp", `
 java_library {
  name: "Robolectric_all-target_upstream",
  srcs: ["Robo.java"]
 }

 java_library {
  name: "Robolectric_all-target",
  srcs: ["Robo.java"]
 }

 java_library {
  name: "mockito-robolectric-prebuilt",
  srcs: ["Mockito.java"]
 }

 java_library {
  name: "truth",
  srcs: ["Truth.java"]
 }

 java_library {
  name: "junitxml",
  srcs: ["JUnitXml.java"]

 }

 java_library {
  name: "ClearcutJunitListenerAar",
  srcs: ["Runtime.java"]
 }

 java_library_host {
  name: "robolectric-host-android_all",
  srcs: ["Runtime.java"]
 }

 android_robolectric_runtimes {
  name: "robolectric-android-all-prebuilts",
  jars: ["android-all/Runtime.jar"],
  lib: "robolectric-host-android_all",
 }
 `),
)

func TestRobolectricJniTest(t *testing.T) {
 t.Parallel()
 if runtime.GOOS != "linux" {
  t.Skip("requires linux")
 }

 ctx := android.GroupFixturePreparers(
  PrepareForIntegrationTestWithJava,
  prepareRobolectricRuntime,
 ).RunTestWithBp(t, `
 android_app {
  name: "inst-target",
  srcs: ["App.java"],
  platform_apis: true,
 }

 cc_library_shared {
  name: "jni-lib1",
  host_supported: true,
  srcs: ["jni.cpp"],
 }

 android_robolectric_test {
  name: "robo-test",
  instrumentation_for: "inst-target",
  srcs: ["FooTest.java"],
  jni_libs: [
   "jni-lib1"
  ],
 }
 `)

 CheckModuleHasDependency(t, ctx.TestContext, "robo-test", "android_common", "jni-lib1")

 // Check that the .so files make it into the output.
 module := ctx.ModuleForTests(t, "robo-test", "android_common")
 module.Output(installPathPrefix + "/robo-test/lib64/jni-lib1.so")

 // Ensure they are listed as "test" modules for code coverage
 expectedTestOnlyModules := []string{
  "robo-test",
 }

 expectedTopLevelTests := []string{
  "robo-test",
 }
 assertTestOnlyAndTopLevel(t, ctx, expectedTestOnlyModules, expectedTopLevelTests)

}