1a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski/* 2a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * Copyright (C) 2016 The Android Open Source Project 3a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * 4a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * Licensed under the Apache License, Version 2.0 (the "License"); 5a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * you may not use this file except in compliance with the License. 6a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * You may obtain a copy of the License at 7a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * 8a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * http://www.apache.org/licenses/LICENSE-2.0 9a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * 10a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * Unless required by applicable law or agreed to in writing, software 11a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * distributed under the License is distributed on an "AS IS" BASIS, 12a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * See the License for the specific language governing permissions and 14a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * limitations under the License. 15a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski */ 16a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 17ae84a18a1fa5dc020fd25a963cb0955c8c31bfefSteve McKaypackage com.android.documentsui; 18a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 19a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.app.Activity; 20fa1e1360999a7ca6d78eb1a2fd560196fe0b4dbbBen Linimport android.app.UiAutomation; 21a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.content.ContentProviderClient; 22a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.content.ContentResolver; 23a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.content.Context; 24a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.content.Intent; 25ceedd1fb5c9e34096bbe1c11e79fa0d96471611eGarfield Tanimport android.os.Bundle; 26a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.os.RemoteException; 2709404b3e765ec99e4db30dbddf4aed5524b8523eSteve McKayimport android.provider.DocumentsContract; 28a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.provider.DocumentsContract.Document; 29a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.support.test.uiautomator.Configurator; 30a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.support.test.uiautomator.UiDevice; 31a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.support.test.uiautomator.UiObjectNotFoundException; 32a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.test.ActivityInstrumentationTestCase2; 33a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskiimport android.view.MotionEvent; 3409404b3e765ec99e4db30dbddf4aed5524b8523eSteve McKay 35b18a526eb3305dd5976e3563f8af31da53cb8cc9Ben Linimport com.android.documentsui.base.Features; 36d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKayimport com.android.documentsui.base.RootInfo; 3785ec0d676297724f211213c7cb188839f1d3601bSteve McKayimport com.android.documentsui.bots.Bots; 385e1acc9f71ca4c2a07ba64ec9c050b20303993e2Steve McKayimport com.android.documentsui.bots.UiBot; 39a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 4085ec0d676297724f211213c7cb188839f1d3601bSteve McKayimport javax.annotation.Nullable; 4185ec0d676297724f211213c7cb188839f1d3601bSteve McKay 42a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski/** 43a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * Provides basic test environment for UI tests: 44a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * - Launches activity 45a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * - Creates and gives access to test root directories and test files 46a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski * - Cleans up the test environment 47a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski */ 48a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewskipublic abstract class ActivityTest<T extends Activity> extends ActivityInstrumentationTestCase2<T> { 49a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 50a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski static final int TIMEOUT = 5000; 51a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 52a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski // Testing files. For custom ones, override initTestFiles(). 53a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String dirName1 = "Dir1"; 5485ec0d676297724f211213c7cb188839f1d3601bSteve McKay public static final String childDir1 = "ChildDir1"; 55a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String fileName1 = "file1.log"; 56a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String fileName2 = "file12.png"; 57a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String fileName3 = "anotherFile0.log"; 58a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String fileName4 = "poodles.text"; 59a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public static final String fileNameNoRename = "NO_RENAMEfile.txt"; 60a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 615e1acc9f71ca4c2a07ba64ec9c050b20303993e2Steve McKay public Bots bots; 62a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public UiDevice device; 63a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public Context context; 64fa1e1360999a7ca6d78eb1a2fd560196fe0b4dbbBen Lin public UiAutomation automation; 65a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 66b18a526eb3305dd5976e3563f8af31da53cb8cc9Ben Lin public Features features; 67a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public RootInfo rootDir0; 68a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public RootInfo rootDir1; 69d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKay protected ContentResolver mResolver; 70d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKay protected DocumentsProviderHelper mDocsHelper; 71d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKay protected ContentProviderClient mClient; 72a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 73a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public ActivityTest(Class<T> activityClass) { 74a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski super(activityClass); 75a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 76a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 77ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska /* 78ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska * Returns the root that will be opened within the activity. 79ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska * By default tests are started with one of the test roots. 80ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska * Override the method if you want to open different root on start. 81ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska * @return Root that will be opened. Return null if you want to open activity's default root. 82ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska */ 8385ec0d676297724f211213c7cb188839f1d3601bSteve McKay protected @Nullable RootInfo getInitialRoot() { 84ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska return rootDir0; 85ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska } 86ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska 87e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski /** 88e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski * Returns the authority of the testing provider begin used. 89e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski * By default it's StubProvider's authority. 90e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski * @return Authority of the provider. 91e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski */ 92e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski protected String getTestingProviderAuthority() { 93ae84a18a1fa5dc020fd25a963cb0955c8c31bfefSteve McKay return StubProvider.DEFAULT_AUTHORITY; 94e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski } 95e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski 96e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski /** 97e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski * Resolves testing roots. 98e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski */ 99e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski protected void setupTestingRoots() throws RemoteException { 100ae84a18a1fa5dc020fd25a963cb0955c8c31bfefSteve McKay rootDir0 = mDocsHelper.getRoot(StubProvider.ROOT_0_ID); 101ae84a18a1fa5dc020fd25a963cb0955c8c31bfefSteve McKay rootDir1 = mDocsHelper.getRoot(StubProvider.ROOT_1_ID); 102e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski } 103e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski 104a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski @Override 105a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public void setUp() throws Exception { 106a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski device = UiDevice.getInstance(getInstrumentation()); 107a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski // NOTE: Must be the "target" context, else security checks in content provider will fail. 108a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski context = getInstrumentation().getTargetContext(); 109fa1e1360999a7ca6d78eb1a2fd560196fe0b4dbbBen Lin automation = getInstrumentation().getUiAutomation(); 110b18a526eb3305dd5976e3563f8af31da53cb8cc9Ben Lin features = new Features.RuntimeFeatures(context.getResources(), null); 1115e1acc9f71ca4c2a07ba64ec9c050b20303993e2Steve McKay 112fa1e1360999a7ca6d78eb1a2fd560196fe0b4dbbBen Lin bots = new Bots(device, automation, context, TIMEOUT); 113a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 114a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski Configurator.getInstance().setToolType(MotionEvent.TOOL_TYPE_MOUSE); 115a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 116a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mResolver = context.getContentResolver(); 117e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski mClient = mResolver.acquireUnstableContentProviderClient(getTestingProviderAuthority()); 118e29e3416fddc34f72234a7ece191289ad3ba3a21Tomasz Mikolajewski mDocsHelper = new DocumentsProviderHelper(getTestingProviderAuthority(), mClient); 119a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 120ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin device.setOrientationNatural(); 1212fca4cc74ab4d3beb2e47447313d20ec13eb33baTomasz Mikolajewski setupTestingRoots(); 1222fca4cc74ab4d3beb2e47447313d20ec13eb33baTomasz Mikolajewski 123a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski launchActivity(); 124a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski resetStorage(); 125ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin 126ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin // Since at the launch of activity, ROOT_0 and ROOT_1 have no files, drawer will 127ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin // automatically open for phone devices. Espresso register click() as (x, y) MotionEvents, 128ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin // so if a drawer is on top of a file we want to select, it will actually click the drawer. 129ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin // Thus to start a clean state, we always try to close first. 130ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin bots.roots.closeDrawer(); 131ceedd1fb5c9e34096bbe1c11e79fa0d96471611eGarfield Tan 132ceedd1fb5c9e34096bbe1c11e79fa0d96471611eGarfield Tan // Configure the provider back to default. 133ceedd1fb5c9e34096bbe1c11e79fa0d96471611eGarfield Tan mDocsHelper.configure(null, Bundle.EMPTY); 134a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 135a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 136a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski @Override 137a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski public void tearDown() throws Exception { 138ebee3dec1da087499d69c53888fb7bb27200aa4fBen Lin device.unfreezeRotation(); 139a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mClient.release(); 140a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski super.tearDown(); 141a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 142a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 143db28d846c6421a2586a8d63d518803b347790e0bJulian Mancini protected void launchActivity() { 144a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski final Intent intent = context.getPackageManager().getLaunchIntentForPackage( 145a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski UiBot.TARGET_PKG); 146a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski intent.addFlags(Intent.FLAG_ACTIVITY_CLEAR_TASK | Intent.FLAG_ACTIVITY_NEW_TASK); 147ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska if (getInitialRoot() != null) { 148c5efea0795e914d411680704fb0f435a543d3907Garfield Tan intent.setAction(Intent.ACTION_VIEW); 149c5efea0795e914d411680704fb0f435a543d3907Garfield Tan intent.setDataAndType(getInitialRoot().getUri(), DocumentsContract.Root.MIME_TYPE_ITEM); 150ec401b7fba9eb822acaf2c2b57c590cf4828afb2Aga Wronska } 151a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski setActivityIntent(intent); 152a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski getActivity(); // Launch the activity. 153a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 154a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 155d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKay protected void resetStorage() throws RemoteException { 156a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mClient.call("clear", null, null); 157a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski device.waitForIdle(); 158a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 159a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 160d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKay protected void initTestFiles() throws RemoteException { 161a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createFolder(rootDir0, dirName1); 162a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createDocument(rootDir0, "text/plain", fileName1); 163a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createDocument(rootDir0, "image/png", fileName2); 164a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createDocumentWithFlags(rootDir0.documentId, "text/plain", fileNameNoRename, 165a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski Document.FLAG_SUPPORTS_WRITE); 166a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 167a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createDocument(rootDir1, "text/plain", fileName3); 168a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski mDocsHelper.createDocument(rootDir1, "text/plain", fileName4); 169a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 170a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 171a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski void assertDefaultContentOfTestDir0() throws UiObjectNotFoundException { 17285ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(fileName1); 17385ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(fileName2); 17485ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(dirName1); 17585ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(fileNameNoRename); 1765e1acc9f71ca4c2a07ba64ec9c050b20303993e2Steve McKay bots.directory.assertDocumentsCount(4); 177a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 178a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski 179a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski void assertDefaultContentOfTestDir1() throws UiObjectNotFoundException { 18085ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(fileName3); 18185ec0d676297724f211213c7cb188839f1d3601bSteve McKay bots.directory.waitForDocument(fileName4); 1825e1acc9f71ca4c2a07ba64ec9c050b20303993e2Steve McKay bots.directory.assertDocumentsCount(2); 183a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski } 184a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski} 185