// Copyright 2022 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 (
 "android/soong/aconfig"
 "runtime"
 "testing"

 "android/soong/android"
 "android/soong/etc"
)

var prepareRavenwoodRuntime = android.GroupFixturePreparers(
 android.FixtureRegisterWithContext(func(ctx android.RegistrationContext) {
  RegisterRavenwoodBuildComponents(ctx)
 }),
 android.FixtureAddTextFile("ravenwood/Android.bp", `
  all_aconfig_declarations {
   name: "all_aconfig_declarations",
  }
  cc_library_shared {
   name: "ravenwood-runtime-jni1",
   host_supported: true,
   srcs: ["jni.cpp"],
  }
  cc_library_shared {
   name: "ravenwood-runtime-jni2",
   host_supported: true,
   srcs: ["jni.cpp"],
   stem: "libred",
   shared_libs: [
    "ravenwood-runtime-jni3",
   ],
  }
  cc_library_shared {
   name: "ravenwood-runtime-jni3",
   host_supported: true,
   srcs: ["jni.cpp"],
  }
  java_library_static {
   name: "framework-minus-apex.ravenwood",
   srcs: ["Framework.java"],
  }
  java_library_static {
   name: "framework-services.ravenwood",
   srcs: ["Services.java"],
  }
  java_library_static {
   name: "ravenwood-runtime-extra",
   stem: "runtime-extra",
   srcs: ["Extra.java"],
  }
  java_library_static {
   name: "framework-rules.ravenwood",
   srcs: ["Rules.java"],
  }
  android_app {
   name: "app1",
   sdk_version: "current",
  }
  android_app {
   name: "app2",
   sdk_version: "current",
  }
  prebuilt_font {
   name: "Font.ttf",
   src: "Font.ttf",
  }
  java_library {
   name: "sub1-1",
  }
  java_library {
   name: "sub2-1",
  }
  java_library {
   name: "sub2-2",
  }
  android_ravenwood_libgroup {
   name: "ravenwood-runtime.sub1",
   libs: [
    "sub1-1",
   ],
   install_name: "ravenwood-runtime",
  }
  android_ravenwood_libgroup {
   name: "ravenwood-runtime.sub2",
   libs: [
    "sub2-1",
    "sub2-2",
   ],
   install_name: "ravenwood-runtime",
  }
  android_ravenwood_libgroup {
   name: "ravenwood-runtime",
   libs: [
    "framework-minus-apex.ravenwood",
    "framework-services.ravenwood",
    "ravenwood-runtime-extra",
   ],
   jni_libs: [
    "ravenwood-runtime-jni1",
    "ravenwood-runtime-jni2",
   ],
   data: [
    {
     path: "ravenwood-data",
     srcs: [":app1"],
    },
    {
     path: "fonts",
     srcs: [":Font.ttf"],
    },
   ],
   subruntimes: [
    "ravenwood-runtime.sub1",
    "ravenwood-runtime.sub2",
   ],
  }
  android_ravenwood_libgroup {
   name: "ravenwood-utils",
   libs: [
    "framework-rules.ravenwood",
   ],
  }
 `),
)

var installPathPrefix = "out/host/linux-x86/testcases"

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

 ctx := android.GroupFixturePreparers(
  PrepareForIntegrationTestWithJava,
  etc.PrepareForTestWithPrebuiltEtc,
  aconfig.PrepareForTestWithAconfigBuildComponents,
  prepareRavenwoodRuntime,
 ).RunTest(t)

 // Verify that our runtime depends on underlying libs
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "framework-minus-apex.ravenwood")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "framework-services.ravenwood")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "ravenwood-runtime-jni")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "sub1-1")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "sub2-1")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-runtime", "android_common", "sub2-2")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-utils", "android_common", "framework-rules.ravenwood")

 // Verify that we've emitted artifacts in expected location
 runtime := ctx.ModuleForTests(t, "ravenwood-runtime", "android_common")
 runtimeOutput := runtime.Output(installPathPrefix + "/ravenwood-runtime/framework-minus-apex.ravenwood.jar")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/framework-services.ravenwood.jar")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/runtime-extra.jar")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/lib64/ravenwood-runtime-jni1.so")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/lib64/libred.so")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/lib64/ravenwood-runtime-jni3.so")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/ravenwood-data/app1.apk")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/fonts/Font.ttf")

 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/etc/all_aconfig_declarations.pb")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/etc/all_aconfig_declarations.textproto")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/maps/all_aconfig_declarations.package.map")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/maps/all_aconfig_declarations.flag.map")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/boot/all_aconfig_declarations.flag.info")
 runtime.Output(installPathPrefix + "/ravenwood-runtime/aconfig/metadata/aconfig/boot/all_aconfig_declarations.val")

 // Check the output files from the subruntimes.
 sub1 := ctx.ModuleForTests(t, "ravenwood-runtime.sub1", "android_common")
 sub1.Output(installPathPrefix + "/ravenwood-runtime/sub1-1.jar")

 sub2 := ctx.ModuleForTests(t, "ravenwood-runtime.sub2", "android_common")
 sub2.Output(installPathPrefix + "/ravenwood-runtime/sub2-1.jar")
 sub2.Output(installPathPrefix + "/ravenwood-runtime/sub2-2.jar")

 // Make sure the ravenwood-runtime output files implicitly depend on output of subruntimes.
 OrderOnly := runtimeOutput.OrderOnly.Strings()
 android.AssertStringListContains(t, "OrderOnly", OrderOnly, installPathPrefix+"/ravenwood-runtime/sub1-1.jar")
 android.AssertStringListContains(t, "OrderOnly", OrderOnly, installPathPrefix+"/ravenwood-runtime/sub2-1.jar")
 android.AssertStringListContains(t, "OrderOnly", OrderOnly, installPathPrefix+"/ravenwood-runtime/sub2-1.jar")

 utils := ctx.ModuleForTests(t, "ravenwood-utils", "android_common")
 utils.Output(installPathPrefix + "/ravenwood-utils/framework-rules.ravenwood.jar")
}

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

 ctx := android.GroupFixturePreparers(
  PrepareForIntegrationTestWithJava,
  etc.PrepareForTestWithPrebuiltEtc,
  aconfig.PrepareForTestWithAconfigBuildComponents,
  prepareRavenwoodRuntime,
 ).RunTestWithBp(t, `
  cc_library_shared {
   name: "jni-lib1",
   host_supported: true,
   srcs: ["jni.cpp"],
  }
  cc_library_shared {
   name: "jni-lib2",
   host_supported: true,
   srcs: ["jni.cpp"],
   stem: "libblue",
   shared_libs: [
    "jni-lib3",
   ],
  }
  cc_library_shared {
   name: "jni-lib3",
   host_supported: true,
   srcs: ["jni.cpp"],
   stem: "libpink",
  }
  java_defaults {
   name: "ravenwood-test-defaults",
   jni_libs: ["jni-lib2"],
  }
  android_ravenwood_test {
   name: "ravenwood-test",
   srcs: ["Test.java"],
   defaults: ["ravenwood-test-defaults"],
   jni_libs: [
    "jni-lib1",
    "ravenwood-runtime-jni2",
   ],
   data: [
    "data/file1.txt",
    "data2/sub/file2",
   ],
   build_resources: true,
   manifest: "AndroidManifest.xml",
   target_resource_apk: "app2",
   sdk_version: "test_current",
   target_sdk_version: "34",
   package_name: "a.b.c",
   target_package_name: "x.y.z",
   instrumentation_class: "androidx.test.runner.AndroidJUnitRunner",
  }
  android_ravenwood_test {
   name: "ravenwood-test-empty",
  }
 `)

 // Verify that our test depends on underlying libs
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-test", "android_common", "ravenwood-buildtime")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-test", "android_common", "ravenwood-utils")
 CheckModuleHasDependency(t, ctx.TestContext, "ravenwood-test", "android_common", "jni-lib")

 module := ctx.ModuleForTests(t, "ravenwood-test", "android_common")
 classpath := module.Rule("javac").Args["classpath"]

 // Verify that we're linking against test_current
 android.AssertStringDoesContain(t, "classpath", classpath, "android_test_stubs_current.jar")
 // Verify that we're linking against utils
 android.AssertStringDoesContain(t, "classpath", classpath, "framework-rules.ravenwood.jar")
 // Verify that we're *NOT* linking against runtime
 android.AssertStringDoesNotContain(t, "classpath", classpath, "framework-minus-apex.ravenwood.jar")
 android.AssertStringDoesNotContain(t, "classpath", classpath, "framework-services.ravenwood.jar")

 // Verify that we've emitted test artifacts in expected location
 outputJar := module.Output(installPathPrefix + "/ravenwood-test/ravenwood-test.jar")
 module.Output(installPathPrefix + "/ravenwood-test/ravenwood-test.config")
 module.Output(installPathPrefix + "/ravenwood-test/ravenwood.properties")
 module.Output(installPathPrefix + "/ravenwood-test/lib64/jni-lib1.so")
 module.Output(installPathPrefix + "/ravenwood-test/lib64/libblue.so")
 module.Output(installPathPrefix + "/ravenwood-test/lib64/libpink.so")
 module.Output(installPathPrefix + "/ravenwood-test/ravenwood-res-apks/ravenwood-res.apk")
 module.Output(installPathPrefix + "/ravenwood-test/ravenwood-res-apks/ravenwood-target-res.apk")
 module.Output(installPathPrefix + "/ravenwood-test/data/file1.txt")
 module.Output(installPathPrefix + "/ravenwood-test/data2/sub/file2")

 module = ctx.ModuleForTests(t, "ravenwood-test-empty", "android_common")
 module.Output(installPathPrefix + "/ravenwood-test-empty/ravenwood.properties")

 // ravenwood-runtime*.so are included in the runtime, so it shouldn't be emitted.
 for _, o := range module.AllOutputs() {
  android.AssertStringDoesNotContain(t, "runtime libs shouldn't be included", o, "/ravenwood-test/lib64/ravenwood-runtime")
 }

 // Verify that we're going to install underlying libs
 orderOnly := outputJar.OrderOnly.Strings()
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-runtime/framework-minus-apex.ravenwood.jar")
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-runtime/framework-services.ravenwood.jar")
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-runtime/lib64/ravenwood-runtime-jni1.so")
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-runtime/lib64/libred.so")
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-runtime/lib64/ravenwood-runtime-jni3.so")
 android.AssertStringListContains(t, "orderOnly", orderOnly, installPathPrefix+"/ravenwood-utils/framework-rules.ravenwood.jar")

 // Ensure they are listed as "test" modules for code coverage
 expectedTestOnlyModules := []string{
  "ravenwood-test",
  "ravenwood-test-empty",
 }
 expectedTopLevelTests := []string{
  "ravenwood-test",
  "ravenwood-test-empty",
 }
 assertTestOnlyAndTopLevel(t, ctx, expectedTestOnlyModules, expectedTopLevelTests)
}