| /* |
| * Copyright (C) 2016 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.documentsui; |
| |
| import static com.android.documentsui.StubProvider.DEFAULT_AUTHORITY; |
| import static com.android.documentsui.StubProvider.ROOT_0_ID; |
| import static com.android.documentsui.StubProvider.ROOT_1_ID; |
| |
| import android.annotation.Nullable; |
| import android.app.Activity; |
| import android.content.ContentProviderClient; |
| import android.content.ContentResolver; |
| import android.content.Context; |
| import android.content.Intent; |
| import android.os.RemoteException; |
| import android.provider.DocumentsContract.Document; |
| import android.support.test.uiautomator.Configurator; |
| import android.support.test.uiautomator.UiDevice; |
| import android.support.test.uiautomator.UiObjectNotFoundException; |
| import android.test.ActivityInstrumentationTestCase2; |
| import android.view.MotionEvent; |
| |
| import com.android.documentsui.bots.DirectoryListBot; |
| import com.android.documentsui.bots.KeyboardBot; |
| import com.android.documentsui.bots.RootsListBot; |
| import com.android.documentsui.bots.UiBot; |
| import com.android.documentsui.model.RootInfo; |
| |
| /** |
| * Provides basic test environment for UI tests: |
| * - Launches activity |
| * - Creates and gives access to test root directories and test files |
| * - Cleans up the test environment |
| */ |
| public abstract class ActivityTest<T extends Activity> extends ActivityInstrumentationTestCase2<T> { |
| |
| static final int TIMEOUT = 5000; |
| |
| // Testing files. For custom ones, override initTestFiles(). |
| public static final String dirName1 = "Dir1"; |
| public static final String fileName1 = "file1.log"; |
| public static final String fileName2 = "file12.png"; |
| public static final String fileName3 = "anotherFile0.log"; |
| public static final String fileName4 = "poodles.text"; |
| public static final String fileNameNoRename = "NO_RENAMEfile.txt"; |
| |
| public Bots bots; |
| public UiDevice device; |
| public Context context; |
| |
| public RootInfo rootDir0; |
| public RootInfo rootDir1; |
| |
| ContentResolver mResolver; |
| DocumentsProviderHelper mDocsHelper; |
| ContentProviderClient mClient; |
| |
| public ActivityTest(Class<T> activityClass) { |
| super(activityClass); |
| } |
| |
| /* |
| * Returns the root that will be opened within the activity. |
| * By default tests are started with one of the test roots. |
| * Override the method if you want to open different root on start. |
| * @return Root that will be opened. Return null if you want to open activity's default root. |
| */ |
| @Nullable |
| protected RootInfo getInitialRoot() { |
| return rootDir0; |
| } |
| |
| @Override |
| public void setUp() throws Exception { |
| device = UiDevice.getInstance(getInstrumentation()); |
| // NOTE: Must be the "target" context, else security checks in content provider will fail. |
| context = getInstrumentation().getTargetContext(); |
| |
| bots = new Bots(device, context, TIMEOUT); |
| |
| Configurator.getInstance().setToolType(MotionEvent.TOOL_TYPE_MOUSE); |
| |
| mResolver = context.getContentResolver(); |
| mClient = mResolver.acquireUnstableContentProviderClient(DEFAULT_AUTHORITY); |
| mDocsHelper = new DocumentsProviderHelper(DEFAULT_AUTHORITY, mClient); |
| |
| rootDir0 = mDocsHelper.getRoot(ROOT_0_ID); |
| rootDir1 = mDocsHelper.getRoot(ROOT_1_ID); |
| |
| launchActivity(); |
| resetStorage(); |
| } |
| |
| @Override |
| public void tearDown() throws Exception { |
| mClient.release(); |
| super.tearDown(); |
| } |
| |
| void launchActivity() { |
| final Intent intent = context.getPackageManager().getLaunchIntentForPackage( |
| UiBot.TARGET_PKG); |
| intent.addFlags(Intent.FLAG_ACTIVITY_CLEAR_TASK | Intent.FLAG_ACTIVITY_NEW_TASK); |
| if (getInitialRoot() != null) { |
| intent.setData(getInitialRoot().getUri()); |
| } |
| setActivityIntent(intent); |
| getActivity(); // Launch the activity. |
| } |
| |
| void resetStorage() throws RemoteException { |
| mClient.call("clear", null, null); |
| device.waitForIdle(); |
| } |
| |
| void initTestFiles() throws RemoteException { |
| mDocsHelper.createFolder(rootDir0, dirName1); |
| mDocsHelper.createDocument(rootDir0, "text/plain", fileName1); |
| mDocsHelper.createDocument(rootDir0, "image/png", fileName2); |
| mDocsHelper.createDocumentWithFlags(rootDir0.documentId, "text/plain", fileNameNoRename, |
| Document.FLAG_SUPPORTS_WRITE); |
| |
| mDocsHelper.createDocument(rootDir1, "text/plain", fileName3); |
| mDocsHelper.createDocument(rootDir1, "text/plain", fileName4); |
| } |
| |
| void assertDefaultContentOfTestDir0() throws UiObjectNotFoundException { |
| bots.directory.assertDocumentsCount(4); |
| bots.directory.assertDocumentsPresent(fileName1, fileName2, dirName1, fileNameNoRename); |
| } |
| |
| void assertDefaultContentOfTestDir1() throws UiObjectNotFoundException { |
| bots.directory.assertDocumentsCount(2); |
| bots.directory.assertDocumentsPresent(fileName3, fileName4); |
| } |
| |
| /** |
| * Handy collection of bots for working with Files app. |
| */ |
| public static final class Bots { |
| public final UiBot main; |
| public final RootsListBot roots; |
| public final DirectoryListBot directory; |
| public final KeyboardBot keyboard; |
| |
| private Bots(UiDevice device, Context context, int timeout) { |
| this.main = new UiBot(device, context, TIMEOUT); |
| this.roots = new RootsListBot(device, context, TIMEOUT); |
| this.directory = new DirectoryListBot(device, context, TIMEOUT); |
| this.keyboard = new KeyboardBot(device, context, TIMEOUT); |
| } |
| } |
| } |