1d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd/* 2d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * Copyright (C) 2015 The Android Open Source Project 3d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * 4d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * Licensed under the Apache License, Version 2.0 (the "License"); 5d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * you may not use this file except in compliance with the License. 6d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * You may obtain a copy of the License at 7d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * 8d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * http://www.apache.org/licenses/LICENSE-2.0 9d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * 10d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * Unless required by applicable law or agreed to in writing, software 11d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * distributed under the License is distributed on an "AS IS" BASIS, 12d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * See the License for the specific language governing permissions and 14d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * limitations under the License. 15d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd */ 16d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 17d3b009ae55651f1e60950342468e3c37fdeb0796Mike Doddpackage com.android.messaging.datamodel; 18d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 19d3b009ae55651f1e60950342468e3c37fdeb0796Mike Doddimport android.os.Bundle; 20d3b009ae55651f1e60950342468e3c37fdeb0796Mike Doddimport android.test.mock.MockCursor; 21d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 22d3b009ae55651f1e60950342468e3c37fdeb0796Mike Doddimport java.util.ArrayList; 23d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 24d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd/** 25d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd * A simple in memory fake cursor that can be used for UI tests. 26d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd */ 27d3b009ae55651f1e60950342468e3c37fdeb0796Mike Doddpublic class FakeCursor extends MockCursor { 28d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd private final ArrayList<Integer> mProjection; 29d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd private final String[] mColumnNamesOfData; 30d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd private final Object[][] mData; 31d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd private int mIndex; 32d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 33d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public FakeCursor(final String[] projection, final String[] columnNames, 34d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final Object[][] data) { 35d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mColumnNamesOfData = columnNames; 36d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mData = data; 37d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mIndex = -1; 38d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mProjection = new ArrayList<Integer>(projection.length); 39d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd for (final String column : projection) { 40d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mProjection.add(getColumnIndex(column)); 41d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 42d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 43d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 44d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public Object getAt(final String columnName, final int row) { 45d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final int dataIdx = getColumnIndex(columnName); 46d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return (dataIdx < 0 || row < 0 || row >= mData.length) ? 0 : mData[row][dataIdx]; 47d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 48d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 49d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 50d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getCount() { 51d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return mData.length; 52d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 53d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 54d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 55d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean isFirst() { 56d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return mIndex == 0; 57d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 58d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 59d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 60d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean isLast() { 61d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return mIndex == mData.length - 1; 62d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 63d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 64d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 65d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean moveToFirst() { 66d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (mData.length == 0) { 67d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return false; 68d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 69d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mIndex = 0; 70d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return true; 71d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 72d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 73d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 74d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean moveToPosition(final int position) { 75d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (position < 0 || position >= mData.length) { 76d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return false; 77d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 78d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mIndex = position; 79d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return true; 80d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 81d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 82d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 83d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getPosition() { 84d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return mIndex; 85d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 86d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 87d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 88d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean moveToPrevious() { 89d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (mIndex <= 0) { 90d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return false; 91d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 92d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mIndex--; 93d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return true; 94d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 95d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 96d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 97d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean moveToNext() { 98d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (mIndex == mData.length - 1) { 99d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return false; 100d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 101d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 102d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd mIndex++; 103d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return true; 104d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 105d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 106d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 107d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getColumnCount() { 108d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return mColumnNamesOfData.length; 109d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 110d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 111d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 112d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getColumnIndex(final String columnName) { 113d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd for (int i = 0 ; i < mColumnNamesOfData.length ; i++) { 114d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (mColumnNamesOfData[i].equals(columnName)) { 115d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return i; 116d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 117d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 118d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return -1; 119d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 120d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 121d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 122d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getColumnIndexOrThrow(final String columnName) { 123d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final int result = getColumnIndex(columnName); 124d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd if (result == -1) { 125d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd throw new IllegalArgumentException(); 126d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 127d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 128d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return result; 129d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 130d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 131d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 132d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public String getString(final int columnIndex) { 133d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final int dataIdx = mProjection.get(columnIndex); 134d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final Object obj = (dataIdx < 0 ? null : mData[mIndex][dataIdx]); 135d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return (obj == null ? null : obj.toString()); 136d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 137d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 138d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 139d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public int getInt(final int columnIndex) { 140d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final int dataIdx = mProjection.get(columnIndex); 141d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return (dataIdx < 0 ? 0 : (Integer) mData[mIndex][dataIdx]); 142d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 143d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 144d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 145d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public long getLong(final int columnIndex) { 146d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd final int dataIdx = mProjection.get(columnIndex); 147d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return (dataIdx < 0 ? 0 : (Long) mData[mIndex][dataIdx]); 148d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 149d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 150d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 151d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public void close() { 152d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 153d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 154d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 155d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public boolean isClosed() { 156d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd return false; 157d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd } 158d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd 159d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd @Override 160d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd public Bundle getExtras() { return null; } 161d3b009ae55651f1e60950342468e3c37fdeb0796Mike Dodd} 162