/*
 * Copyright (C) 2010 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.quicksearchbox;

import android.app.SearchManager;
import android.database.Cursor;
import android.test.AndroidTestCase;

import java.util.Arrays;
import java.util.HashSet;

/**
 * Tests to verify that custom columns from a suggestion cursor get propagated through QSB properly.
 */
public class SuggestionCursorWithExtrasTest extends AndroidTestCase {

    public void testSuggestionCursorExtraRowString() {
        checkSuggestionCursorExtraColumnValue("extra_text", "Extra text");
    }

    public void testSuggestionCursorExtraRowInteger() {
        checkSuggestionCursorExtraColumnValue("extra_int", 42);
    }

    public void testSuggestionCursorExtraRowFloat() {
        checkSuggestionCursorExtraColumnValue("extra_float", new Float(Math.E));
    }

    public void testSuggestionCursorExtraRowNull() {
        checkSuggestionCursorExtraColumnValue("extra_null", null);
    }

    public void testCursorExtraRowString() {
        checkExtraRowString("stringy-string");
        checkExtraRowString("");
        checkExtraRowString(null);
    }

    private void checkExtraRowString(String value) {
        String column = "extra_string";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getString(c.getColumnIndex(column)));
    }

    public void testCursorExtraRowInt() {
        checkCursorExtraInt(42);
        checkCursorExtraInt(0);
        checkCursorExtraInt(-42);
        checkCursorExtraInt(Integer.MAX_VALUE);
        checkCursorExtraInt(Integer.MIN_VALUE);
    }

    public void checkCursorExtraInt(int value) {
        String column = "extra_int";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getInt(c.getColumnIndex(column)));
    }

    public void testCursorExtraRowDouble() {
        checkCursorExtraRowDouble(Math.PI);
        checkCursorExtraRowDouble(-Math.PI);
        checkCursorExtraRowDouble(0);
        checkCursorExtraRowDouble(Double.MAX_VALUE);
        checkCursorExtraRowDouble(Double.MIN_VALUE);
    }

    public void checkCursorExtraRowDouble(double value) {
        String column = "extra_double";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getDouble(c.getColumnIndex(column)));
    }

    public void testCursorExtraRowFloat() {
        checkCursorExtraRowFloat((float) Math.E);
        checkCursorExtraRowFloat((float) -Math.E);
        checkCursorExtraRowFloat(0f);
        checkCursorExtraRowFloat(Float.MAX_VALUE);
        checkCursorExtraRowFloat(Float.MIN_VALUE);
    }

    public void checkCursorExtraRowFloat(float value) {
        String column = "extra_float";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getFloat(c.getColumnIndex(column)));
    }

    public void testCursorExtraRowLong() {
        checkExtraRowLong(0xfeed0beefl);
        checkExtraRowLong(-0xfeed0beefl);
        checkExtraRowLong(0);
        checkExtraRowLong(Long.MIN_VALUE);
        checkExtraRowLong(Long.MAX_VALUE);
    }

    private void checkExtraRowLong(long value) {
        String column = "extra_long";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getLong(c.getColumnIndex(column)));
    }

    public void testCursorExtraRowShort() {
        checkCursorExtraRowShort((short) 0xabc);
        checkCursorExtraRowShort((short) -0xabc);
        checkCursorExtraRowShort((short) 0);
        checkCursorExtraRowShort(Short.MAX_VALUE);
        checkCursorExtraRowShort(Short.MIN_VALUE);
    }

    private void checkCursorExtraRowShort(short value) {
        String column = "extra_short";
        Cursor c = createCursorWithExtras(column, value);
        assertEquals("Extra column value", value, c.getShort(c.getColumnIndex(column)));
    }

    private Cursor createCursorWithExtras(String columnName, Object columnValue) {
        MockSuggestionProviderCursor expectedCursor = new MockSuggestionProviderCursor(
                new String[]{"_id",   SearchManager.SUGGEST_COLUMN_TEXT_1, columnName });
        expectedCursor.addRow(       0,       "Text 1",                            columnValue);

        // this roughly approcimates what happens to suggestions
        CursorBackedSourceResult suggestions = new CursorBackedSourceResult(
                MockSource.SOURCE_1, "", expectedCursor);
        assertEquals("Number of suggestions", 1, suggestions.getCount());
        suggestions.moveTo(0);
        SuggestionCursorBackedCursor observedCursor = new SuggestionCursorBackedCursor(suggestions);

        assertEquals("Cursor rows", 1, observedCursor.getCount());
        HashSet<String> rows = new HashSet<String>();
        rows.addAll(Arrays.asList(observedCursor.getColumnNames()));
        assertTrue("Extra column present in cursor", rows.contains(columnName));
        observedCursor.moveToFirst();
        return observedCursor;
    }

    private void checkSuggestionCursorExtraColumnValue(String columnName, Object columnValue) {
        MockSuggestionProviderCursor cursor = new MockSuggestionProviderCursor(
                new String[]{"_id",   SearchManager.SUGGEST_COLUMN_TEXT_1, columnName });
        cursor.addRow(       0,       "Text 1",                            columnValue);

        CursorBackedSourceResult suggestions = new CursorBackedSourceResult(
                MockSource.SOURCE_1, "", cursor);

        assertEquals("Suggestions count matches cursor count",
                cursor.getCount(), suggestions.getCount());

        cursor.moveToFirst();
        suggestions.moveTo(0);

        SuggestionExtras extras = suggestions.getExtras();
        assertNotNull("Suggestions extras", extras);
        assertTrue("Extra column missing", extras.getExtraColumnNames().contains(columnName));
        assertTrue("Spurious columns", extras.getExtraColumnNames().size() == 1);
        Object extraValue = extras.getExtra(columnName);
        if (columnValue == null) {
            assertEquals("Wrong extra value", columnValue, extraValue);
        } else {
            assertNotNull("Extra value null", extraValue);
            if (columnValue == null) {
                assertEquals("Extra value wrong", columnValue, extraValue);
            } else {
                assertEquals("Extra value wrong", columnValue.toString(), extraValue);
            }
        }
    }

}