1/*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License
15 */
16
17package com.android.providers.contacts.util;
18
19import static com.android.providers.contacts.util.DbQueryUtils.checkForSupportedColumns;
20import static com.android.providers.contacts.util.DbQueryUtils.concatenateClauses;
21import static com.android.providers.contacts.util.DbQueryUtils.escapeLikeValue;
22
23import android.content.ContentValues;
24import android.test.suitebuilder.annotation.SmallTest;
25
26import com.android.common.content.ProjectionMap;
27import com.android.providers.contacts.EvenMoreAsserts;
28
29import junit.framework.TestCase;
30
31/**
32 * Unit tests for the {@link DbQueryUtils} class.
33 * Run the test like this:
34 * <code>
35 * runtest -c com.android.providers.contacts.util.DBQueryUtilsTest contactsprov
36 * </code>
37 */
38@SmallTest
39public class DBQueryUtilsTest extends TestCase {
40    public void testGetEqualityClause() {
41        assertEquals("(foo = 'bar')", DbQueryUtils.getEqualityClause("foo", "bar"));
42        assertEquals("(foo = 2)", DbQueryUtils.getEqualityClause("foo", 2));
43    }
44
45    public void testGetInEqualityClause() {
46        assertEquals("(foo != 2)", DbQueryUtils.getInequalityClause("foo", 2));
47    }
48
49    public void testConcatenateClauses() {
50        assertEquals("(first)", concatenateClauses("first"));
51        assertEquals("(first) AND (second)", concatenateClauses("first", "second"));
52        assertEquals("(second)", concatenateClauses("second", null));
53        assertEquals("(second)", concatenateClauses(null, "second"));
54        assertEquals("(second)", concatenateClauses(null, "second", null));
55        assertEquals("(a) AND (b) AND (c)", concatenateClauses(null, "a", "b", null, "c"));
56        assertEquals("(WHERE \"a\" = \"b\")", concatenateClauses(null, "WHERE \"a\" = \"b\""));
57    }
58
59    public void testCheckForSupportedColumns() {
60        final ProjectionMap projectionMap = new ProjectionMap.Builder()
61                .add("A").add("B").add("C").build();
62        final ContentValues values = new ContentValues();
63        values.put("A", "?");
64        values.put("C", "?");
65        // No exception expected.
66        checkForSupportedColumns(projectionMap, values);
67        // Expect exception for invalid column.
68        EvenMoreAsserts.assertThrows(IllegalArgumentException.class, new Runnable() {
69            @Override
70            public void run() {
71                values.put("D", "?");
72                checkForSupportedColumns(projectionMap, values);
73            }
74        });
75    }
76
77    public void testEscapeLikeValuesEscapesUnderscores() {
78        StringBuilder sb = new StringBuilder();
79        DbQueryUtils.escapeLikeValue(sb, "my_test_string", '\\');
80        assertEquals("my\\_test\\_string", sb.toString());
81
82        sb = new StringBuilder();
83        DbQueryUtils.escapeLikeValue(sb, "_test_", '\\');
84        assertEquals("\\_test\\_", sb.toString());
85    }
86
87    public void testEscapeLikeValuesEscapesPercents() {
88        StringBuilder sb = new StringBuilder();
89        escapeLikeValue(sb, "my%test%string", '\\');
90        assertEquals("my\\%test\\%string", sb.toString());
91
92        sb = new StringBuilder();
93        escapeLikeValue(sb, "%test%", '\\');
94        assertEquals("\\%test\\%", sb.toString());
95    }
96
97    public void testEscapeLikeValuesNoChanges() {
98        StringBuilder sb = new StringBuilder();
99        escapeLikeValue(sb, "my test string", '\\');
100        assertEquals("my test string", sb.toString());
101    }
102}
103