libraries/sts-common-util/device-side/src/com/android/sts/common/util/StsExtraBusinessLogicTestCase.java

/*
 * Copyright (C) 2021 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.sts.common.util;

import com.android.compatibility.common.util.ExtraBusinessLogicTestCase;

import android.os.Build;
import android.util.Log;

import androidx.test.InstrumentationRegistry;

import org.junit.Rule;
import org.junit.runner.Description;

import java.time.LocalDate;
import java.util.List;

/** The device-side implementation of StsLogic. */
public class StsExtraBusinessLogicTestCase extends ExtraBusinessLogicTestCase implements StsLogic {

    private LocalDate deviceSpl = null;
    @Rule public DescriptionProvider descriptionProvider = new DescriptionProvider();

    protected StsExtraBusinessLogicTestCase() {
        mDependentOnBusinessLogic = false;
    }

    @Override
    public List<String> getExtraBusinessLogics() {
        String stsDynamicPlan =
                InstrumentationRegistry.getArguments().getString("sts-dynamic-plan");
        switch (stsDynamicPlan) {
            case "incremental":
                return StsLogic.STS_EXTRA_BUSINESS_LOGIC_INCREMENTAL;
            case "full":
                return StsLogic.STS_EXTRA_BUSINESS_LOGIC_FULL;
            default:
                throw new RuntimeException(
                        "Could not find Dynamic STS plan in InstrumentationRegistry arguments");
        }
    }

    @Override
    public Description getTestDescription() {
        return descriptionProvider.getDescription();
    }

    @Override
    public LocalDate getDeviceSpl() {
        if (deviceSpl == null) {
            deviceSpl = SplUtils.localDateFromSplString(Build.VERSION.SECURITY_PATCH);
        }
        return deviceSpl;
    }

    /**
     * Specify the latest release bulletin. Control this from the command-line with the following:
     * --test-arg
     * com.android.tradefed.testtype.AndroidJUnitTest:instrumentation-arg:release-bulletin-spl:=2020-06
     */
    @Override
    public LocalDate getReleaseBulletinSpl() {
        String releaseBulletinSpl =
                InstrumentationRegistry.getArguments().getString("release-bulletin-spl");
        if (releaseBulletinSpl == null) {
            return null;
        }
        // bulletin is released by month; add any day - only the year and month are compared.
        releaseBulletinSpl =
                String.format("%s-%02d", releaseBulletinSpl, SplUtils.Type.PARTIAL.day);
        return SplUtils.localDateFromSplString(releaseBulletinSpl);
    }

    @Override
    public void logInfo(String logTag, String format, Object... args) {
        Log.i(logTag, String.format(format, args));
    }

    @Override
    public void logDebug(String logTag, String format, Object... args) {
        Log.d(logTag, String.format(format, args));
    }

    @Override
    public void logWarn(String logTag, String format, Object... args) {
        Log.w(logTag, String.format(format, args));
    }

    @Override
    public void logError(String logTag, String format, Object... args) {
        Log.e(logTag, String.format(format, args));
    }
}