// Copyright 2021 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 (
 "sort"
 "strings"
 "testing"

 "android/soong/android"
)

func intPtr(v int) *int {
 return &v
}

func TestPrebuiltApis_SystemModulesCreation(t *testing.T) {
 t.Parallel()
 result := android.GroupFixturePreparers(
  prepareForJavaTest,
  FixtureWithPrebuiltApis(map[string][]string{
   "31":      {},
   "32":      {},
   "current": {},
  }),
 ).RunTest(t)

 sdkSystemModules := []string{}
 result.VisitAllModules(func(module android.Module) {
  name := android.RemoveOptionalPrebuiltPrefix(module.Name())
  if strings.HasPrefix(name, "sdk_") && strings.HasSuffix(name, "_system_modules") {
   sdkSystemModules = append(sdkSystemModules, name)
  }
 })
 sort.Strings(sdkSystemModules)
 expected := []string{
  // 31 only has public system modules.
  "sdk_public_31_system_modules",

  // 32 and current both have public and module-lib system modules.
  "sdk_public_32_system_modules",
  "sdk_module-lib_32_system_modules",
  "sdk_public_current_system_modules",
  "sdk_module-lib_current_system_modules",
 }
 sort.Strings(expected)
 android.AssertArrayString(t, "sdk system modules", expected, sdkSystemModules)
}

func TestPrebuiltApis_WithExtensions(t *testing.T) {
 t.Parallel()
 runTestWithBaseExtensionLevel := func(v int) (foo_input, bar_input, baz_input string) {
  result := android.GroupFixturePreparers(
   prepareForJavaTest,
   android.FixtureModifyProductVariables(func(variables android.FixtureProductVariables) {
    variables.Platform_base_sdk_extension_version = intPtr(v)
   }),
   FixtureWithPrebuiltApisAndExtensions(map[string][]string{
    "31":      {"foo"},
    "32":      {"foo", "bar", "baz"},
    "current": {"foo", "bar"},
   }, map[string][]string{
    "1": {"foo"},
    "2": {"foo", "bar"},
   }),
  ).RunTest(t)
  foo_input = result.ModuleForTests(t, "foo.api.public.latest", "").Rule("generator").Implicits[0].String()
  bar_input = result.ModuleForTests(t, "bar.api.public.latest", "").Rule("generator").Implicits[0].String()
  baz_input = result.ModuleForTests(t, "baz.api.public.latest", "").Rule("generator").Implicits[0].String()
  return
 }
 // Extension 2 is the latest for both foo and bar, finalized after the base extension version.
 foo_input, bar_input, baz_input := runTestWithBaseExtensionLevel(1)
 android.AssertStringEquals(t, "Expected latest foo = extension level 2", "prebuilts/sdk/extensions/2/public/api/foo.txt", foo_input)
 android.AssertStringEquals(t, "Expected latest bar = extension level 2", "prebuilts/sdk/extensions/2/public/api/bar.txt", bar_input)
 android.AssertStringEquals(t, "Expected latest baz = api level 32", "prebuilts/sdk/32/public/api/baz.txt", baz_input)

 // Extension 2 is the latest for both foo and bar, finalized together with 32
 foo_input, bar_input, baz_input = runTestWithBaseExtensionLevel(2)
 android.AssertStringEquals(t, "Expected latest foo = extension level 2", "prebuilts/sdk/extensions/2/public/api/foo.txt", foo_input)
 android.AssertStringEquals(t, "Expected latest bar = extension level 2", "prebuilts/sdk/extensions/2/public/api/bar.txt", bar_input)
 android.AssertStringEquals(t, "Expected latest baz = api level 32", "prebuilts/sdk/32/public/api/baz.txt", baz_input)

 // Extension 3 is the current extension, but it has not yet been finalized.
 foo_input, bar_input, baz_input = runTestWithBaseExtensionLevel(3)
 android.AssertStringEquals(t, "Expected latest foo = extension level 2", "prebuilts/sdk/extensions/2/public/api/foo.txt", foo_input)
 android.AssertStringEquals(t, "Expected latest bar = extension level 2", "prebuilts/sdk/extensions/2/public/api/bar.txt", bar_input)
 android.AssertStringEquals(t, "Expected latest baz = api level 32", "prebuilts/sdk/32/public/api/baz.txt", baz_input)
}

func TestPrebuiltApis_WithMixedVersionCodes(t *testing.T) {
 t.Parallel()
 runTestWithIncrementalApi := func() (foo_input, bar_input, baz_input string) {
  result := android.GroupFixturePreparers(
   prepareForJavaTest,
   FixtureWithPrebuiltApis(map[string][]string{
    "33.0":    {"foo"},
    "34":      {"foo", "bar", "baz"},
    "34.1":    {"foo", "bar"},
    "current": {"foo", "bar"},
   }),
  ).RunTest(t)
  foo_input = result.ModuleForTests(t, "foo.api.public.latest", "").Rule("generator").Implicits[0].String()
  bar_input = result.ModuleForTests(t, "bar.api.public.latest", "").Rule("generator").Implicits[0].String()
  baz_input = result.ModuleForTests(t, "baz.api.public.latest", "").Rule("generator").Implicits[0].String()
  return
 }
 // 34 is the latest for baz, 34.1 is the latest for both foo & bar
 foo_input, bar_input, baz_input := runTestWithIncrementalApi()
 android.AssertStringEquals(t, "Expected latest foo = api level 34.1", "prebuilts/sdk/34.1/public/api/foo.txt", foo_input)
 android.AssertStringEquals(t, "Expected latest bar = api level 34.1", "prebuilts/sdk/34.1/public/api/bar.txt", bar_input)
 android.AssertStringEquals(t, "Expected latest baz = api level 34", "prebuilts/sdk/34/public/api/baz.txt", baz_input)
}