summaryrefslogtreecommitdiffstats
path: root/tests/src/com/android/dialer/dialpad/SmartDialNameMatcherTest.java
blob: 3c481d0c7f5b5cebb6c2469b373b43e3bda5ee61 (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
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
/*
 * Copyright (C) 2012 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.dialer.dialpad;

import android.test.suitebuilder.annotation.SmallTest;
import android.test.suitebuilder.annotation.Suppress;
import android.util.Log;

import com.android.dialer.dialpad.SmartDialNameMatcher;

import java.text.Normalizer;
import java.util.ArrayList;

import junit.framework.TestCase;

@SmallTest
public class SmartDialNameMatcherTest extends TestCase {
    private static final String TAG = "SmartDialNameMatcherTest";

    public void testMatches() {
        // Test to ensure that all alphabetic characters are covered
        checkMatches("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
                "22233344455566677778889999" + "22233344455566677778889999", true, 0, 26 * 2);
        // Should fail because of a mistyped 2 instead of 9 in the second last character
        checkMatches("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
                "22233344455566677778889999" + "22233344455566677778889929", false, 0, 0);

        // Basic name test
        checkMatches("joe", "5", true, 0, 1);
        checkMatches("joe", "56", true, 0, 2);
        checkMatches("joe", "563", true, 0, 3);

        // Matches only word boundary.
        checkMatches("joe", "63", false, 0, 0);
        checkMatches("joe oe", "63", true, 4, 6);

        // Test for a match across word boundaries
        checkMatches("joe oe", "56363", true, 0, 6);
    }

    public void testMatches_repeatedLetters() {
        checkMatches("aaaaaaaaaa", "2222222222", true, 0, 10);
        // Fails because of one extra 2
        checkMatches("aaaaaaaaaa", "22222222222", false, 0, 0);
        checkMatches("zzzzzzzzzz zzzzzzzzzz", "99999999999999999999", true, 0, 21);
    }

    public void testMatches_repeatedSpaces() {
        checkMatches("William     J  Smith", "9455426576", true, 0, 17);
        checkMatches("William     J  Smith", "576", true, 12, 17);
        // Fails because we start at non-word boundary
        checkMatches("William     J  Smith", "6576", false, 0, 0);
    }


    public void testMatches_Initial() {
        // wjs matches (W)illiam (J)ohn (S)mith
        checkMatches("William John Smith", "957", true, 0, 1, 8, 9, 13, 14);
        // wjsmit matches (W)illiam (J)ohn (Smit)h
        checkMatches("William John Smith", "957648", true, 0, 1, 8, 9, 13, 17);
        // wjohn matches (W)illiam (John) Smith
        checkMatches("William John Smith", "95646", true, 0, 1, 8, 12);
        // jsmi matches William (J)ohn (Smi)th
        checkMatches("William John Smith", "5764", true, 8, 9, 13, 16);
        // make sure multiple spaces don't mess things up
        checkMatches("William        John   Smith", "5764", true, 15, 16, 22, 25);
    }

    public void testMatches_InitialWithSeparator() {
        // wjs matches (W)illiam (J)ohn (S)mith
        checkMatches("William John-Smith", "957", true, 0, 1, 8, 9, 13, 14);
        // wjsmit matches (W)illiam (J)ohn-(OShe)a
        checkMatches("William John-O'Shea", "956743", true, 0, 1, 8, 9, 13, 18);
        // wjohn matches (W)illiam-(John) Smith
        checkMatches("William-John Smith", "95646", true, 0, 1, 8, 12);
        // jsmi matches William (J)ohn-(Smi)th
        checkMatches("William John-Smith", "5764", true, 8, 9, 13, 16);
        // make sure multiple spaces don't mess things up
        checkMatches("William        John---Smith", "5764", true, 15, 16, 22, 25);

        checkMatches("Berkeley Hair-Studio", "788346", true, 14, 20);
    }

    public void testMatches_repeatedSeparators() {
        // Simple match for single token
        checkMatches("John,,,,,Doe", "5646", true, 0, 4);
        // Match across tokens
        checkMatches("John,,,,,Doe", "56463", true, 0, 10);
        // Match token after chain of separators
        checkMatches("John,,,,,Doe", "363", true, 9, 12);
    }

    public void testMatches_LatinMix() {
        // Latin + Chinese characters
        checkMatches("Lee王力Wang宏", "59264", true, 0, 1, 5, 9);
        // Latin + Japanese characters
        checkMatches("千Abcd佳智Efgh佳IJKL", "222333444555", true, 1, 16);
        // Latin + Arabic characters
        checkMatches("Peterعبد الرحمنJames", "752637", true, 0, 1, 15, 20);
    }

    public void testMatches_umlaut() {
        checkMatches("ÄÖÜäöü", "268268", true, 0, 6);
    }

    // TODO: Great if it was treated as "s" or "ss. Figure out if possible without prefix trie?
    @Suppress
    public void testMatches_germanSharpS() {
        checkMatches("ß", "s", true, 0, 1);
        checkMatches("ß", "ss", true, 0, 1);
    }

    // TODO: Add this and make it work
    @Suppress
    public void testMatches_greek() {
        // http://en.wikipedia.org/wiki/Greek_alphabet
        fail("Greek letters aren't supported yet.");
    }

    // TODO: Add this and make it work
    @Suppress
    public void testMatches_cyrillic() {
        // http://en.wikipedia.org/wiki/Cyrillic_script
        fail("Cyrillic letters aren't supported yet.");
    }


    public void testMatches_NumberBasic() {
        // Simple basic examples that start the match from the start of the number
        checkMatchesNumber("5103337596", "510", true, 0, 3);
        checkMatchesNumber("5103337596", "511", false, 0, 0);
        checkMatchesNumber("5103337596", "5103337596", true, 0, 10);
        checkMatchesNumber("123-456-789", "123456789", true, 0, 11);
        checkMatchesNumber("123-456-789", "123456788", false, 0, 0);
        checkMatchesNumber("09999999999", "099", true, 0, 3);
    }

    public void testMatches_NumberWithCountryCode() {
        // These matches should ignore the country prefix
        // USA (+1)
        checkMatchesNumber("+15103337596", "5103337596", true, 2, 12);
        checkMatchesNumber("+15103337596", "15103337596", true, 0, 12);

        // Singapore (+65)
        checkMatchesNumber("+6591776930", "6591", true, 0, 5);
        checkMatchesNumber("+6591776930", "9177", true, 3, 7);
        checkMatchesNumber("+6591776930", "5917", false, 3, 7);

        // Hungary (+36)
        checkMatchesNumber("+3612345678", "361234", true, 0, 7);
        checkMatchesNumber("+3612345678", "1234", true, 3, 7);

        // Hongkong (+852)
        checkMatchesNumber("+852 2222 2222", "85222222222", true, 0, 14);
        checkMatchesNumber("+852 2222 3333", "2222", true, 5, 9);

        // Invalid (+854)
        checkMatchesNumber("+854 1111 2222", "8541111", true, 0, 9);
        checkMatchesNumber("+854 1111 2222", "1111", false, 0, 0);
    }

    public void testMatches_NumberNANP() {
        // An 11 digit number prefixed with 1 should be matched by the 10 digit number, as well as
        // the 7 digit number (without area code)
        checkMatchesNumber("1-510-333-7596", "5103337596", true, true, 2, 14);
        checkMatchesNumber("1-510-333-7596", "3337596", true, true, 6, 14);

        // Invalid NANP numbers should not be matched
        checkMatchesNumber("1-510-333-759", "510333759", false, true, 0, 0);
        checkMatchesNumber("510-333-759", "333759", false, true, 0, 0);

        // match should fail if NANP flag is switched off
        checkMatchesNumber("1-510-333-7596", "3337596", false, false, 0, 0);

        // A 10 digit number without a 1 prefix should be matched by the 7 digit number
        checkMatchesNumber("(650) 292 2323", "2922323", true, true, 6, 14);
        checkMatchesNumber("(650) 292 2323", "6502922323", true, true, 0, 14);
        // match should fail if NANP flag is switched off
        checkMatchesNumber("(650) 292 2323", "2922323", false, false, 0, 0);
        // But this should still match (since it is the full number)
        checkMatchesNumber("(650) 292 2323", "6502922323", true, false, 0, 14);
    }


    private void checkMatchesNumber(String number, String query, boolean expectedMatches,
            int matchStart, int matchEnd) {
        checkMatchesNumber(number, query, expectedMatches, false, matchStart, matchEnd);
    }

    private void checkMatchesNumber(String number, String query, boolean expectedMatches,
            boolean matchNanp, int matchStart, int matchEnd) {
        final SmartDialMatchPosition pos = SmartDialNameMatcher.matchesNumber(number, query,
                matchNanp);
        assertEquals(expectedMatches, pos != null);
        if (expectedMatches) {
            assertEquals("start", matchStart, pos.start);
            assertEquals("end", matchEnd, pos.end);
        }
    }

    private void checkMatches(String displayName, String query, boolean expectedMatches,
            int... expectedMatchPositions) {
        final SmartDialNameMatcher matcher = new SmartDialNameMatcher(query);
        final ArrayList<SmartDialMatchPosition> matchPositions =
                new ArrayList<SmartDialMatchPosition>();
        final boolean matches = matcher.matchesCombination(
                displayName, query, matchPositions);
        Log.d(TAG, "query=" + query + "  text=" + displayName
                + "  nfd=" + Normalizer.normalize(displayName, Normalizer.Form.NFD)
                + "  nfc=" + Normalizer.normalize(displayName, Normalizer.Form.NFC)
                + "  nfkd=" + Normalizer.normalize(displayName, Normalizer.Form.NFKD)
                + "  nfkc=" + Normalizer.normalize(displayName, Normalizer.Form.NFKC)
                + "  matches=" + matches);
        assertEquals("matches", expectedMatches, matches);
        final int length = expectedMatchPositions.length;
        assertEquals(length % 2, 0);
        if (matches) {
            for (int i = 0; i < length/2; i++) {
                assertEquals("start", expectedMatchPositions[i * 2], matchPositions.get(i).start);
                assertEquals("end", expectedMatchPositions[i * 2 + 1], matchPositions.get(i).end);
            }
        }
    }

}