DocumentsProviderHelper.java revision d9caa6ab53aa784acaf241c0ded3c4ae2d342bf8
199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay/* 299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * Copyright (C) 2015 The Android Open Source Project 399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * 499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * Licensed under the Apache License, Version 2.0 (the "License"); 599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * you may not use this file except in compliance with the License. 699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * You may obtain a copy of the License at 799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * 899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * http://www.apache.org/licenses/LICENSE-2.0 999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * 1099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * Unless required by applicable law or agreed to in writing, software 1199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * distributed under the License is distributed on an "AS IS" BASIS, 1299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * See the License for the specific language governing permissions and 1499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * limitations under the License. 1599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay */ 1699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 1799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKaypackage com.android.documentsui; 1899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 19bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static android.provider.DocumentsContract.buildChildDocumentsUri; 20bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static android.provider.DocumentsContract.buildDocumentUri; 21bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static android.provider.DocumentsContract.buildRootsUri; 22d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKayimport static com.android.documentsui.base.DocumentInfo.getCursorString; 2322797f589b5251585b069622de0417a0c240ac35Steve McKayimport static com.android.internal.util.Preconditions.checkArgument; 24bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static junit.framework.Assert.assertEquals; 25bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static junit.framework.Assert.assertNotNull; 26bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport static junit.framework.Assert.fail; 2799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 2899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.content.ContentProviderClient; 2999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.database.Cursor; 3099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.net.Uri; 31bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.os.Bundle; 32bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.os.ParcelFileDescriptor; 33bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.os.ParcelFileDescriptor.AutoCloseInputStream; 34bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.os.ParcelFileDescriptor.AutoCloseOutputStream; 3599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.os.RemoteException; 3699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.provider.DocumentsContract; 3799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.provider.DocumentsContract.Document; 3899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport android.provider.DocumentsContract.Root; 39bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.support.annotation.Nullable; 40bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport android.test.MoreAsserts; 4122797f589b5251585b069622de0417a0c240ac35Steve McKayimport android.text.TextUtils; 4299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 43d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKayimport com.android.documentsui.base.DocumentInfo; 44d080506e3aa8547605cd4783eb660775d7d2b8eeSteve McKayimport com.android.documentsui.base.RootInfo; 45d9caa6ab53aa784acaf241c0ded3c4ae2d342bf8Steve McKayimport com.android.documentsui.roots.RootCursorWrapper; 4699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 47bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport com.google.android.collect.Lists; 48bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 4999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport libcore.io.IoUtils; 50bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport libcore.io.Streams; 51bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 52bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.io.IOException; 53bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.util.ArrayList; 54bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.util.List; 5599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 5699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay/** 5799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * Provides support for creation of documents in a test settings. 5899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay */ 5999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKaypublic class DocumentsProviderHelper { 6099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 6199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay private final String mAuthority; 62bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay private final ContentProviderClient mClient; 6399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 6499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public DocumentsProviderHelper(String authority, ContentProviderClient client) { 6522797f589b5251585b069622de0417a0c240ac35Steve McKay checkArgument(!TextUtils.isEmpty(authority)); 6699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay mAuthority = authority; 67bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay mClient = client; 6899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 6999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 70bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public RootInfo getRoot(String documentId) throws RemoteException { 71bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay final Uri rootsUri = buildRootsUri(mAuthority); 7299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay Cursor cursor = null; 7399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay try { 7499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay cursor = mClient.query(rootsUri, null, null, null, null); 7599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay while (cursor.moveToNext()) { 76bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (documentId.equals(getCursorString(cursor, Root.COLUMN_ROOT_ID))) { 7799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return RootInfo.fromRootsCursor(mAuthority, cursor); 7899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 7999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 80bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throw new IllegalArgumentException("Can't find matching root for id=" + documentId); 8199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } catch (Exception e) { 82bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throw new RuntimeException("Can't load root for id=" + documentId , e); 8399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } finally { 8499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay IoUtils.closeQuietly(cursor); 8599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 8699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 8799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 8899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createDocument(Uri parentUri, String mimeType, String name) { 8999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay if (name.contains("/")) { 9099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay throw new IllegalArgumentException("Name and mimetype probably interposed."); 9199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 9299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay try { 93bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri uri = DocumentsContract.createDocument(mClient, parentUri, mimeType, name); 94bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return uri; 9599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } catch (RemoteException e) { 96a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski throw new RuntimeException("Couldn't create document: " + name + " with mimetype " 97a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski + mimeType, e); 9899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 9999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 10099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 101bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createDocument(String parentId, String mimeType, String name) { 102bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 103bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(parentUri, mimeType, name); 104bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 105bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 106bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createDocument(RootInfo root, String mimeType, String name) { 107bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(root.documentId, mimeType, name); 108bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 109bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 110fe7f53647db0b1b50dff9489a77656046c20bf7bTomasz Mikolajewski public Uri createDocumentWithFlags(String documentId, String mimeType, String name, int flags, 111fe7f53647db0b1b50dff9489a77656046c20bf7bTomasz Mikolajewski String... streamTypes) 11218410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska throws RemoteException { 11318410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Bundle in = new Bundle(); 11418410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putInt(StubProvider.EXTRA_FLAGS, flags); 11518410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(StubProvider.EXTRA_PARENT_ID, documentId); 11618410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(Document.COLUMN_MIME_TYPE, mimeType); 11718410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(Document.COLUMN_DISPLAY_NAME, name); 118fe7f53647db0b1b50dff9489a77656046c20bf7bTomasz Mikolajewski in.putStringArrayList(StubProvider.EXTRA_STREAM_TYPES, Lists.newArrayList(streamTypes)); 11918410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 12018410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Bundle out = mClient.call("createDocumentWithFlags", null, in); 12118410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Uri uri = out.getParcelable(DocumentsContract.EXTRA_URI); 12218410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska return uri; 12318410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska } 12418410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 12599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createFolder(Uri parentUri, String name) { 12699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return createDocument(parentUri, Document.MIME_TYPE_DIR, name); 12799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 12899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 129bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createFolder(String parentId, String name) { 130bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 131bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(parentUri, Document.MIME_TYPE_DIR, name); 13299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 13399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 13499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createFolder(RootInfo root, String name) { 13599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return createDocument(root, Document.MIME_TYPE_DIR, name); 13699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 137bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 138bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void writeDocument(Uri documentUri, byte[] contents) 139bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws RemoteException, IOException { 140bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay ParcelFileDescriptor file = mClient.openFile(documentUri, "w", null); 141bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (AutoCloseOutputStream out = new AutoCloseOutputStream(file)) { 142bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay out.write(contents, 0, contents.length); 143bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 144bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 145bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 146bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public byte[] readDocument(Uri documentUri) throws RemoteException, IOException { 147bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay ParcelFileDescriptor file = mClient.openFile(documentUri, "r", null); 148bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay byte[] buf = null; 149bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (AutoCloseInputStream in = new AutoCloseInputStream(file)) { 150bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay buf = Streams.readFully(in); 151bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 152bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return buf; 153bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 154bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 155bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(Uri parentUri, int expected) throws Exception { 156bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 157bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertEquals("Incorrect file count after copy", expected, children.size()); 158bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 159bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 160bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(String parentId, int expected) throws Exception { 161bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 162bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertEquals("Incorrect file count after copy", expected, children.size()); 163bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 164bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 165bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(RootInfo root, int expected) throws Exception { 166bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertChildCount(root.documentId, expected); 167bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 168bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 169bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(Uri parentUri, String name) throws Exception { 170bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 171bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 172bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName) && !child.isDirectory()) { 173bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return; 174bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 175bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 176bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Could not find file named=" + name + " in children " + children); 177bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 178bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 179bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(String parentId, String name) throws Exception { 180bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 181bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasFile(parentUri, name); 182bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 183bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 184bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(RootInfo root, String name) throws Exception { 185bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasFile(root.documentId, name); 186bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 187bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 188bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(Uri parentUri, String name) throws Exception { 189bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 190bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 191bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName) && child.isDirectory()) { 192bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return; 193bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 194bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 195bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Could not find name=" + name + " in children " + children); 196bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 197bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 198bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(String parentId, String name) throws Exception { 199bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 200bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasDirectory(parentUri, name); 201bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 202bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 203bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(RootInfo root, String name) throws Exception { 204bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasDirectory(root.documentId, name); 205bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 206bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 207bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(Uri parentUri, String name) throws Exception { 208bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 209bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 210bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 211bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Found name=" + name + " in children " + children); 212bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 213bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 214bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 215bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 216bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(String parentId, String name) throws Exception { 217bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 218bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertDoesNotExist(parentUri, name); 219bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 220bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 221bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(RootInfo root, String name) throws Exception { 222bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertDoesNotExist(root.getUri(), name); 223bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 224bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 225bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public @Nullable DocumentInfo findFile(String parentId, String name) 226bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws Exception { 227bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 228bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 229bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 230bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 231bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 232bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 233bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 234bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 235bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 236bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public DocumentInfo findDocument(String parentId, String name) throws Exception { 237bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 238bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 239bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 240bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 241bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 242bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 243bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 244bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 245bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 246bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public DocumentInfo findDocument(Uri parentUri, String name) throws Exception { 247bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 248bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 249bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 250bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 251bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 252bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 253bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 254bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 255bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 256bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public List<DocumentInfo> listChildren(Uri parentUri) throws Exception { 257bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay String id = DocumentsContract.getDocumentId(parentUri); 258bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return listChildren(id); 259bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 260bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 261bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public List<DocumentInfo> listChildren(String documentId) throws Exception { 262bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri uri = buildChildDocumentsUri(mAuthority, documentId); 263bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = new ArrayList<>(); 264bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (Cursor cursor = mClient.query(uri, null, null, null, null, null)) { 26522797f589b5251585b069622de0417a0c240ac35Steve McKay Cursor wrapper = new RootCursorWrapper(mAuthority, "totally-fake", cursor, 100); 26622797f589b5251585b069622de0417a0c240ac35Steve McKay while (wrapper.moveToNext()) { 26722797f589b5251585b069622de0417a0c240ac35Steve McKay children.add(DocumentInfo.fromDirectoryCursor(wrapper)); 268bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 269bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 270bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return children; 271bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 272bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 273bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertFileContents(Uri documentUri, byte[] expected) throws Exception { 27422797f589b5251585b069622de0417a0c240ac35Steve McKay MoreAsserts.assertEquals( 27522797f589b5251585b069622de0417a0c240ac35Steve McKay "Copied file contents differ", 27622797f589b5251585b069622de0417a0c240ac35Steve McKay expected, readDocument(documentUri)); 277bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 278bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 279bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertFileContents(String parentId, String fileName, byte[] expected) 280bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws Exception { 281bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay DocumentInfo file = findFile(parentId, fileName); 282bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertNotNull(file); 283bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertFileContents(file.derivedUri, expected); 284bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 285bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 286bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay /** 287bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay * A helper method for StubProvider only. Won't work with other providers. 288bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay * @throws RemoteException 289bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay */ 290bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createVirtualFile( 291bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay RootInfo root, String path, String mimeType, byte[] content, String... streamTypes) 292bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws RemoteException { 293bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 294bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Bundle args = new Bundle(); 295bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(StubProvider.EXTRA_ROOT, root.rootId); 296bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(StubProvider.EXTRA_PATH, path); 297bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(Document.COLUMN_MIME_TYPE, mimeType); 298bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putStringArrayList(StubProvider.EXTRA_STREAM_TYPES, Lists.newArrayList(streamTypes)); 299bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putByteArray(StubProvider.EXTRA_CONTENT, content); 300bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 301bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Bundle result = mClient.call("createVirtualFile", null, args); 302bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay String documentId = result.getString(Document.COLUMN_DOCUMENT_ID); 303bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 304bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return DocumentsContract.buildDocumentUri(mAuthority, documentId); 305bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 30618410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 30799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay} 308