Add support for search in DownloadManager. (9f63f94c) · Commits · e / os / android_frameworks_base

core/java/android/app/DownloadManager.java

+38 −5

Original line number	Diff line number	Diff line
		@@ -839,6 +839,7 @@ public class DownloadManager {

		private long[] mIds = null;
		private Integer mStatusFlags = null;
		private String mFilterString = null;
		private String mOrderByColumn = Downloads.Impl.COLUMN_LAST_MODIFICATION;
		private int mOrderDirection = ORDER_DESCENDING;
		private boolean mOnlyIncludeVisibleInDownloadsUi = false;
		@@ -852,6 +853,17 @@ public class DownloadManager {
		return this;
		}

		/**
		*
		* Include only the downloads that contains the given string in its name.
		* @return this object
		* @hide
		*/
		public Query setFilterByString(@Nullable String filter) {
		mFilterString = filter;
		return this;
		}

		/**
		* Include only downloads with status matching any the given status flags.
		* @param flags any combination of the STATUS_* bit flags
		@@ -911,9 +923,20 @@ public class DownloadManager {
		List<String> selectionParts = new ArrayList<String>();
		String[] selectionArgs = null;

		int whereArgsCount = (mIds == null) ? 0 : mIds.length;
		whereArgsCount = (mFilterString == null) ? whereArgsCount : whereArgsCount + 1;
		selectionArgs = new String[whereArgsCount];

		if (whereArgsCount > 0) {
		if (mIds != null) {
		selectionParts.add(getWhereClauseForIds(mIds));
		selectionArgs = getWhereArgsForIds(mIds);
		getWhereArgsForIds(mIds, selectionArgs);
		}

		if (mFilterString != null) {
		selectionParts.add(Downloads.Impl.COLUMN_TITLE + " LIKE ?");
		selectionArgs[selectionArgs.length - 1] = "%" + mFilterString + "%";
		}
		}

		if (mStatusFlags != null) {
		@@ -1486,12 +1509,22 @@ public class DownloadManager {
		*/
		static String[] getWhereArgsForIds(long[] ids) {
		String[] whereArgs = new String[ids.length];
		return getWhereArgsForIds(ids, whereArgs);
		}

		/**
		* Get selection args for a clause returned by {@link #getWhereClauseForIds(long[])}
		* and write it to the supplied args array.
		*/
		static String[] getWhereArgsForIds(long[] ids, String[] args) {
		assert(args.length >= ids.length);
		for (int i = 0; i < ids.length; i++) {
		whereArgs[i] = Long.toString(ids[i]);
		args[i] = Long.toString(ids[i]);
		}
		return whereArgs;
		return args;
		}


		/**
		* This class wraps a cursor returned by DownloadProvider -- the "underlying cursor" -- and
		* presents a different set of columns, those defined in the DownloadManager.COLUMN_* constants.

packages/DocumentsUI/tests/src/com/android/documentsui/SearchViewUiTest.java

+40 −0

Original line number	Diff line number	Diff line
		@@ -72,6 +72,32 @@ public class SearchViewUiTest extends ActivityTest<FilesActivity> {
		bots.main.assertSearchTextField(false, query);
		}

		public void testSearchDownloads() throws Exception {
		initTestFiles();
		bots.roots.openRoot(ROOT_0_ID);

		bots.directory.copyFilesToClipboard(fileName1, fileName2);
		device.waitForIdle();

		bots.roots.openRoot("Downloads");
		bots.directory.pasteFilesFromClipboard();

		//TODO: linben Why do we need to click on Downloads again so this will work?
		bots.roots.openRoot("Downloads");
		device.waitForIdle();

		String query = "file12";
		bots.main.openSearchView();
		bots.main.setSearchQuery(query);

		device.pressEnter();

		bots.directory.assertDocumentsCountOnList(true, 1);
		bots.directory.assertDocumentsPresent(fileName2);

		device.pressBack();
		}

		@Suppress
		public void testSearchResultsFound_ClearsOnBack() throws Exception {
		initTestFiles();
		@@ -151,4 +177,18 @@ public class SearchViewUiTest extends ActivityTest<FilesActivity> {
		bots.main.assertSearchTextFiledAndIcon(false, false);
		}

		@Override
		public void tearDown() throws Exception {
		try {
		// Proper clean up of #testSearchDownloads
		bots.directory.clickDocument(fileName1 + ".txt");
		bots.directory.clickDocument(fileName2);
		device.waitForIdle();
		bots.main.menuDelete().click();
		bots.main.findDialogOkButton().click();
		} catch (Exception e) {
		} finally {
		super.tearDown();
		}
		}
		}

packages/DocumentsUI/tests/src/com/android/documentsui/bots/DirectoryListBot.java

+12 −0

Original line number	Diff line number	Diff line
		@@ -30,6 +30,7 @@ import android.support.test.uiautomator.UiObject2;
		import android.support.test.uiautomator.UiObjectNotFoundException;
		import android.support.test.uiautomator.UiSelector;
		import android.support.test.uiautomator.Until;
		import android.view.KeyEvent;
		import android.view.MotionEvent;

		import junit.framework.Assert;
		@@ -129,6 +130,17 @@ public class DirectoryListBot extends BaseBot {
		return doc;
		}

		public void copyFilesToClipboard(String...labels) throws UiObjectNotFoundException {
		for (String label: labels) {
		clickDocument(label);
		}
		mDevice.pressKeyCode(KeyEvent.KEYCODE_C, KeyEvent.META_CTRL_ON);
		}

		public void pasteFilesFromClipboard() {
		mDevice.pressKeyCode(KeyEvent.KEYCODE_V, KeyEvent.META_CTRL_ON);
		}

		public UiObject2 getSnackbar(String message) {
		return mDevice.wait(Until.findObject(By.text(message)), mTimeout);
		}