254 lines
8.2 KiB
Java
254 lines
8.2 KiB
Java
/*
|
|
* 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;
|
|
|
|
import static com.android.documentsui.StubProvider.EXTRA_SIZE;
|
|
import static com.android.documentsui.StubProvider.ROOT_0_ID;
|
|
import static com.android.documentsui.StubProvider.ROOT_1_ID;
|
|
|
|
import android.content.BroadcastReceiver;
|
|
import android.content.Context;
|
|
import android.content.Intent;
|
|
import android.content.IntentFilter;
|
|
import android.net.Uri;
|
|
import android.os.Bundle;
|
|
import android.os.RemoteException;
|
|
import android.util.Log;
|
|
|
|
import androidx.test.filters.LargeTest;
|
|
|
|
import com.android.documentsui.files.FilesActivity;
|
|
import com.android.documentsui.filters.HugeLongTest;
|
|
import com.android.documentsui.services.TestNotificationService;
|
|
|
|
import java.util.concurrent.CountDownLatch;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
/**
|
|
* This class tests the below points.
|
|
* - Cancel copying or moving file before starting it.
|
|
* - Cancel during copying or moving file.
|
|
*/
|
|
@LargeTest
|
|
public class CancelFromNotificationUiTest extends ActivityTest<FilesActivity> {
|
|
private static final String TAG = "CancelFromNotificationUiTest";
|
|
|
|
private static final String TARGET_FILE = "stub.data";
|
|
|
|
private static final int BUFFER_SIZE = 10 * 1024 * 1024;
|
|
|
|
private static final int WAIT_TIME_SECONDS = 120;
|
|
|
|
private final BroadcastReceiver mReceiver = new BroadcastReceiver() {
|
|
@Override
|
|
public void onReceive(Context context, Intent intent) {
|
|
String action = intent.getAction();
|
|
if (TestNotificationService.ACTION_OPERATION_RESULT.equals(action)) {
|
|
mOperationExecuted = intent.getBooleanExtra(
|
|
TestNotificationService.EXTRA_RESULT, false);
|
|
if (!mOperationExecuted) {
|
|
mErrorReason = intent.getStringExtra(
|
|
TestNotificationService.EXTRA_ERROR_REASON);
|
|
}
|
|
mCountDownLatch.countDown();
|
|
}
|
|
}
|
|
};
|
|
|
|
private CountDownLatch mCountDownLatch;
|
|
|
|
private boolean mOperationExecuted;
|
|
|
|
private String mErrorReason;
|
|
|
|
public CancelFromNotificationUiTest() {
|
|
super(FilesActivity.class);
|
|
}
|
|
|
|
@Override
|
|
public void setUp() throws Exception {
|
|
super.setUp();
|
|
|
|
// super.setUp() method will change the storage size to 100MB.
|
|
// So, reset the storage size again to 500MB.
|
|
Bundle bundle = new Bundle();
|
|
bundle.putLong(EXTRA_SIZE, 500L);
|
|
// Set a flag to prevent many refreshes.
|
|
bundle.putBoolean(StubProvider.EXTRA_ENABLE_ROOT_NOTIFICATION, false);
|
|
mDocsHelper.configure(null, bundle);
|
|
|
|
try {
|
|
bots.notifications.setNotificationAccess(getActivity(), true);
|
|
} catch (Exception e) {
|
|
Log.d(TAG, "Cannot set notification access. ", e);
|
|
}
|
|
|
|
initTestFiles();
|
|
|
|
IntentFilter filter = new IntentFilter();
|
|
filter.addAction(TestNotificationService.ACTION_OPERATION_RESULT);
|
|
context.registerReceiver(mReceiver, filter);
|
|
context.sendBroadcast(new Intent(
|
|
TestNotificationService.ACTION_CHANGE_CANCEL_MODE));
|
|
|
|
mOperationExecuted = false;
|
|
mErrorReason = "No response from Notification";
|
|
mCountDownLatch = new CountDownLatch(1);
|
|
}
|
|
|
|
@Override
|
|
public void tearDown() throws Exception {
|
|
mCountDownLatch.countDown();
|
|
mCountDownLatch = null;
|
|
|
|
context.unregisterReceiver(mReceiver);
|
|
try {
|
|
bots.notifications.setNotificationAccess(getActivity(), false);
|
|
} catch (Exception e) {
|
|
Log.d(TAG, "Cannot set notification access. ", e);
|
|
}
|
|
super.tearDown();
|
|
}
|
|
|
|
@Override
|
|
public void initTestFiles() throws RemoteException {
|
|
try {
|
|
createStubFile();
|
|
} catch (Exception e) {
|
|
fail("Initialization failed. " + e.toString());
|
|
}
|
|
}
|
|
|
|
private void createStubFile() throws Exception {
|
|
Uri uri = mDocsHelper.createDocument(rootDir0, "*/*", TARGET_FILE);
|
|
byte[] stubByte = new byte[BUFFER_SIZE];
|
|
mDocsHelper.writeDocument(uri, stubByte);
|
|
for (int i = 0; i < 49; i++) {
|
|
stubByte = null;
|
|
stubByte = new byte[BUFFER_SIZE];
|
|
mDocsHelper.writeAppendDocument(uri, stubByte, stubByte.length);
|
|
}
|
|
}
|
|
|
|
@HugeLongTest
|
|
public void testCopyDocument_Cancel() throws Exception {
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
|
|
bots.directory.findDocument(TARGET_FILE);
|
|
device.waitForIdle();
|
|
|
|
bots.directory.selectDocument(TARGET_FILE, 1);
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickToolbarOverflowItem(context.getResources().getString(R.string.menu_copy));
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickDialogCancelButton();
|
|
device.waitForIdle();
|
|
|
|
bots.directory.waitForDocument(TARGET_FILE);
|
|
}
|
|
|
|
@HugeLongTest
|
|
public void testCopyDocument_CancelFromNotification() throws Exception {
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
bots.directory.findDocument(TARGET_FILE);
|
|
device.waitForIdle();
|
|
|
|
bots.directory.selectDocument(TARGET_FILE, 1);
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickToolbarOverflowItem(context.getResources().getString(R.string.menu_copy));
|
|
device.waitForIdle();
|
|
|
|
bots.roots.openRoot(ROOT_1_ID);
|
|
bots.main.clickDialogOkButton();
|
|
device.waitForIdle();
|
|
|
|
try {
|
|
mCountDownLatch.await(WAIT_TIME_SECONDS, TimeUnit.SECONDS);
|
|
} catch (Exception e) {
|
|
fail("Cannot wait because of error." + e.toString());
|
|
}
|
|
|
|
assertTrue(mErrorReason, mOperationExecuted);
|
|
|
|
bots.roots.openRoot(ROOT_1_ID);
|
|
device.waitForIdle();
|
|
assertFalse(bots.directory.hasDocuments(TARGET_FILE));
|
|
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
device.waitForIdle();
|
|
assertTrue(bots.directory.hasDocuments(TARGET_FILE));
|
|
}
|
|
|
|
@HugeLongTest
|
|
public void testMoveDocument_Cancel() throws Exception {
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
|
|
bots.directory.findDocument(TARGET_FILE);
|
|
device.waitForIdle();
|
|
|
|
bots.directory.selectDocument(TARGET_FILE, 1);
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickToolbarOverflowItem(context.getResources().getString(R.string.menu_move));
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickDialogCancelButton();
|
|
device.waitForIdle();
|
|
|
|
bots.directory.waitForDocument(TARGET_FILE);
|
|
}
|
|
|
|
@HugeLongTest
|
|
// (TODO: b/156756197) : Deflake tests
|
|
// Notice because this class inherits JUnit3 TestCase, the right way to suppress a test
|
|
// is by removing "test" from prefix, instead of adding @Ignore.
|
|
public void ignored_testMoveDocument_CancelFromNotification() throws Exception {
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
bots.directory.findDocument(TARGET_FILE);
|
|
device.waitForIdle();
|
|
|
|
bots.directory.selectDocument(TARGET_FILE, 1);
|
|
device.waitForIdle();
|
|
|
|
bots.main.clickToolbarOverflowItem(context.getResources().getString(R.string.menu_move));
|
|
device.waitForIdle();
|
|
|
|
bots.roots.openRoot(ROOT_1_ID);
|
|
bots.main.clickDialogOkButton();
|
|
device.waitForIdle();
|
|
|
|
try {
|
|
mCountDownLatch.await(WAIT_TIME_SECONDS, TimeUnit.SECONDS);
|
|
} catch (Exception e) {
|
|
fail("Cannot wait because of error." + e.toString());
|
|
}
|
|
|
|
assertTrue(mErrorReason, mOperationExecuted);
|
|
|
|
bots.roots.openRoot(ROOT_1_ID);
|
|
device.waitForIdle();
|
|
assertFalse(bots.directory.hasDocuments(TARGET_FILE));
|
|
|
|
bots.roots.openRoot(ROOT_0_ID);
|
|
device.waitForIdle();
|
|
assertTrue(bots.directory.hasDocuments(TARGET_FILE));
|
|
}
|
|
}
|