/*
 * Copyright (C) 2025 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 android.content.pm.cts_root;

import static android.content.pm.cts_root.utils.Constants.MODE_CRASH;
import static android.content.pm.cts_root.utils.Constants.MODE_INCOMPLETE_NETWORK_UNAVAILABLE;
import static android.content.pm.cts_root.utils.Constants.MODE_INCOMPLETE_UNKNOWN;
import static android.content.pm.cts_root.utils.Constants.MODE_REJECT;
import static android.content.pm.cts_root.utils.Constants.MODE_REJECT_WITH_EXTENSION_RESPONSE;
import static android.content.pm.cts_root.utils.Constants.MODE_TIMEOUT;

import static com.google.common.truth.Truth.assertThat;

import android.content.pm.ApplicationInfo;
import android.content.pm.PackageInstaller;
import android.content.pm.PackageManager;
import android.os.Build;
import android.util.Pair;

/**
 * Privileged installer targeting 36 will only get a failure status code and the user action intent
 * as an extra in the result bundle.
 */
public class DeveloperVerificationPrivInstallerTarget36Test
        extends DeveloperVerificationInstallerTestBase {

    static {
        try {
            final ApplicationInfo installerInfo = sContext.getPackageManager().getPackageInfo(
                    sInstallerPackageName, /* flags= */ 0).applicationInfo;
            // Check that the target sdk of the installer matches with the expectation
            assertThat(installerInfo).isNotNull();
            assertThat(installerInfo.targetSdkVersion).isEqualTo(Build.VERSION_CODES.BAKLAVA);
        } catch (PackageManager.NameNotFoundException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    protected TestParams getInstallWhenVerifierCrashTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_CRASH);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE
                || sDefaultPolicy
                == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_BLOCK_FAIL_OPEN) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS);
        } else {
            // TODO(b/399436145): currently we haven't found a good way to trigger the verifier
            // service to disconnect which should simulate a crash. As a result, the verification
            // is considered timeout because the verifier doesn't respond in time. Fix this so that
            // the error message reflect the disconnection of the service.
            testParamsBuilder
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller.DEVELOPER_VERIFICATION_FAILED_REASON_UNKNOWN)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verification timed out; "
                            + "missing a response from the verifier within the time limit")
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }

    @Override
    protected TestParams getInstallWhenVerifierTimeoutTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_TIMEOUT);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE
                || sDefaultPolicy
                == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_BLOCK_FAIL_OPEN) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS);
        } else {
            testParamsBuilder
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller.DEVELOPER_VERIFICATION_FAILED_REASON_UNKNOWN)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verification timed out; missing a response from the "
                            + "verifier within the time limit")
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }

    @Override
    protected TestParams getInstallWhenVerifierIncompleteUnknownTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_INCOMPLETE_UNKNOWN);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS);
        } else {
            testParamsBuilder
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller.DEVELOPER_VERIFICATION_FAILED_REASON_UNKNOWN)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verification cannot be completed because of unknown reasons.")
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }

    @Override
    protected TestParams getInstallWithVerifierIncompleteNetworkUnavailableTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_INCOMPLETE_NETWORK_UNAVAILABLE);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS);
        } else {
            testParamsBuilder
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller
                                    .DEVELOPER_VERIFICATION_FAILED_REASON_NETWORK_UNAVAILABLE)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verification cannot be completed because of unavailable network.")
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }

    @Override
    protected TestParams getInstallWithVerifierRejectTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_REJECT);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS);
        } else {
            testParamsBuilder
                    .setVerifierRejectMessage(REJECT_MESSAGE)
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller.DEVELOPER_VERIFICATION_FAILED_REASON_DEVELOPER_BLOCKED)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verifier rejected the installation with message: "
                            + REJECT_MESSAGE)
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }

    @Override
    protected TestParams getInstallWithVerifierRejectWithExtensionResponseTestParams() {
        TestParams.TestParamsBuilder testParamsBuilder = getTestParamsBuilder()
                .setVerifierMode(MODE_REJECT_WITH_EXTENSION_RESPONSE);
        if (sDefaultPolicy == PackageInstaller.DEVELOPER_VERIFICATION_POLICY_NONE) {
            testParamsBuilder.setStatusCodeExpected(PackageInstaller.STATUS_SUCCESS)
                    .setExtensionResponseKeyValuePairExpected(
                            new Pair<>(EXTENSION_RESPONSE_KEY, EXTENSION_RESPONSE_VALUE));
        } else {
            testParamsBuilder.setVerifierRejectMessage(REJECT_MESSAGE)
                    .setExtensionResponseKeyValuePairExpected(
                            new Pair<>(EXTENSION_RESPONSE_KEY, EXTENSION_RESPONSE_VALUE))
                    .setStatusCodeExpected(PackageInstaller.STATUS_FAILURE_ABORTED)
                    .setVerificationFailureReasonExpected(
                            PackageInstaller.DEVELOPER_VERIFICATION_FAILED_REASON_DEVELOPER_BLOCKED)
                    .setErrorMessageExpected("INSTALL_FAILED_VERIFICATION_FAILURE: "
                            + "Verifier rejected the installation with message: "
                            + REJECT_MESSAGE)
                    .setHasUserActionIntentExpected(true);
        }
        return testParamsBuilder.build();
    }
}