summaryrefslogtreecommitdiffstats
path: root/tests/src/com/android/messaging/datamodel/FakeCursor.java
blob: 59d1b89756ea37de16015e70c7b54043f478986c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.messaging.datamodel;

import android.os.Bundle;
import android.test.mock.MockCursor;

import java.util.ArrayList;

/**
 * A simple in memory fake cursor that can be used for UI tests.
 */
public class FakeCursor extends MockCursor {
    private final ArrayList<Integer> mProjection;
    private final String[] mColumnNamesOfData;
    private final Object[][] mData;
    private int mIndex;

    public FakeCursor(final String[] projection, final String[] columnNames,
            final Object[][] data) {
        mColumnNamesOfData = columnNames;
        mData = data;
        mIndex = -1;
        mProjection = new ArrayList<Integer>(projection.length);
        for (final String column : projection) {
            mProjection.add(getColumnIndex(column));
        }
    }

    public Object getAt(final String columnName, final int row) {
        final int dataIdx = getColumnIndex(columnName);
        return (dataIdx < 0 || row < 0 || row >= mData.length) ? 0 : mData[row][dataIdx];
    }

    @Override
    public int getCount() {
        return mData.length;
    }

    @Override
    public boolean isFirst() {
        return mIndex == 0;
    }

    @Override
    public boolean isLast() {
        return mIndex == mData.length - 1;
    }

    @Override
    public boolean moveToFirst() {
        if (mData.length == 0) {
            return false;
        }
        mIndex = 0;
        return true;
    }

    @Override
    public boolean moveToPosition(final int position) {
        if (position < 0 || position >= mData.length) {
            return false;
        }
        mIndex = position;
        return true;
    }

    @Override
    public int getPosition() {
        return mIndex;
    }

    @Override
    public boolean moveToPrevious() {
        if (mIndex <= 0) {
            return false;
        }
        mIndex--;
        return true;
    }

    @Override
    public boolean moveToNext() {
        if (mIndex == mData.length - 1) {
            return false;
        }

        mIndex++;
        return true;
    }

    @Override
    public int getColumnCount() {
        return mColumnNamesOfData.length;
    }

    @Override
    public int getColumnIndex(final String columnName) {
        for (int i = 0 ; i < mColumnNamesOfData.length ; i++) {
            if (mColumnNamesOfData[i].equals(columnName)) {
                return i;
            }
        }
        return -1;
    }

    @Override
    public int getColumnIndexOrThrow(final String columnName) {
        final int result = getColumnIndex(columnName);
        if (result == -1) {
            throw new IllegalArgumentException();
        }

        return result;
    }

    @Override
    public String getString(final int columnIndex) {
        final int dataIdx = mProjection.get(columnIndex);
        final Object obj = (dataIdx < 0 ? null : mData[mIndex][dataIdx]);
        return (obj == null ? null : obj.toString());
    }

    @Override
    public int getInt(final int columnIndex) {
        final int dataIdx = mProjection.get(columnIndex);
        return (dataIdx < 0 ? 0 : (Integer) mData[mIndex][dataIdx]);
    }

    @Override
    public long getLong(final int columnIndex) {
        final int dataIdx = mProjection.get(columnIndex);
        return (dataIdx < 0 ? 0 : (Long) mData[mIndex][dataIdx]);
    }

    @Override
    public void close() {
    }

    @Override
    public boolean isClosed() {
        return false;
    }

    @Override
    public Bundle getExtras() { return null; }
}