/****************************************************************************

**

** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the Qt Mobility Components.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "ut_qtcontacts_trackerplugin.h"

#include <QMap>

#include <QPair>

#include <QUuid>

#include <QtTracker/Tracker>

#include <QtTracker/ontologies/nco.h>

#include <QtTracker/ontologies/nie.h>

#include <qcontactfilters.h>

#include <QContactChangeLogFilter>

#include <qtcontacts.h>

#include <trackerchangelistener.h>

#include <qcontactrelationshipsaverequest.h>

#include <qcontactrelationshipfetchrequest.h>

#include <qtrackercontactidfetchrequest.h>

#include <qcontacttrackerbackend_p.h>

#include <qtrackercontactasyncrequest.h>

#include "contactmanager.h"

// update this when creating debian package

const QString PATH_TO_SPARQL_TESTS("./ut_qtcontacts_trackerplugin_data");

ut_qtcontacts_trackerplugin::ut_qtcontacts_trackerplugin()

{

}

void ut_qtcontacts_trackerplugin::initTestCase()

{

    QMap<QString, QString> trackerEngineParams;

    trackerEngine = new QContactTrackerEngine(trackerEngineParams);

    errorMap = new QMap<int, QContactManager::Error>();

}

void ut_qtcontacts_trackerplugin::testContacts()

{

    QContact c1;

    QContact c2;

    trackerEngine->saveContact(&c1, error);

    trackerEngine->saveContact(&c2, error);

    QVERIFY2((error == QContactManager::NoError),"Saving contact");

    QList<QContactLocalId> contacts = trackerEngine->contactIds(queryFilter, sortOrders, error);

    QVERIFY2(contacts.contains(c1.localId()), "Previously added contact is not found");

    QVERIFY2(contacts.contains(c2.localId()), "Previously added contact is not found");

}

void ut_qtcontacts_trackerplugin::testContact()

{

    // Test invalid contact id

    QContact invalidContact = trackerEngine->contact_impl( -1, QStringList(), error);

    QVERIFY(error != QContactManager::NoError);

    // Add a contact

    QContact newContact;

    const QContactLocalId oldid = newContact.localId();

    QVERIFY( trackerEngine->saveContact( &newContact, error ) );

    QContactLocalId id = newContact.localId();

    QVERIFY( id != oldid );

    // Find the added contact

    QContact c = trackerEngine->contact_impl( id, QStringList(), error );

    QVERIFY( c.localId() == newContact.localId() );

}

void ut_qtcontacts_trackerplugin::testSaveName()

{

    QContact c;

    QContactLocalId initialId = c.localId();

    int detailsAdded = 0;

    QMap<QString,QString> nameValues;

    QContactName name;

    nameValues.insert(QLatin1String(QContactName::FieldPrefix), "Mr");

    nameValues.insert(QLatin1String(QContactName::FieldFirst), "John");

    nameValues.insert(QLatin1String(QContactName::FieldMiddle), "Rupert");

    nameValues.insert(QLatin1String(QContactName::FieldLast), "Doe");

//    nameValues.insert(QContactName::FieldSuffix, "III");

    foreach (QString field, nameValues.keys()) {

        name.setValue(field, nameValues.value(field));

    }

    c.saveDetail(&name);

    QContactNickname nick;

    nick.setValue(QLatin1String(QContactNickname::FieldNickname), "Johnny");

    c.saveDetail(&nick);

    QVERIFY(c.detail<QContactName>().prefix() == "Mr");

    QVERIFY(c.detail<QContactName>().firstName() == "John");

    QVERIFY(c.detail<QContactName>().middleName() == "Rupert");

    QVERIFY(c.detail<QContactName>().lastName() == "Doe");

    QVERIFY(c.detail<QContactNickname>().nickname() == "Johnny");

    detailsAdded++;

    trackerEngine->saveContact(&c, error);

    QCOMPARE(error,  QContactManager::NoError);

    QVERIFY(c.localId() != initialId);

    QContact contact = this->contact(c.localId());

    QList<QContactName> details = contact.details<QContactName>();

    QList<QContactNickname> details2 = contact.details<QContactNickname>();

    QCOMPARE(details.count(), detailsAdded);

    QCOMPARE(details2.count(), detailsAdded);

    // Name is unique

    foreach(QString field, nameValues.keys()) {

        QCOMPARE(details.at(0).value(field), nameValues.value(field));

    }

    QCOMPARE(details2.at(0).value(QLatin1String(QContactNickname::FieldNickname)), QString("Johnny"));

    // Try changing the name of the saved contact.

    {

        QMap<QString,QString> nameValues;

        QContactName name = c.detail<QContactName>();

        nameValues.insert(QLatin1String(QContactName::FieldPrefix), "Mr2");

        nameValues.insert(QLatin1String(QContactName::FieldFirst), "John2");

        nameValues.insert(QLatin1String(QContactName::FieldMiddle), "Rupert2");

        nameValues.insert(QLatin1String(QContactName::FieldLast), "");

        //    nameValues.insert(QContactName::FieldSuffix, "III");

        foreach (QString field, nameValues.keys()) {

            name.setValue(field, nameValues.value(field));

        }

        c.saveDetail(&name);

        QContactNickname nick = c.detail<QContactNickname>();

        nick.setValue(QLatin1String(QContactNickname::FieldNickname), "Johnny2");

        c.saveDetail(&nick);

        QVERIFY(trackerEngine->saveContact(&c, error));

        QCOMPARE(error,  QContactManager::NoError);

        QVERIFY(c.localId() != initialId);

        QContact contact = trackerEngine->contact_impl(c.localId(), QStringList(), error);

        QCOMPARE(error,  QContactManager::NoError);

        QList<QContactName> details = contact.details<QContactName>();

        QList<QContactNickname> details2 = contact.details<QContactNickname>();

        QCOMPARE(details.count(), detailsAdded);

        QCOMPARE(details2.count(), detailsAdded);

        // Name is unique

        foreach(QString field, nameValues.keys()) {

            QCOMPARE(details.at(0).value(field), nameValues.value(field));

        }

        QCOMPARE(details2.at(0).value(QLatin1String(QContactNickname::FieldNickname)), QString("Johnny2"));

        // now try to add new name detail fails - this is how currently unique fields are implemented

        // so cover it in unit tests

        QContactName name1;

        name1.setValue(QContactName::FieldFirst, "Something that wont be stored as name is unique");

        c.saveDetail(&name1);

        // validate that unique name is not saved

        QVERIFY(!trackerEngine->saveContact(&c, error));

        details = contact.details<QContactName>();

        details2 = contact.details<QContactNickname>();

        QCOMPARE(details.count(), detailsAdded);

        QCOMPARE(details2.count(), detailsAdded);

    }

}

void ut_qtcontacts_trackerplugin::testSavePhoneNumber()

{

    // use the same values for 2 contacts

    for (int i = 0; i <2; i++ )

    {

    QContact c;

    QContactLocalId initialId = c.localId();

    int detailsAdded = 0;

    QContactName name;

    name.setFirstName("I have phone numbers");

    name.setLastName("Girl");

    c.saveDetail(&name);

    // key: phonenumber; value: context,subtype

    QMap<QString,QPair<QString,QString> > phoneValues;

    phoneValues.insert("(704)486-6472", QPair<QString,QString>(QLatin1String(QContactDetail::ContextHome), QString()));

    phoneValues.insert("(765)957-1663", QPair<QString,QString>(QLatin1String(QContactDetail::ContextHome), QString()));

    phoneValues.insert("(999)888-1111", QPair<QString,QString>(QLatin1String(QContactDetail::ContextHome),

                                                               QLatin1String(QContactPhoneNumber::SubTypeMobile)));

    phoneValues.insert("(792)123-6113", QPair<QString,QString>(QLatin1String(QContactDetail::ContextWork), QString()));

    phoneValues.insert("(918)491-7361", QPair<QString,QString>(QLatin1String(QContactDetail::ContextWork),

                                                  QLatin1String(QContactPhoneNumber::SubTypeMobile)));

    phoneValues.insert("(412)670-1514", QPair<QString,QString>(QLatin1String(QContactDetail::ContextWork),

                                                  QLatin1String(QContactPhoneNumber::SubTypeCar)));

    QMap<QString,QPair<QString,QString> > formattedPhoneValues;

    foreach (QString number, phoneValues.keys()) {

        QContactPhoneNumber phone;

        phone.setNumber(number);

        // Stripped automatically on saving RFC 3966 visual-separators reg exp "[(|-|.|)| ]"

        formattedPhoneValues.insert(QString(number).replace( QRegExp("[\\(|" \

                "\\-|" \

                "\\.|" \

                "\\)|" \

                " ]"), ""),phoneValues.value(number));

        if (!phoneValues.value(number).first.isEmpty()) {

            phone.setContexts(phoneValues.value(number).first);

        }

        if (!phoneValues.value(number).second.isEmpty()) {

            phone.setSubTypes(phoneValues.value(number).second);

        }

        c.saveDetail(&phone);

        detailsAdded++;

    }

    trackerEngine->saveContact(&c, error);

    QCOMPARE(error,  QContactManager::NoError);

    QVERIFY(c.localId() != initialId);

    // wait for commit transaction to be done, no signals yet

    for(int i = 0; i < 100; i++)

    {

        usleep(10000);

        QCoreApplication::processEvents();

    }

    // verify with synchronous read too

    QContact contact = trackerEngine->contact_impl(c.localId(), QStringList(), error);

    QCOMPARE(error,  QContactManager::NoError);

    QList<QContactPhoneNumber> details = contact.details<QContactPhoneNumber>();

    QCOMPARE(details.count(), detailsAdded);

    foreach (QContactPhoneNumber detail, details) {

        // Verify that the stored values and attributes are the same as given

        QVERIFY(formattedPhoneValues.contains(detail.number()));

        QCOMPARE(detail.contexts()[0], formattedPhoneValues.value(detail.number()).first);

        if( formattedPhoneValues.value(detail.number()).second.isEmpty()) // default empty is voice

            QCOMPARE(detail.subTypes()[0], QLatin1String(QContactPhoneNumber::SubTypeVoice));

        else

            QCOMPARE(detail.subTypes()[0], formattedPhoneValues.value(detail.number()).second);

    }

    // edit one of numbers . values, context and subtypes and save again

    QString editedPhoneValue = "+7044866473";

    QContactPhoneNumber phone = details[0];

    phone.setNumber(editedPhoneValue);

    phone.setContexts(QContactDetail::ContextWork);

    phone.setSubTypes(QContactPhoneNumber::SubTypeMobile);

    c = contact;

    c.saveDetail(&phone);

    trackerEngine->saveContact(&c, error);

    QCOMPARE(error,  QContactManager::NoError);

    c = this->contact(c.localId(), QStringList()<<QContactPhoneNumber::DefinitionName);

    details = c.details<QContactPhoneNumber>();

    QCOMPARE(details.count(), detailsAdded);

    bool found = false;

    foreach (QContactPhoneNumber detail, details) {

        if(detail.number() == phone.number())

        {

            found = true;

            QVERIFY(detail.subTypes().contains(QContactPhoneNumber::SubTypeMobile));

            QVERIFY(detail.contexts().contains(QContactPhoneNumber::ContextWork));

            break;

        }

    }

    QVERIFY(found);

    }

}

void ut_qtcontacts_trackerplugin::testPhoneNumberContext()

{

    QContact c;

    QContactPhoneNumber phone;

    phone.setContexts(QContactDetail::ContextHome);

    phone.setNumber("555-888");

    phone.setSubTypes(QContactPhoneNumber::SubTypeMobile);

    c.saveDetail(&phone);

    QContact contactToSave = c;

    // Let's do this all twice, first time save new detail, and next iteration change the context

    for (int iterations = 0; iterations < 2; iterations++) {

        QVERIFY(trackerEngine->saveContact(&contactToSave, error));

        // wait for commit transaction to be done, no signals yet

        for(int i = 0; i < 100; i++) {

            usleep(10000);

            QCoreApplication::processEvents();

        }

        QContactFetchRequest request;

        QContactLocalIdFilter filter;

        QList<QContactLocalId> ids;

        ids.append(contactToSave.localId());

        filter.setIds(ids);

        request.setFilter(filter);

        QStringList details;

        details << QContactPhoneNumber::DefinitionName;

        request.setDefinitionRestrictions(details);

        Slots slot;

        QObject::connect(&request, SIGNAL(resultsAvailable()),

                &slot, SLOT(resultsAvailable()));

        trackerEngine->startRequest(&request);

        for(int i = 0; i < 100; i++) {

            usleep(100000);

            QCoreApplication::processEvents();

            if(request.isFinished() )

                break;

        }

        // if it takes more, then something is wrong

        QVERIFY(request.isFinished());

        QVERIFY(!slot.contacts.isEmpty());

        QContact contactToTest;

        foreach (QContact savedContact, slot.contacts) {

            if (savedContact.localId() == contactToSave.localId()) {

                contactToTest = savedContact;

            }

        }

        QVERIFY(contactToTest.localId() == contactToSave.localId()); // Just to be sure we got the saved contact

        qDebug()<<contactToTest.details<QContactPhoneNumber>().count();

        QVERIFY(contactToTest.details<QContactPhoneNumber>().count() == 1);

        if (0 == iterations) {

            // perform context change

            QContactPhoneNumber phoneToEdit = contactToTest.detail<QContactPhoneNumber>();

            phoneToEdit.setContexts(QContactDetail::ContextWork);

            contactToTest.saveDetail(&phoneToEdit);

            contactToSave = contactToTest;

        }

        QVERIFY(contactToTest.details<QContactPhoneNumber>().count() == 1);

    }

}

void ut_qtcontacts_trackerplugin::testWritingOnlyWorkMobile()

{

    QContact c;

    QContactPhoneNumber phone;

    phone.setContexts(QContactDetail::ContextWork);

    phone.setNumber("555999");

    phone.setSubTypes(QContactPhoneNumber::SubTypeMobile);

    c.saveDetail(&phone);

    QContact& contactToSave = c;

    QVERIFY(trackerEngine->saveContact(&contactToSave, error));

    // wait for commit transaction to be done, no signals yet

    for(int i = 0; i < 100; i++) {

        usleep(10000);

        QCoreApplication::processEvents();

    }

    QContactFetchRequest request;

    QContactLocalIdFilter filter;

    QList<QContactLocalId> ids;

    ids.append(contactToSave.localId());

    filter.setIds(ids);

    request.setFilter(filter);

    QStringList details;

    details << QContactPhoneNumber::DefinitionName;

    request.setDefinitionRestrictions(details);

    Slots slot;

    QObject::connect(&request, SIGNAL(resultsAvailable()),

            &slot, SLOT(resultsAvailable()));

    trackerEngine->startRequest(&request);

    for(int i = 0; i < 100; i++) {

        usleep(100000);

        QCoreApplication::processEvents();

        if(request.isFinished() )

            break;

    }

    // if it takes more, then something is wrong

    QVERIFY(request.isFinished());

    QVERIFY(!slot.contacts.isEmpty());

    QContact contactToTest;

    foreach (QContact savedContact, slot.contacts) {

        if (savedContact.localId() == c.localId()) {

            contactToTest = savedContact;

        }

    }

    QVERIFY(contactToTest.localId() == c.localId()); // Just to be sure we got the saved contact

    QVERIFY(contactToTest.details<QContactPhoneNumber>().count() == 1);

    QVERIFY(contactToTest.detail<QContactPhoneNumber>().number() == phone.number());

    QVERIFY(contactToTest.detail<QContactPhoneNumber>().subTypes() == phone.subTypes());

    QVERIFY(contactToTest.detail<QContactPhoneNumber>().contexts() == phone.contexts());

}

void ut_qtcontacts_trackerplugin::testSaveAddress()

{

    QContact c;

    QContactName name;

    name.setFirstName("Aruba & Barbados");

    name.setLastName("Girl");

    c.saveDetail(&name);

    QContactLocalId initialId = c.localId();

    int detailsAdded = 0;

    // List of pairs of field-value map and context

    typedef QMap<QString,QString> typeAddress;

    typedef QPair<typeAddress,QString> typeAddressWithContext;

    QList<typeAddressWithContext> addressValues;

    // TODO check status of 137174 and other libqttracker1pre6 bugs before refactoring

    typeAddress values;

    values.insert(QLatin1String(QContactAddress::FieldCountry), "Barbados");

    values.insert(QLatin1String(QContactAddress::FieldPostcode), "55555");

    values.insert(QLatin1String(QContactAddress::FieldStreet), "Martindales Rd");

    values.insert(QLatin1String(QContactAddress::FieldRegion), "Bridgetown");

    addressValues.append(typeAddressWithContext(values, QLatin1String(QContactDetail::ContextHome)));

    values.clear();

    values.insert(QLatin1String(QContactAddress::FieldCountry), "Aruba");

    values.insert(QLatin1String(QContactAddress::FieldPostcode), "44444");

    values.insert(QLatin1String(QContactAddress::FieldStreet), "Brazilie Straat");

    values.insert(QLatin1String(QContactAddress::FieldRegion), "Oranjestad");

    addressValues.append(typeAddressWithContext(values, QLatin1String(QContactDetail::ContextHome)));

    values.clear();

    values.insert(QLatin1String(QContactAddress::FieldCountry), "ArubaWork");

    values.insert(QLatin1String(QContactAddress::FieldPostcode), "44445");

    values.insert(QLatin1String(QContactAddress::FieldStreet), "Sunset Blvd");

    values.insert(QLatin1String(QContactAddress::FieldRegion), "Oranjestad");

    addressValues.append(typeAddressWithContext(values, QLatin1String(QContactDetail::ContextHome)));

    foreach (typeAddressWithContext addressWithContext, addressValues) {

        QContactAddress address;

        foreach (QString field, addressWithContext.first.keys()) {

            address.setValue(field, addressWithContext.first.value(field));

        }

        address.setContexts(addressWithContext.second);

        c.saveDetail(&address);

        detailsAdded++;

    }

    trackerEngine->saveContact(&c, error);

    QCOMPARE(error,  QContactManager::NoError);

    QVERIFY(c.localId() != initialId);

    QContact contact = trackerEngine->contact_impl(c.localId(), QStringList(), error);

    QList<QContactAddress> details = contact.details<QContactAddress>();

    QCOMPARE(details.count(), detailsAdded);

    bool found = false;

    // Test if inserted values are found in some of the details

    foreach (typeAddressWithContext addressWithContext, addressValues) {

        foreach (QContactAddress detail, details) {

            foreach (QString field, addressWithContext.first.keys()) {

                found = (detail.value(field) == addressWithContext.first.value(field));

                if (!found)

                    break;

            }

            if (found)

                break;

        }

        QVERIFY2(found, "Inserted detail was not found in the fetched details");

    }

}

void ut_qtcontacts_trackerplugin::testSaveEmailAddress()

{

    QContact c;

    QContactLocalId initialId = c.localId();

    int detailsAdded = 0;

    QMap<QString,QString> values;

    values.insert("john.does@hotmail.com", QContactDetail::ContextHome);

    values.insert("john.doe@gmail.com", QContactDetail::ContextWork);

    values.insert("john.doe@nokia.com", QContactDetail::ContextWork);

    values.insert("john.doe@johndoe.com", QContactDetail::ContextHome);

    foreach(QString address, values.keys()) {

        QContactEmailAddress emailAddress;

        emailAddress.setEmailAddress(address);

        emailAddress.setContexts(values.value(address));

        c.saveDetail(&emailAddress);

        detailsAdded++;

    }

    QContactName name;

    name.setFirstName("Jo");

    name.setLastName("H N Doe");

    c.saveDetail(&name);

    trackerEngine->saveContact(&c, error);

    QCOMPARE(error,  QContactManager::NoError);

    QVERIFY(c.localId() != initialId);

    QContact contact = trackerEngine->contact_impl(c.localId(), QStringList(), error);

    QList<QContactEmailAddress> details = contact.details<QContactEmailAddress>();

    QCOMPARE(details.count(), detailsAdded);

    foreach (QContactEmailAddress detail, details) {

        QString address = detail.value(QContactEmailAddress::FieldEmailAddress);