// 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 testsuites

import (
 "android/soong/android"
 "testing"
)

func TestBuildTestList(t *testing.T) {
 t.Parallel()
 ctx := android.GroupFixturePreparers(
  prepareForFakeTestSuite,
  android.PrepareForTestWithHostTools("soong_zip"),
  android.FixtureRegisterWithContext(func(ctx android.RegistrationContext) {
   ctx.RegisterParallelSingletonType("testsuites", testSuiteFilesFactory)
  }),
 ).RunTestWithBp(t, `
  fake_module {
   name: "module1",
   outputs: [
    "Test1/Test1.config",
    "Test1/Test1.apk",
   ],
   test_suites: ["ravenwood-tests"],
  }
  fake_module {
   name: "module2",
   outputs: [
    "Test2/Test21/Test21.config",
    "Test2/Test21/Test21.apk",
   ],
   test_suites: ["ravenwood-tests", "robolectric-tests"],
  }
  fake_module {
   name: "module_without_config",
   outputs: [
    "BadTest/BadTest.jar",
   ],
   test_suites: ["robolectric-tests"],
  }
 `)

 config := ctx.SingletonForTests(t, "testsuites")

 wantContents := map[string]string{
  "out/soong/packaging/robolectric-tests_list": `host/testcases/Test2/Test21/Test21.config
`,
  "out/soong/packaging/ravenwood-tests_list": `host/testcases/Test1/Test1.config
host/testcases/Test2/Test21/Test21.config
`,
 }
 for file, want := range wantContents {
  got := android.ContentFromFileRuleForTests(t, ctx.TestContext, config.Output(file))

  if want != got {
   t.Errorf("want %q, got %q", want, got)
  }
 }
}

type fake_module struct {
 android.ModuleBase
 props struct {
  Outputs     []string
  Test_suites []string
 }
}

func fakeTestSuiteFactory() android.Module {
 module := &fake_module{}
 module.AddProperties(&module.props)
 android.InitAndroidModule(module)
 return module
}

var prepareForFakeTestSuite = android.GroupFixturePreparers(
 android.FixtureRegisterWithContext(func(ctx android.RegistrationContext) {
  ctx.RegisterModuleType("fake_module", fakeTestSuiteFactory)
 }),
)

func (f *fake_module) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 for _, output := range f.props.Outputs {
  f := android.PathForModuleOut(ctx, output)
  ctx.InstallFile(pathForTestCases(ctx), output, f)
 }

 android.SetProvider(ctx, android.TestSuiteInfoProvider, android.TestSuiteInfo{
  TestSuites: f.TestSuites(),
 })
}

func (f *fake_module) TestSuites() []string {
 return f.props.Test_suites
}