1 /*
2  * Copyright (C) 2017 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.tradefed.targetprep.suite;
18 
19 import com.android.tradefed.config.OptionClass;
20 import com.android.tradefed.invoker.TestInformation;
21 import com.android.tradefed.result.error.InfraErrorIdentifier;
22 import com.android.tradefed.targetprep.TargetSetupError;
23 import com.android.tradefed.targetprep.TestAppInstallSetup;
24 
25 import java.io.File;
26 import java.io.FileNotFoundException;
27 
28 /**
29  * Installs specified APKs for Suite configuration: either from $ANDROID_TARGET_OUT_TESTCASES
30  * variable or the ROOT_DIR in build info.
31  */
32 @OptionClass(alias = "apk-installer")
33 public class SuiteApkInstaller extends TestAppInstallSetup {
34 
35     /** {@inheritDoc} */
36     @Override
getLocalPathForFilename(TestInformation testInfo, String apkFileName)37     protected File getLocalPathForFilename(TestInformation testInfo, String apkFileName)
38             throws TargetSetupError {
39         File apkFile = null;
40         try {
41             apkFile = testInfo.getDependencyFile(apkFileName, true);
42         } catch (FileNotFoundException e) {
43             throw new TargetSetupError(
44                     String.format("%s not found", apkFileName),
45                     e,
46                     testInfo.getDevice().getDeviceDescriptor(),
47                     InfraErrorIdentifier.ARTIFACT_NOT_FOUND);
48         }
49         return apkFile;
50     }
51 }
52