1 /*
2  * Copyright (C) 2020 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 android.scopedstorage.cts.host;
18 
19 import static com.google.common.truth.Truth.assertThat;
20 
21 import com.android.tradefed.device.ITestDevice;
22 import com.android.tradefed.testtype.DeviceJUnit4ClassRunner;
23 
24 import org.junit.AfterClass;
25 import org.junit.Before;
26 import org.junit.runner.RunWith;
27 
28 @RunWith(DeviceJUnit4ClassRunner.class)
29 public class PublicVolumeLegacyHostTest extends LegacyStorageHostTest {
30     /** Used to clean up the virtual volume after the test */
31     private static ITestDevice sDevice = null;
32     private boolean mIsPublicVolumeSetup = false;
33 
setupNewPublicVolume()34     private void setupNewPublicVolume() throws Exception {
35         if (!mIsPublicVolumeSetup) {
36             assertThat(runDeviceTests("android.scopedstorage.cts",
37                     "android.scopedstorage.cts.PublicVolumeTestHelper", "setupNewPublicVolume"))
38                             .isTrue();
39             mIsPublicVolumeSetup = true;
40         }
41     }
42 
setupDevice()43     private void setupDevice() {
44         if (sDevice == null) {
45             sDevice = getDevice();
46         }
47     }
48 
49     /**
50      * Runs the given phase of PublicVolumeTest by calling into the device.
51      * Throws an exception if the test phase fails.
52      */
53     @Override
runDeviceTest(String phase)54     void runDeviceTest(String phase) throws Exception {
55         assertThat(runDeviceTests("android.scopedstorage.cts.legacy",
56                 "android.scopedstorage.cts.legacy.PublicVolumeLegacyTest", phase)).isTrue();
57     }
58 
59     @Before
setup()60     public void setup() throws Exception {
61         setupDevice();
62         setupNewPublicVolume();
63         super.setup();
64     }
65 
66     @AfterClass
deletePublicVolumes()67     public static void deletePublicVolumes() throws Exception {
68         if (sDevice != null) {
69             sDevice.executeShellCommand("sm set-virtual-disk false");
70         }
71     }
72 }
73