DocumentsProviderHelper.java revision a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5
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; 2299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport static com.android.documentsui.model.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 43bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport com.android.documentsui.model.DocumentInfo; 4499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport com.android.documentsui.model.RootInfo; 4599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 46bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport com.google.android.collect.Lists; 47bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 4899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKayimport libcore.io.IoUtils; 49bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport libcore.io.Streams; 50bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 51bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.io.IOException; 52bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.util.ArrayList; 53bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKayimport java.util.List; 5499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 5599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay/** 5699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay * Provides support for creation of documents in a test settings. 5799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay */ 5899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKaypublic class DocumentsProviderHelper { 5999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 6099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay private final String mAuthority; 61bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay private final ContentProviderClient mClient; 6299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 6399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public DocumentsProviderHelper(String authority, ContentProviderClient client) { 6422797f589b5251585b069622de0417a0c240ac35Steve McKay checkArgument(!TextUtils.isEmpty(authority)); 6599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay mAuthority = authority; 66bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay mClient = client; 6799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 6899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 69bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public RootInfo getRoot(String documentId) throws RemoteException { 70bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay final Uri rootsUri = buildRootsUri(mAuthority); 7199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay Cursor cursor = null; 7299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay try { 7399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay cursor = mClient.query(rootsUri, null, null, null, null); 7499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay while (cursor.moveToNext()) { 75bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (documentId.equals(getCursorString(cursor, Root.COLUMN_ROOT_ID))) { 7699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return RootInfo.fromRootsCursor(mAuthority, cursor); 7799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 7899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 79bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throw new IllegalArgumentException("Can't find matching root for id=" + documentId); 8099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } catch (Exception e) { 81bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throw new RuntimeException("Can't load root for id=" + documentId , e); 8299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } finally { 8399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay IoUtils.closeQuietly(cursor); 8499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 8599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 8699bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 8799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createDocument(Uri parentUri, String mimeType, String name) { 8899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay if (name.contains("/")) { 8999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay throw new IllegalArgumentException("Name and mimetype probably interposed."); 9099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 9199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay try { 92bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri uri = DocumentsContract.createDocument(mClient, parentUri, mimeType, name); 93bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return uri; 9499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } catch (RemoteException e) { 95a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski throw new RuntimeException("Couldn't create document: " + name + " with mimetype " 96a2ad8dfc42a4fcd5391423605b5d48fa28ee7fc5Tomasz Mikolajewski + mimeType, e); 9799bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 9899bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 9999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 100bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createDocument(String parentId, String mimeType, String name) { 101bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 102bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(parentUri, mimeType, name); 103bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 104bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 105bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createDocument(RootInfo root, String mimeType, String name) { 106bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(root.documentId, mimeType, name); 107bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 108bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 10918410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska public Uri createDocumentWithFlags(String documentId, String mimeType, String name, int flags) 11018410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska throws RemoteException { 11118410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Bundle in = new Bundle(); 11218410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putInt(StubProvider.EXTRA_FLAGS, flags); 11318410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(StubProvider.EXTRA_PARENT_ID, documentId); 11418410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(Document.COLUMN_MIME_TYPE, mimeType); 11518410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska in.putString(Document.COLUMN_DISPLAY_NAME, name); 11618410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 11718410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Bundle out = mClient.call("createDocumentWithFlags", null, in); 11818410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska Uri uri = out.getParcelable(DocumentsContract.EXTRA_URI); 11918410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska return uri; 12018410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska } 12118410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 12299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createFolder(Uri parentUri, String name) { 12399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return createDocument(parentUri, Document.MIME_TYPE_DIR, name); 12499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 12599bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 126bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createFolder(String parentId, String name) { 127bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 128bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return createDocument(parentUri, Document.MIME_TYPE_DIR, name); 12999bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 13099bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay 13199bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay public Uri createFolder(RootInfo root, String name) { 13299bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay return createDocument(root, Document.MIME_TYPE_DIR, name); 13399bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay } 134bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 135bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void writeDocument(Uri documentUri, byte[] contents) 136bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws RemoteException, IOException { 137bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay ParcelFileDescriptor file = mClient.openFile(documentUri, "w", null); 138bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (AutoCloseOutputStream out = new AutoCloseOutputStream(file)) { 139bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay out.write(contents, 0, contents.length); 140bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 141bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 142bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 143bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public byte[] readDocument(Uri documentUri) throws RemoteException, IOException { 144bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay ParcelFileDescriptor file = mClient.openFile(documentUri, "r", null); 145bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay byte[] buf = null; 146bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (AutoCloseInputStream in = new AutoCloseInputStream(file)) { 147bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay buf = Streams.readFully(in); 148bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 149bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return buf; 150bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 151bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 152bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(Uri parentUri, int expected) throws Exception { 153bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 154bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertEquals("Incorrect file count after copy", expected, children.size()); 155bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 156bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 157bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(String parentId, int expected) throws Exception { 158bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 159bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertEquals("Incorrect file count after copy", expected, children.size()); 160bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 161bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 162bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertChildCount(RootInfo root, int expected) throws Exception { 163bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertChildCount(root.documentId, expected); 164bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 165bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 166bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(Uri parentUri, String name) throws Exception { 167bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 168bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 169bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName) && !child.isDirectory()) { 170bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return; 171bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 172bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 173bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Could not find file named=" + name + " in children " + children); 174bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 175bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 176bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(String parentId, String name) throws Exception { 177bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 178bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasFile(parentUri, name); 179bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 180bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 181bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasFile(RootInfo root, String name) throws Exception { 182bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasFile(root.documentId, name); 183bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 184bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 185bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(Uri parentUri, String name) throws Exception { 186bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 187bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 188bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName) && child.isDirectory()) { 189bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return; 190bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 191bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 192bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Could not find name=" + name + " in children " + children); 193bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 194bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 195bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(String parentId, String name) throws Exception { 196bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 197bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasDirectory(parentUri, name); 198bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 199bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 200bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertHasDirectory(RootInfo root, String name) throws Exception { 201bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertHasDirectory(root.documentId, name); 202bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 203bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 204bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(Uri parentUri, String name) throws Exception { 205bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 206bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 207bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 208bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay fail("Found name=" + name + " in children " + children); 209bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 210bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 211bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 212bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 213bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(String parentId, String name) throws Exception { 214bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri parentUri = buildDocumentUri(mAuthority, parentId); 215bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertDoesNotExist(parentUri, name); 216bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 217bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 218bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertDoesNotExist(RootInfo root, String name) throws Exception { 219bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertDoesNotExist(root.getUri(), name); 220bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 221bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 222bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public @Nullable DocumentInfo findFile(String parentId, String name) 223bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws Exception { 224bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 225bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 226bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 227bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 228bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 229bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 230bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 231bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 232bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 233bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public DocumentInfo findDocument(String parentId, String name) throws Exception { 234bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentId); 235bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 236bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 237bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 238bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 239bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 240bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 241bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 242bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 243bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public DocumentInfo findDocument(Uri parentUri, String name) throws Exception { 244bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = listChildren(parentUri); 245bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay for (DocumentInfo child : children) { 246bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay if (name.equals(child.displayName)) { 247bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return child; 248bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 249bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 250bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return null; 251bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 252bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 253bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public List<DocumentInfo> listChildren(Uri parentUri) throws Exception { 254bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay String id = DocumentsContract.getDocumentId(parentUri); 255bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return listChildren(id); 256bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 257bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 258bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public List<DocumentInfo> listChildren(String documentId) throws Exception { 259bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Uri uri = buildChildDocumentsUri(mAuthority, documentId); 260bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay List<DocumentInfo> children = new ArrayList<>(); 261bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay try (Cursor cursor = mClient.query(uri, null, null, null, null, null)) { 26222797f589b5251585b069622de0417a0c240ac35Steve McKay Cursor wrapper = new RootCursorWrapper(mAuthority, "totally-fake", cursor, 100); 26322797f589b5251585b069622de0417a0c240ac35Steve McKay while (wrapper.moveToNext()) { 26422797f589b5251585b069622de0417a0c240ac35Steve McKay children.add(DocumentInfo.fromDirectoryCursor(wrapper)); 265bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 266bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 267bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return children; 268bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 269bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 270bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertFileContents(Uri documentUri, byte[] expected) throws Exception { 27122797f589b5251585b069622de0417a0c240ac35Steve McKay MoreAsserts.assertEquals( 27222797f589b5251585b069622de0417a0c240ac35Steve McKay "Copied file contents differ", 27322797f589b5251585b069622de0417a0c240ac35Steve McKay expected, readDocument(documentUri)); 274bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 275bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 276bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public void assertFileContents(String parentId, String fileName, byte[] expected) 277bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws Exception { 278bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay DocumentInfo file = findFile(parentId, fileName); 279bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertNotNull(file); 280bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay assertFileContents(file.derivedUri, expected); 281bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 282bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 283bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay /** 284bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay * A helper method for StubProvider only. Won't work with other providers. 285bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay * @throws RemoteException 286bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay */ 287bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay public Uri createVirtualFile( 288bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay RootInfo root, String path, String mimeType, byte[] content, String... streamTypes) 289bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay throws RemoteException { 290bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 291bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Bundle args = new Bundle(); 292bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(StubProvider.EXTRA_ROOT, root.rootId); 293bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(StubProvider.EXTRA_PATH, path); 294bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putString(Document.COLUMN_MIME_TYPE, mimeType); 295bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putStringArrayList(StubProvider.EXTRA_STREAM_TYPES, Lists.newArrayList(streamTypes)); 296bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay args.putByteArray(StubProvider.EXTRA_CONTENT, content); 297bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 298bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay Bundle result = mClient.call("createVirtualFile", null, args); 299bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay String documentId = result.getString(Document.COLUMN_DOCUMENT_ID); 300bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay 301bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay return DocumentsContract.buildDocumentUri(mAuthority, documentId); 302bbeba5265c5baa42c6db93fc8030c6055747da4dSteve McKay } 30318410b7e10fb6e5841c1a309e706aa2d68552a6bAga Wronska 30499bcc6aa4dbcfaf6206d9bee496f7388fa9ab17cSteve McKay} 305