TestContentProvider.java revision a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7 
1// Copyright 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5/**
6 * Content provider for testing content URLs.
7 */
8
9package org.chromium.android_webview.test;
10
11import android.content.ContentProvider;
12import android.content.ContentValues;
13import android.content.Context;
14import android.content.res.AssetFileDescriptor;
15import android.database.AbstractCursor;
16import android.database.Cursor;
17import android.net.Uri;
18import android.os.ParcelFileDescriptor;
19import android.util.Log;
20
21import java.io.FileOutputStream;
22import java.io.IOException;
23import java.util.HashMap;
24import java.util.Map;
25
26// Note: if you move this class, make sure you have also updated AndroidManifest.xml
27public class TestContentProvider extends ContentProvider {
28    private static final String AUTHORITY =
29            "org.chromium.android_webview.test.TestContentProvider";
30    private static final String CONTENT_SCHEME = "content://";
31    private static final String CONTENT_TYPE = "image/png";
32    private static final String GET_RESOURCE_REQUEST_COUNT = "get_resource_request_count";
33    private static final String RESET_RESOURCE_REQUEST_COUNT = "reset_resource_request_count";
34    private static final String TAG = "TestContentProvider";
35    private enum ColumnIndex {
36        RESOURCE_REQUEST_COUNT_COLUMN,
37    };
38    private final Map<String, Integer> mResourceRequestCount;
39
40    public static String createContentUrl(String target) {
41        return CONTENT_SCHEME + AUTHORITY + "/" + target;
42    }
43
44    private static Uri createRequestUri(final String target, String resource) {
45        return Uri.parse(createContentUrl(target) + "?" + resource);
46    }
47
48    public static int getResourceRequestCount(Context context, String resource) {
49        Uri uri = createRequestUri(GET_RESOURCE_REQUEST_COUNT, resource);
50        final Cursor cursor = context.getContentResolver().query(uri, null, null, null, null);
51        try {
52            cursor.moveToFirst();
53            return cursor.getInt(ColumnIndex.RESOURCE_REQUEST_COUNT_COLUMN.ordinal());
54        } finally {
55            cursor.close();
56        }
57    }
58
59    public static void resetResourceRequestCount(Context context, String resource) {
60        Uri uri = createRequestUri(RESET_RESOURCE_REQUEST_COUNT, resource);
61        // A null cursor is returned for this request.
62        context.getContentResolver().query(uri, null, null, null, null);
63    }
64
65    public TestContentProvider() {
66        super();
67        mResourceRequestCount = new HashMap<String, Integer>();
68    }
69
70    @Override
71    public boolean onCreate() {
72        return true;
73    }
74
75    @Override
76    public AssetFileDescriptor openAssetFile(Uri uri, String mode) {
77        String resource = uri.getLastPathSegment();
78        if (mResourceRequestCount.containsKey(resource)) {
79            mResourceRequestCount.put(resource, mResourceRequestCount.get(resource) + 1);
80        } else {
81            mResourceRequestCount.put(resource, 1);
82        }
83        return createImage();
84    }
85
86    @Override
87    public String getType(Uri uri) {
88        return CONTENT_TYPE;
89    }
90
91    @Override
92    public int update(Uri uri, ContentValues values, String where,
93                      String[] whereArgs) {
94        return 0;
95    }
96
97    @Override
98    public int delete(Uri uri, String selection, String[] selectionArgs) {
99        return 0;
100    }
101
102    @Override
103    public Uri insert(Uri uri, ContentValues values) {
104        return null;
105    }
106
107    /**
108     * Cursor object for retrieving resource request counters.
109     */
110    private static class ProviderStateCursor extends AbstractCursor {
111        private final int mResourceRequestCount;
112
113        public ProviderStateCursor(int resourceRequestCount) {
114            mResourceRequestCount = resourceRequestCount;
115        }
116
117        @Override
118        public boolean isNull(int columnIndex) {
119            return columnIndex != ColumnIndex.RESOURCE_REQUEST_COUNT_COLUMN.ordinal();
120        }
121
122        @Override
123        public int getCount() {
124            return 1;
125        }
126
127        @Override
128        public int getType(int columnIndex) {
129            if (columnIndex == ColumnIndex.RESOURCE_REQUEST_COUNT_COLUMN.ordinal()) {
130                return Cursor.FIELD_TYPE_INTEGER;
131            }
132            return Cursor.FIELD_TYPE_NULL;
133        }
134
135        private void unsupported() {
136            throw new UnsupportedOperationException();
137        }
138
139        @Override
140        public double getDouble(int columnIndex) {
141            unsupported();
142            return 0.0;
143        }
144
145        @Override
146        public float getFloat(int columnIndex) {
147            unsupported();
148            return 0.0f;
149        }
150
151        @Override
152        public int getInt(int columnIndex) {
153            if (columnIndex == ColumnIndex.RESOURCE_REQUEST_COUNT_COLUMN.ordinal()) {
154                return mResourceRequestCount;
155            }
156            return -1;
157        }
158
159        @Override
160        public short getShort(int columnIndex) {
161            unsupported();
162            return 0;
163        }
164
165        @Override
166        public long getLong(int columnIndex) {
167            return getInt(columnIndex);
168        }
169
170        @Override
171        public String getString(int columnIndex) {
172            unsupported();
173            return null;
174        }
175
176        @Override
177        public String[] getColumnNames() {
178            return new String[] { GET_RESOURCE_REQUEST_COUNT };
179        }
180    }
181
182    @Override
183    public Cursor query(Uri uri, String[] projection, String selection,
184                        String[] selectionArgs, String sortOrder) {
185        String action = uri.getLastPathSegment();
186        String resource = uri.getQuery();
187        if (GET_RESOURCE_REQUEST_COUNT.equals(action)) {
188            return new ProviderStateCursor(
189                mResourceRequestCount.containsKey(resource) ?
190                mResourceRequestCount.get(resource) : 0);
191        } else if (RESET_RESOURCE_REQUEST_COUNT.equals(action)) {
192            mResourceRequestCount.put(resource, 0);
193        }
194        return null;
195    }
196
197    // 1x1 black dot png image.
198    private static final byte[] IMAGE = {
199        (byte)0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a, 0x00, 0x00, 0x00,
200        0x0d, 0x49, 0x48, 0x44, 0x52, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
201        0x01, 0x08, 0x00, 0x00, 0x00, 0x00, 0x3a, 0x7e, (byte)0x9b, 0x55, 0x00,
202        0x00, 0x00, 0x01, 0x73, 0x52, 0x47, 0x42, 0x00, (byte)0xae, (byte)0xce,
203        0x1c, (byte)0xe9, 0x00, 0x00, 0x00, 0x0d, 0x49, 0x44, 0x41, 0x54, 0x08,
204        0x1d, 0x01, 0x02, 0x00, (byte)0xfd, (byte)0xff, 0x00, 0x00, 0x00, 0x02,
205        0x00, 0x01, (byte)0xcd, (byte)0xe3, (byte)0xd1, 0x2b, 0x00, 0x00, 0x00,
206        0x00, 0x49, 0x45, 0x4e, 0x44, (byte)0xae, 0x42, 0x60, (byte)0x82
207    };
208
209    private static AssetFileDescriptor createImage() {
210        ParcelFileDescriptor[] pfds = null;
211        FileOutputStream fileOut = null;
212        try {
213            try {
214                pfds = ParcelFileDescriptor.createPipe();
215                fileOut = new FileOutputStream(pfds[1].getFileDescriptor());
216                fileOut.write(IMAGE);
217                fileOut.flush();
218                return new AssetFileDescriptor(pfds[0], 0, -1);
219            } finally {
220                if (fileOut != null) fileOut.close();
221                if (pfds != null && pfds[1] != null) pfds[1].close();
222            }
223        } catch (IOException e) {
224            Log.e(TAG, e.getMessage(), e);
225        }
226        return null;
227    }
228}
229