build-system/gradle-core/src/main/java/com/android/build/gradle/internal/test/TestDataImpl.kt

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * 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 com.android.build.gradle.internal.test

import com.android.build.api.variant.impl.BuiltArtifactsImpl
import com.android.build.api.variant.impl.BuiltArtifactsLoaderImpl
import com.android.build.gradle.internal.component.DeviceTestCreationConfig
import com.android.build.gradle.internal.test.BuiltArtifactsSplitOutputMatcher.computeBestOutput
import com.android.builder.testing.api.DeviceConfigProvider
import org.gradle.api.file.Directory
import org.gradle.api.file.FileCollection
import org.gradle.api.provider.Provider
import org.gradle.api.tasks.Input
import java.io.File

/**
 * Implementation of [TestData] on top of a [DeviceTestCreationConfig]
 */
class TestDataImpl(
    namespace: Provider<String>,
    testConfig: DeviceTestCreationConfig,
    testApkDir: Provider<Directory>,
    testedApksDir: Provider<Directory>?,
    privacySandboxSdkApks: FileCollection?,
    privacySandboxCompatSdkApksDir: Provider<Directory>?,
    additionalSdkSupportedApkSplits: Provider<Directory>?,
    extraInstrumentationTestRunnerArgs: Map<String, String>
) : AbstractTestDataImpl(
    namespace,
    testConfig,
    testApkDir,
    testedApksDir,
    privacySandboxSdkApks,
    privacySandboxCompatSdkApksDir,
    additionalSdkSupportedApkSplits,
    extraInstrumentationTestRunnerArgs
) {
    @get: Input
    val supportedAbis: Set<String> =
        testConfig.nativeBuildCreationConfig?.supportedAbis ?: emptySet()


    override val libraryType =
        testConfig.services.provider { testConfig.mainVariant.componentType.isAar }

    override val testedApksFinder: ApksFinder
        get() = _testedApksFinder ?: TestedApksFinder(
                testedApksDir
                        ?.let { BuiltArtifactsLoaderImpl().load(it) },
                privacySandboxCompatSdkApks?.let {
                    if (it.isPresent) {
                        BuiltArtifactsLoaderImpl().load(it)
                    } else {
                        null
                    }
                },
                additionalSdkSupportedSplitApks?.let {
                    if (it.isPresent) {
                        BuiltArtifactsLoaderImpl().load(it)
                    } else {
                        null
                    }
                },
                supportedAbis
        )
                .also {
                    _testedApksFinder = it
                }

    private var _testedApksFinder: TestedApksFinder? = null

    internal class TestedApksFinder(
        private val testedApkBuiltArtifacts: BuiltArtifactsImpl?,
        private val privacySandboxCompatSdkApksBuiltArtifacts: BuiltArtifactsImpl?,
        private val additionalSdkSupportApkSplitsBuiltArtifacts: BuiltArtifactsImpl?,
        private val supportedAbis: Set<String>
    ): ApksFinder {

        override fun findApks(deviceConfigProvider: DeviceConfigProvider): List<File> {
            testedApkBuiltArtifacts ?: return emptyList()
            val apks = mutableListOf<File>()
            apks += computeBestOutput(deviceConfigProvider, testedApkBuiltArtifacts, supportedAbis)
            // Add additional splits
            if (deviceConfigProvider.supportsPrivacySandbox) {
                additionalSdkSupportApkSplitsBuiltArtifacts?.let {
                    apks += it.elements.map { File(it.outputFile) }
                }
            } else {
                privacySandboxCompatSdkApksBuiltArtifacts?.let {
                    apks += it.elements.map { File(it.outputFile) }
                }
            }
            return apks
        }
    }

}