summaryrefslogtreecommitdiff
path: root/libraries/sts-common-util/host-side/src/com/android/sts/common/tradefed/testtype/StsExtraBusinessLogicHostTestBase.java
blob: 80c5cc95bbc221d2d4e2037141d056caea5b2644 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
 * 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.tradefed.testtype;

import com.android.compatibility.common.tradefed.testtype.ExtraBusinessLogicHostTestBase;
import com.android.ddmlib.Log;
import com.android.sts.common.util.DescriptionProvider;
import com.android.sts.common.util.SplUtils;
import com.android.sts.common.util.StsLogic;
import com.android.sts.common.util.UnameVersionHost;
import com.android.tradefed.device.DeviceNotAvailableException;

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

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

/** The host-side implementation of StsLogic. */
public class StsExtraBusinessLogicHostTestBase extends ExtraBusinessLogicHostTestBase
        implements StsLogic {

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

    public StsExtraBusinessLogicHostTestBase() {
        super();
        mDependentOnBusinessLogic = false;
    }

    @Override
    public List<String> getExtraBusinessLogics() {
        // set in test/sts/tools/sts-tradefed/res/config/sts-base-dynamic-*.xml
        String stsDynamicPlan = getBuild().getBuildAttributes().get("sts-dynamic-plan");
        return StsLogic.getExtraBusinessLogicForPlan(stsDynamicPlan);
    }

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

    @Override
    public LocalDate getPlatformSpl() {
        if (deviceSpl == null) {
            try {
                String splString = getDevice().getProperty("ro.build.version.security_patch");
                deviceSpl = SplUtils.localDateFromSplString(splString);
            } catch (DeviceNotAvailableException e) {
                throw new RuntimeException("couldn't get the security patch level", e);
            }
        }
        return deviceSpl;
    }

    @Override
    public Optional<LocalDate> getKernelBuildDate() {
        try {
            return new UnameVersionHost(getDevice()).parseBuildTimestamp();
        } catch (DeviceNotAvailableException e) {
            throw new RuntimeException("Couldn't get the kernel build timestamp", e);
        }
    }

    @Override
    public boolean shouldUseKernelSpl() {
        // set in test/sts/tools/sts-tradefed/res/config/sts-base-use-kernel-spl.xml
        String useKernelSplString = getBuild().getBuildAttributes().get("sts-use-kernel-spl");
        return Boolean.parseBoolean(useKernelSplString);
    }

    @Override
    public boolean shouldSkipMainlineTests() {
        return Boolean.parseBoolean(
                getBuild().getBuildAttributes().get("sts-should-skip-mainline"));
    }

    /**
     * Specify the latest release bulletin. Control this from the command-line with the following
     * command line argument: --build-attribute "release-bulletin-spl=2021-06"
     */
    @Override
    public LocalDate getReleaseBulletinSpl() {
        // set manually with command-line args at runtime
        String releaseBulletinSpl = getBuild().getBuildAttributes().get("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));
    }
}