Add test for cancellig copy / move by Notification (418e5a3e) · Commits · e / os / android_packages_apps_DocumentsUI

tests/AndroidManifest.xml

+9 −0

Original line number	Diff line number	Diff line
		@@ -100,6 +100,15 @@
		<action android:name="android.content.action.DOCUMENTS_PROVIDER" />
		</intent-filter>
		</provider>

		<service android:name="com.android.documentsui.services.TestNotificationService"
		android:permission="android.permission.BIND_NOTIFICATION_LISTENER_SERVICE"
		android:process="com.android.documentsui">
		<intent-filter>
		<action android:name="android.service.notification.NotificationListenerService" />
		</intent-filter>
		</service>

		</application>

		<instrumentation android:name="android.support.test.runner.AndroidJUnitRunner"

tests/common/com/android/documentsui/DocumentsProviderHelper.java

+9 −0

Original line number	Diff line number	Diff line
		@@ -144,6 +144,15 @@ public class DocumentsProviderHelper {
		waitForWrite();
		}

		public void writeAppendDocument(Uri documentUri, byte[] contents)
		throws RemoteException, IOException {
		ParcelFileDescriptor file = mClient.openFile(documentUri, "wa", null);
		try (AutoCloseOutputStream out = new AutoCloseOutputStream(file)) {
		out.write(contents);
		}
		waitForWrite();
		}

		public void waitForWrite() throws RemoteException {
		mClient.call("waitForWrite", null, null);
		}

tests/common/com/android/documentsui/StubProvider.java

+11 −2

Original line number	Diff line number	Diff line
		@@ -74,7 +74,7 @@ public class StubProvider extends DocumentsProvider {
		private static final String TAG = "StubProvider";

		private static final String STORAGE_SIZE_KEY = "documentsui.stubprovider.size";
		private static int DEFAULT_ROOT_SIZE = 1024 * 1024 * 100; // 100 MB.
		private static int DEFAULT_ROOT_SIZE = 1024 * 1024 * 500; // 500 MB.

		private static final String[] DEFAULT_ROOT_PROJECTION = new String[] {
		Root.COLUMN_ROOT_ID, Root.COLUMN_FLAGS, Root.COLUMN_TITLE, Root.COLUMN_DOCUMENT_ID,
		@@ -357,6 +357,10 @@ public class StubProvider extends DocumentsProvider {
		if ("w".equals(mode)) {
		return startWrite(document);
		}
		if ("wa".equals(mode)) {
		return startWrite(document, true);
		}


		throw new FileNotFoundException();
		}
		@@ -423,6 +427,11 @@ public class StubProvider extends DocumentsProvider {

		private ParcelFileDescriptor startWrite(final StubDocument document)
		throws FileNotFoundException {
		return startWrite(document, false);
		}

		private ParcelFileDescriptor startWrite(final StubDocument document, boolean append)
		throws FileNotFoundException {
		ParcelFileDescriptor[] pipe;
		try {
		pipe = ParcelFileDescriptor.createReliablePipe();
		@@ -438,7 +447,7 @@ public class StubProvider extends DocumentsProvider {
		try {
		Log.d(TAG, "Opening write stream on file " + document.documentId);
		inputStream = new ParcelFileDescriptor.AutoCloseInputStream(readPipe);
		outputStream = new FileOutputStream(document.file);
		outputStream = new FileOutputStream(document.file, append);
		byte[] buffer = new byte[32 * 1024];
		int bytesToRead;
		int bytesRead = 0;

tests/common/com/android/documentsui/bots/UiBot.java

+1 −1

Original line number	Diff line number	Diff line
		@@ -229,7 +229,7 @@ public class UiBot extends Bots.BaseBot {
		onView(withId(android.R.id.button2)).perform(click());
		}

		UiObject findMenuLabelWithName(String label) {
		public UiObject findMenuLabelWithName(String label) {
		UiSelector selector = new UiSelector().text(label);
		return mDevice.findObject(selector);
		}

tests/common/com/android/documentsui/services/TestNotificationService.java

0 → 100644

+211 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright (C) 2017 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.services;

		import android.app.Notification;
		import android.app.PendingIntent;
		import android.content.Context;
		import android.content.Intent;
		import android.content.IntentFilter;
		import android.content.BroadcastReceiver;
		import android.service.notification.NotificationListenerService;
		import android.service.notification.StatusBarNotification;
		import android.view.View;
		import android.view.ViewGroup;
		import android.widget.FrameLayout;
		import android.widget.ProgressBar;
		import android.widget.RemoteViews;


		/**
		* This class receives a callback when Notification is posted or removed
		* and monitors the Notification status.
		* And, this sends the operation's result by Broadcast.
		*/
		public class TestNotificationService extends NotificationListenerService {
		public static final String ACTION_CHANGE_CANCEL_MODE =
		"com.android.documentsui.services.TestNotificationService.ACTION_CHANGE_CANCEL_MODE";

		public static final String ACTION_CHANGE_EXECUTION_MODE =
		"com.android.documentsui.services.TestNotificationService.ACTION_CHANGE_EXECUTION_MODE";

		public static final String ACTION_OPERATION_RESULT =
		"com.android.documentsui.services.TestNotificationService.ACTION_OPERATION_RESULT";

		public static final String EXTRA_RESULT =
		"com.android.documentsui.services.TestNotificationService.EXTRA_RESULT";

		public static final String EXTRA_ERROR_REASON =
		"com.android.documentsui.services.TestNotificationService.EXTRA_ERROR_REASON";

		public enum MODE {
		CANCEL_MODE,
		EXECUTION_MODE;
		}

		private String DOCUMENTSUI= "com.android.documentsui";

		private FrameLayout mFrameLayout = null;

		private ProgressBar mProgressBar = null;

		private MODE mCurrentMode = MODE.CANCEL_MODE;

		private boolean mCancelled = false;

		private final BroadcastReceiver mReceiver = new BroadcastReceiver() {
		@Override
		public void onReceive(Context context, Intent intent) {
		String action = intent.getAction();
		if (ACTION_CHANGE_CANCEL_MODE.equals(action)) {
		mCurrentMode = MODE.CANCEL_MODE;
		} else if (ACTION_CHANGE_EXECUTION_MODE.equals(action)) {
		mCurrentMode = MODE.EXECUTION_MODE;
		}
		}
		};

		@Override
		public void onCreate() {
		mFrameLayout = new FrameLayout(getBaseContext());
		IntentFilter filter = new IntentFilter();
		filter.addAction(ACTION_CHANGE_CANCEL_MODE);
		filter.addAction(ACTION_CHANGE_EXECUTION_MODE);
		registerReceiver(mReceiver, filter);
		}

		@Override
		public int onStartCommand(Intent intent, int flags, int startId) {
		return START_STICKY;
		}

		@Override
		public void onDestroy() {
		unregisterReceiver(mReceiver);
		mProgressBar = null;
		mFrameLayout.removeAllViews();
		mFrameLayout = null;
		}

		@Override
		public void onNotificationPosted(StatusBarNotification sbn) {
		String pkgName = sbn.getPackageName();
		if (!pkgName.equals(DOCUMENTSUI)) {
		return;
		}

		if (MODE.CANCEL_MODE.equals(mCurrentMode)) {
		mCancelled = doCancel(sbn.getNotification());
		}
		}

		@Override
		public void onNotificationRemoved(StatusBarNotification sbn) {
		String pkgName = sbn.getPackageName();
		if (!pkgName.equals(DOCUMENTSUI)) {
		return;
		}

		Intent intent = new Intent(ACTION_OPERATION_RESULT);
		if (MODE.CANCEL_MODE.equals(mCurrentMode)) {
		intent.putExtra(EXTRA_RESULT, mCancelled);
		if (!mCancelled) {
		intent.putExtra(EXTRA_ERROR_REASON, "Cannot executed cancel");
		}
		} else if (MODE.EXECUTION_MODE.equals(mCurrentMode)) {
		boolean isStartProgress = isStartProgress(sbn.getNotification());
		intent.putExtra(EXTRA_RESULT, isStartProgress);
		if (!isStartProgress) {
		intent.putExtra(EXTRA_ERROR_REASON, "Progress does not displayed correctly.");
		}
		}
		sendBroadcast(intent);
		}

		private boolean doCancel(Notification noti) {
		if (!isStartProgress(noti)) {
		return false;
		}

		Notification.Action aList [] = noti.actions;
		if (aList == null) {
		return false;
		}

		boolean result = false;
		for (Notification.Action item : aList) {
		if (item.title.equals("Cancel")) {
		try {
		item.actionIntent.send();
		result = true;
		} catch (PendingIntent.CanceledException e) {
		}
		}
		}
		return result;
		}

		private boolean isStartProgress(Notification notifiction) {
		ProgressBar progressBar = getProgresssBar(getRemoteViews(notifiction));
		return (progressBar != null) ? progressBar.getProgress() > 0 : false;
		}

		private RemoteViews getRemoteViews(Notification notifiction) {
		Notification.Builder builder = Notification.Builder.recoverBuilder(
		getBaseContext(), notifiction);
		if (builder == null) {
		return null;
		}

		return builder.createContentView();
		}

		private ProgressBar getProgresssBar(RemoteViews remoteViews) {
		if (remoteViews == null) {
		return null;
		}

		View view = remoteViews.apply(getBaseContext(), mFrameLayout);
		return getProgressBarImpl(view);
		}

		private ProgressBar getProgressBarImpl(View view) {
		if (view == null \|\| !(view instanceof ViewGroup)) {
		return null;
		}

		ViewGroup viewGroup = (ViewGroup)view;
		if (viewGroup.getChildCount() <= 0) {
		return null;
		}

		ProgressBar result = null;
		for (int i = 0; i < viewGroup.getChildCount(); i++) {
		View v = viewGroup.getChildAt(i);
		if (v instanceof ProgressBar) {
		result = ((ProgressBar)v);
		break;
		} else if (v instanceof ViewGroup) {
		result = getProgressBarImpl(v);
		if (result != null) {
		break;
		}
		}
		}
		return result;
		}
		}