summaryrefslogtreecommitdiff
path: root/tests/src/com/android/dialer/calllog/ContactInfoHelperTest.java
blob: 6d3e86042619d85a694af60c25bb1b37f5eb37c9 (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
/*
 * 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.dialer.calllog;

import android.net.Uri;
import android.provider.ContactsContract.Contacts;
import android.provider.ContactsContract.PhoneLookup;
import android.test.AndroidTestCase;
import android.test.suitebuilder.annotation.MediumTest;

import com.android.contacts.common.test.mocks.ContactsMockContext;
import com.android.contacts.common.test.mocks.MockContentProvider.Query;

import junit.framework.Assert;

@MediumTest
public class ContactInfoHelperTest extends AndroidTestCase {

    private static final String TEST_COUNTRY_ISO = "US";
    private static final String TEST_DISPLAY_NAME = "Display Name";
    private static final String TEST_DISPLAY_NAME_ALTERNATIVE = "Name, Display";
    private static final String[] TEST_DISPLAY_NAME_ALTERNATIVE_ROW = new String[]{null,
            TEST_DISPLAY_NAME_ALTERNATIVE};
    private static final String TEST_LOOKUP_KEY = "lookupKey";
    private static final String[] TEST_LOOKUP_ROW = new String[]{null, TEST_DISPLAY_NAME,
            null, null, null, null, null, TEST_LOOKUP_KEY, null};

    private Uri displayNameAlternativeUri;
    private ContactsMockContext mContext;
    private ContactInfo mContactInfo;
    private ContactInfoHelper mContactInfoHelper;

    @Override
    public void setUp() throws Exception {
        super.setUp();

        displayNameAlternativeUri = Uri.withAppendedPath(Contacts.CONTENT_LOOKUP_URI,
                TEST_LOOKUP_KEY);
        mContext = new ContactsMockContext(getContext());
        mContactInfo = new ContactInfo();
        mContactInfo.name = TEST_DISPLAY_NAME;
        mContactInfo.nameAlternative = TEST_DISPLAY_NAME_ALTERNATIVE;
        mContactInfoHelper = new ContactInfoHelper(mContext, TEST_COUNTRY_ISO);
    }

    public void testLookupContactFromUriNullUri() {
        Assert.assertNull(mContactInfoHelper.lookupContactFromUri(null));
    }

    public void testLookupContactFromUriNoResults() {
        setUpQueryExpectations(PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI,
                PhoneQuery.PHONE_LOOKUP_PROJECTION);

        Assert.assertEquals(ContactInfo.EMPTY, mContactInfoHelper.lookupContactFromUri(
                PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI));
        mContext.verify();
    }

    public void testLookupContactFromUriNoDisplayNameAlternative() {
        setUpQueryExpectations(PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI,
                PhoneQuery.PHONE_LOOKUP_PROJECTION, TEST_LOOKUP_ROW);
        setUpQueryExpectations(displayNameAlternativeUri,
                PhoneQuery.DISPLAY_NAME_ALTERNATIVE_PROJECTION);

        ContactInfo contactInfo = mContactInfoHelper.lookupContactFromUri(
                PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI);
        Assert.assertEquals(TEST_DISPLAY_NAME, contactInfo.name);
        Assert.assertNull(contactInfo.nameAlternative);
        mContext.verify();
    }

    public void testLookupContactFromUriWithDisplayNameAlternative() {
        setUpQueryExpectations(PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI,
                PhoneQuery.PHONE_LOOKUP_PROJECTION, TEST_LOOKUP_ROW);
        setUpQueryExpectations(displayNameAlternativeUri,
                PhoneQuery.DISPLAY_NAME_ALTERNATIVE_PROJECTION, TEST_DISPLAY_NAME_ALTERNATIVE_ROW);

        ContactInfo contactInfo = mContactInfoHelper.lookupContactFromUri(
                PhoneLookup.ENTERPRISE_CONTENT_FILTER_URI);
        Assert.assertEquals(TEST_DISPLAY_NAME, contactInfo.name);
        Assert.assertEquals(TEST_DISPLAY_NAME_ALTERNATIVE, contactInfo.nameAlternative);
        mContext.verify();
    }

    /*
    * Sets up query expectations to return the given row for all queries for the given
    * uri and projection. If row is null, an empty cursor is returned for query calls
    */
    private void setUpQueryExpectations(Uri uri, String[] projection, String...row) {
        Query query = mContext.getContactsProvider().expectQuery(uri)
                .withProjection(projection).withAnySelection().withAnySortOrder();
        if (row == null || row.length == 0) {
            query.returnEmptyCursor();
            return;
        }
        query.returnRow(row);
    }
}