// Copyright 2019 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 (
 "reflect"
 "testing"

 "android/soong/android"
 "android/soong/cc"
)

var prepareForJavaAndroidMkTest = android.GroupFixturePreparers(
 prepareForJavaTest,
 android.PrepareForTestWithAndroidMk,
)

func TestRequired(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  java_library {
   name: "foo",
   srcs: ["a.java"],
   required: ["libfoo"],
  }
 `)

 mod := ctx.ModuleForTests(t, "foo", "android_common").Module()
 info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)

 expected := []string{"libfoo"}
 actual := info.PrimaryInfo.EntryMap["LOCAL_REQUIRED_MODULES"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected required modules - expected: %q, actual: %q", expected, actual)
 }
}

func TestHostdex(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  java_library {
   name: "foo",
   srcs: ["a.java"],
   hostdex: true,
  }
 `)

 mod := ctx.ModuleForTests(t, "foo", "android_common").Module()
 info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)
 if len(info.ExtraInfo) != 1 {
  t.Errorf("one extra entry is expected, but got %d", len(info.ExtraInfo))
 }

 mainEntries := &info.PrimaryInfo
 expected := []string{"foo"}
 actual := mainEntries.EntryMap["LOCAL_MODULE"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected module name - expected: %q, actual: %q", expected, actual)
 }

 subEntries := &info.ExtraInfo[0]
 expected = []string{"foo-hostdex"}
 actual = subEntries.EntryMap["LOCAL_MODULE"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected module name - expected: %q, actual: %q", expected, actual)
 }
}

func TestHostdexRequired(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  java_library {
   name: "foo",
   srcs: ["a.java"],
   hostdex: true,
   required: ["libfoo"],
  }
 `)

 mod := ctx.ModuleForTests(t, "foo", "android_common").Module()
 info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)
 if len(info.ExtraInfo) != 1 {
  t.Errorf("one extra entries is expected, but got %d", len(info.ExtraInfo))
 }

 if len(info.ExtraInfo) != 1 {
  t.Errorf("one extra entries is expected, but got %d", len(info.ExtraInfo))
 }

 mainEntries := &info.PrimaryInfo

 expected := []string{"libfoo"}
 actual := mainEntries.EntryMap["LOCAL_REQUIRED_MODULES"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected required modules - expected: %q, actual: %q", expected, actual)
 }

 subEntries := &info.ExtraInfo[0]

 expected = []string{"libfoo"}
 actual = subEntries.EntryMap["LOCAL_REQUIRED_MODULES"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected required modules - expected: %q, actual: %q", expected, actual)
 }
}

func TestHostdexSpecificRequired(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  java_library {
   name: "foo",
   srcs: ["a.java"],
   hostdex: true,
   target: {
    hostdex: {
     required: ["libfoo"],
    },
   },
  }
 `)

 mod := ctx.ModuleForTests(t, "foo", "android_common").Module()
 info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)
 if len(info.ExtraInfo) != 1 {
  t.Fatalf("one extra entries is expected, but got %d", len(info.ExtraInfo))
 }

 mainEntries := &info.PrimaryInfo

 if r, ok := mainEntries.EntryMap["LOCAL_REQUIRED_MODULES"]; ok {
  t.Errorf("Unexpected required modules: %q", r)
 }

 if len(info.ExtraInfo) != 1 {
  t.Fatalf("one extra entries is expected, but got %d", len(info.ExtraInfo))
 }

 subEntries := &info.ExtraInfo[0]

 expected := []string{"libfoo"}
 actual := subEntries.EntryMap["LOCAL_REQUIRED_MODULES"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected required modules - expected: %q, actual: %q", expected, actual)
 }
}

func TestJavaSdkLibrary_RequireXmlPermissionFile(t *testing.T) {
 t.Parallel()
 result := android.GroupFixturePreparers(
  prepareForJavaAndroidMkTest,
  PrepareForTestWithJavaSdkLibraryFiles,
  FixtureWithLastReleaseApis("foo-shared_library", "foo-no_shared_library"),
 ).RunTestWithBp(t, `
  java_sdk_library {
   name: "foo-shared_library",
   srcs: ["a.java"],
  }
  java_sdk_library {
   name: "foo-no_shared_library",
   srcs: ["a.java"],
   shared_library: false,
  }
  `)

 // Verify the existence of internal modules
 result.ModuleForTests(t, "foo-shared_library.xml", "android_common")

 testCases := []struct {
  moduleName string
  expected   []string
 }{
  {"foo-shared_library", []string{"foo-shared_library.impl", "foo-shared_library.xml"}},
  {"foo-no_shared_library", []string{"foo-no_shared_library.impl"}},
 }
 for _, tc := range testCases {
  mod := result.ModuleForTests(t, tc.moduleName, "android_common").Module()
  info := android.AndroidMkInfoForTest(t, result.TestContext, mod)
  actual := info.PrimaryInfo.EntryMap["LOCAL_REQUIRED_MODULES"]
  if !reflect.DeepEqual(tc.expected, actual) {
   t.Errorf("Unexpected required modules - expected: %q, actual: %q", tc.expected, actual)
  }
 }
}

func TestImportSoongDexJar(t *testing.T) {
 t.Parallel()
 result := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  java_import {
   name: "my-java-import",
   jars: ["a.jar"],
   prefer: true,
   compile_dex: true,
  }
 `)

 mod := result.Module("my-java-import", "android_common")
 info := android.AndroidMkInfoForTest(t, result.TestContext, mod)
 expectedSoongDexJar := "out/soong/.intermediates/my-java-import/android_common/dex/my-java-import.jar"
 actualSoongDexJar := info.PrimaryInfo.EntryMap["LOCAL_SOONG_DEX_JAR"]

 android.AssertStringPathsRelativeToTopEquals(t, "LOCAL_SOONG_DEX_JAR", result.Config, []string{expectedSoongDexJar}, actualSoongDexJar)
}

func TestAndroidTestHelperApp_LocalDisableTestConfig(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(t, `
  android_test_helper_app {
   name: "foo",
   srcs: ["a.java"],
  }
 `)

 mod := ctx.ModuleForTests(t, "foo", "android_common").Module()
 info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)

 expected := []string{"true"}
 actual := info.PrimaryInfo.EntryMap["LOCAL_DISABLE_TEST_CONFIG"]
 if !reflect.DeepEqual(expected, actual) {
  t.Errorf("Unexpected flag value - expected: %q, actual: %q", expected, actual)
 }
}

func TestGetOverriddenPackages(t *testing.T) {
 t.Parallel()
 ctx := prepareForJavaAndroidMkTest.RunTestWithBp(
  t, `
  android_app {
   name: "foo",
   srcs: ["a.java"],
   sdk_version: "current",
   overrides: ["qux"]
  }

  override_android_app {
   name: "foo_override",
   base: "foo",
   overrides: ["bar"]
  }
  `)

 expectedVariants := []struct {
  name        string
  moduleName  string
  variantName string
  overrides   []string
 }{
  {
   name:        "foo",
   moduleName:  "foo",
   variantName: "android_common",
   overrides:   []string{"qux"},
  },
  {
   name:        "foo",
   moduleName:  "foo_override",
   variantName: "android_common_foo_override",
   overrides:   []string{"bar", "foo"},
  },
 }

 for _, expected := range expectedVariants {
  mod := ctx.ModuleForTests(t, expected.name, expected.variantName).Module()
  info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)
  actual := info.PrimaryInfo.EntryMap["LOCAL_OVERRIDES_PACKAGES"]

  android.AssertDeepEquals(t, "overrides property", expected.overrides, actual)
 }
}

func TestJniAsRequiredDeps(t *testing.T) {
 t.Parallel()
 ctx := android.GroupFixturePreparers(
  PrepareForTestWithJavaDefaultModules,
  cc.PrepareForTestWithCcDefaultModules,
  android.PrepareForTestWithAndroidMk,
 ).RunTestWithBp(t, `
  android_app {
   name: "app",
   jni_libs: ["libjni"],
   platform_apis: true,
  }

  android_app {
   name: "app_embedded",
   jni_libs: ["libjni"],
   platform_apis: true,
   use_embedded_native_libs: true,
  }

  cc_library {
   name: "libjni",
   system_shared_libs: [],
   stl: "none",
  }
  `)

 testcases := []struct {
  name     string
  expected []string
 }{
  {
   name:     "app",
   expected: []string{"libjni:64"},
  },
  {
   name:     "app_embedded",
   expected: nil,
  },
 }

 for _, tc := range testcases {
  mod := ctx.ModuleForTests(t, tc.name, "android_common").Module()
  info := android.AndroidMkInfoForTest(t, ctx.TestContext, mod)
  required := info.PrimaryInfo.EntryMap["LOCAL_REQUIRED_MODULES"]
  android.AssertDeepEquals(t, "unexpected required deps", tc.expected, required)
 }
}