/*

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

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:

*

*/

/*

 * castoragetestadd.cpp

 *

 *  Created on: 2009-08-05

 *      Author: michal.czerwiec

 */

#include "castoragetest.h"

#include "castoragefactory.h"

#include "castorage.h"

#include "cainnerentry.h"

#include "casqlquery.h"

#include "casqlitestorage.h"

#include "castoragetestsqlcommands.h"

#include "caarraycleanup.inl"

#include <QMap>

#include <QString>

// ============================ MEMBER FUNCTIONS ===============================

// ---------------------------------------------------------------------------

//

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAdd()

{

    __UHEAP_MARK;

    TInt entryId(0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          mStorage->AddL(innerEntry);

          entryId = innerEntry->GetId();

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    QVERIFY(entryId > 0);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddEmptyIcon()

{

    TInt entryId(0);

    __UHEAP_MARK;

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(KNullDesC, KNullDesC, KNullDesC);

          mStorage->AddL(innerEntry);

          entryId = innerEntry->GetId();

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    QVERIFY(entryId > 0);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithIcon()

{

    __UHEAP_MARK;

    TInt entryId(0);

    TInt iconId(0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("qtg_large_applications_user"), KNullDesC, KNullDesC);

          mStorage->AddL(innerEntry);

          entryId = innerEntry->GetId();

          iconId = innerEntry->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    QVERIFY(entryId > 0);

    QVERIFY(iconId > 0);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithExistingIcon()

{

    __UHEAP_MARK;

    TInt entryId1(0);

    TInt iconId1(0);

    TInt entryId2(0);

    TInt iconId2(0);

    TInt countIconsAfterFirstAdd (0);

    TInt countIconsAfterSecondAdd (0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("filename"), _L("skinId"), _L("268458321"));

          mStorage->AddL(innerEntry);

          CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterFirstAdd);

          CleanupStack::PopAndDestroy(query);

          // Add second entry with the some icon

          CCaInnerEntry *innerEntry2 = CCaInnerEntry::NewLC();

          innerEntry2->SetTextL(KTestName2);

          innerEntry2->SetEntryTypeNameL(KTypeName);

          innerEntry2->SetFlags(flag);

          innerEntry2->SetRole(CCaInnerQuery::Group);

          innerEntry2->SetIconDataL(_L("filename"), _L("skinId"), _L("268458321"));

          mStorage->AddL(innerEntry2);

          query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterSecondAdd);

          CleanupStack::PopAndDestroy(query);

          entryId1 = innerEntry->GetId();

          iconId1 = innerEntry->GetIconId();

          entryId2 = innerEntry2->GetId();

          iconId2 = innerEntry2->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry2);

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    // No new icon created; Icon which is the same should be shared by newly added entries

    QCOMPARE(countIconsAfterFirstAdd, countIconsAfterSecondAdd);

    QVERIFY(entryId1 > 0);

    QVERIFY(iconId1 > 0);

    QVERIFY(entryId2 > 0);

    QVERIFY(iconId2 > 0);

    QVERIFY(!(entryId1 == entryId2));

    QCOMPARE(iconId1, iconId2);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithExistingIconOnlyFileName()

{

    __UHEAP_MARK;

    TInt entryId1(0);

    TInt iconId1(0);

    TInt entryId2(0);

    TInt iconId2(0);

    TInt countIconsAfterFirstAdd (0);

    TInt countIconsAfterSecondAdd (0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("qtg_large_applications_user"), KNullDesC, KNullDesC);

          mStorage->AddL(innerEntry);

          CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterFirstAdd);

          CleanupStack::PopAndDestroy(query);

          // Add second entry with the some icon

          CCaInnerEntry *innerEntry2 = CCaInnerEntry::NewLC();

          innerEntry2->SetTextL(KTestName2);

          innerEntry2->SetEntryTypeNameL(KTypeName);

          innerEntry2->SetFlags(flag);

          innerEntry2->SetRole(CCaInnerQuery::Group);

          innerEntry2->SetIconDataL(_L("qtg_large_applications_user"), KNullDesC, KNullDesC);

          mStorage->AddL(innerEntry2);

          query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterSecondAdd);

          CleanupStack::PopAndDestroy(query);

          entryId1 = innerEntry->GetId();

          iconId1 = innerEntry->GetIconId();

          entryId2 = innerEntry2->GetId();

          iconId2 = innerEntry2->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry2);

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    // No new icon created; Icon which is the same should be shared by newly added entries

    QCOMPARE(countIconsAfterFirstAdd, countIconsAfterSecondAdd);

    QVERIFY(entryId1 > 0);

    QVERIFY(iconId1 > 0);

    QVERIFY(entryId2 > 0);

    QVERIFY(iconId2 > 0);

    QVERIFY(!(entryId1 == entryId2));

    QCOMPARE(iconId1, iconId2);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithPartlyMatchingIcon()

{

    __UHEAP_MARK;

    TInt entryId1(0);

    TInt iconId1(0);

    TInt entryId2(0);

    TInt iconId2(0);

    TInt countIconsAfterFirstAdd (0);

    TInt countIconsAfterSecondAdd (0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("filename"), _L("skinId"), _L("268458321"));

          mStorage->AddL(innerEntry);

          CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterFirstAdd);

          CleanupStack::PopAndDestroy(query);

          // Add second entry with the some icon

          CCaInnerEntry *innerEntry2 = CCaInnerEntry::NewLC();

          innerEntry2->SetTextL(KTestName2);

          innerEntry2->SetEntryTypeNameL(KTypeName);

          innerEntry2->SetFlags(flag);

          innerEntry2->SetRole(CCaInnerQuery::Group);

          innerEntry2->SetIconDataL(_L("filename"), _L("skinId2"), _L("111111111"));

          mStorage->AddL(innerEntry2);

          query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterSecondAdd);

          CleanupStack::PopAndDestroy(query);

          entryId1 = innerEntry->GetId();

          iconId1 = innerEntry->GetIconId();

          entryId2 = innerEntry2->GetId();

          iconId2 = innerEntry2->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry2);

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    // New icon should be added

    QCOMPARE(countIconsAfterSecondAdd - countIconsAfterFirstAdd, 1);

    QVERIFY(entryId1 > 0);

    QVERIFY(iconId1 > 0);

    QVERIFY(entryId2 > 0);

    QVERIFY(iconId2 > 0);

    QVERIFY(!(entryId1 == entryId2));

    QVERIFY(!(iconId1 == iconId2));

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithPartlyMatchingIconNotAllParams()

{

    __UHEAP_MARK;

    TInt entryId1(0);

    TInt iconId1(0);

    TInt entryId2(0);

    TInt iconId2(0);

    TInt countIconsAfterFirstAdd (0);

    TInt countIconsAfterSecondAdd (0);

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("filename"), _L("skinId"), KNullDesC);

          mStorage->AddL(innerEntry);

          CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterFirstAdd);

          CleanupStack::PopAndDestroy(query);

          // Add second entry with the some icon

          CCaInnerEntry *innerEntry2 = CCaInnerEntry::NewLC();

          innerEntry2->SetTextL(KTestName2);

          innerEntry2->SetEntryTypeNameL(KTypeName);

          innerEntry2->SetFlags(flag);

          innerEntry2->SetRole(CCaInnerQuery::Group);

          innerEntry2->SetIconDataL(KNullDesC, _L("skinId"), KNullDesC);

          mStorage->AddL(innerEntry2);

          query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterSecondAdd);

          CleanupStack::PopAndDestroy(query);

          entryId1 = innerEntry->GetId();

          iconId1 = innerEntry->GetIconId();

          entryId2 = innerEntry2->GetId();

          iconId2 = innerEntry2->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry2);

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    // New icon should be added

    QCOMPARE(countIconsAfterSecondAdd - countIconsAfterFirstAdd, 1);

    QVERIFY(entryId1 > 0);

    QVERIFY(iconId1 > 0);

    QVERIFY(entryId2 > 0);

    QVERIFY(iconId2 > 0);

    QVERIFY(!(entryId1 == entryId2));

    QVERIFY(!(iconId1 == iconId2));

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddWithNotCompleteIcon()

{

    TInt entryId(0);

    TInt iconId(0);

    __UHEAP_MARK;

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("fake"), KNullDesC, KNullDesC);

          mStorage->AddL(innerEntry);

          entryId = innerEntry->GetId();

          iconId = innerEntry->GetIconId();

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    QVERIFY(entryId > 0);

    QVERIFY(iconId > 0);

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddUpdateIcon()

{

    TInt entryId(0);

    TInt iconId(0);

    TInt updatedEntryId(0);

    TInt updatedIconId(0);

    TInt countIconsAfterAdd (0);

    TInt countIconsAfterUpdate (0);

    RBuf updatedIconFileName;

    RBuf updatedIconSkinId;

    RBuf updatedIconAppId;

    TInt updatedIconIdFormIcon(0);

    __UHEAP_MARK;

    updatedIconFileName.CreateL( KMaxFileName );

    updatedIconSkinId.CreateL( KMaxFileName );

    updatedIconAppId.CreateL( KMaxUidName );

    TRAPD(err,

          CCaInnerEntry *innerEntry = CCaInnerEntry::NewLC();

          innerEntry->SetTextL(KTestName);

          innerEntry->SetEntryTypeNameL(KTypeName);

          innerEntry->SetFlags(flag);

          innerEntry->SetRole(CCaInnerQuery::Group);

          innerEntry->SetIconDataL(_L("filename"), _L("skinId"), _L("268458321"));

          mStorage->AddL(innerEntry);

          // Check db after add with icon

          CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterAdd);

          CleanupStack::PopAndDestroy(query);

          entryId = innerEntry->GetId();

          iconId = innerEntry->GetIconId();

          innerEntry->SetTextL(KTestNameUpdate);

          innerEntry->SetFlags(flagUpdate);

          innerEntry->SetIconDataL( _L("filename2"), _L("skinId2"), _L("537001519"));

          mStorage->AddL(innerEntry);

          // Check db after update with icon

          query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);

          query->SetQueryL(KSQLGetIdCountFromIconTable);

          query->PrepareL();

          query->ExecuteL(countIconsAfterUpdate);

          CleanupStack::PopAndDestroy(query);

          updatedEntryId = innerEntry->GetId();

          updatedIconId = innerEntry->GetIconId();

          updatedIconFileName = innerEntry->Icon()->FileName();

          updatedIconSkinId= innerEntry->Icon()->SkinId();

          updatedIconAppId = innerEntry->Icon()->ApplicationId();

          updatedIconIdFormIcon = innerEntry->Icon()->Id();

          CleanupStack::PopAndDestroy(innerEntry);

         );

    QCOMPARE(err, KErrNone);

    QVERIFY(entryId > 0);

    QVERIFY(iconId > 0);

    QCOMPARE(updatedEntryId, entryId);

    QCOMPARE(updatedIconId, iconId);

    QCOMPARE(updatedIconIdFormIcon, updatedIconId);

    QVERIFY(!updatedIconFileName.Compare(_L("filename2")));

    QVERIFY(!updatedIconSkinId.Compare(_L("skinId2")));

    QVERIFY(!updatedIconAppId.Compare(_L("537001519")));

    // No new icon should be added to DB

    QCOMPARE(countIconsAfterAdd, countIconsAfterUpdate);

    updatedIconFileName.Close();

    updatedIconSkinId.Close();

    updatedIconAppId.Close();    

    __UHEAP_MARKEND;

}

// ---------------------------------------------------------------------------

//

void CCAStorageTest::testAddUpdateIconOnlyFewParams()

{

    TInt entryId(0);

    TInt iconId(0);

    TInt updatedEntryId(0);

    TInt updatedIconId(0);

    TInt countIconsAfterAdd (0);

    TInt countIconsAfterUpdate (0);

    RBuf updatedIconFileName;

    RBuf updatedIconSkinId;

    RBuf updatedIconAppId;

    TInt updatedIconIdFormIcon(0);

    __UHEAP_MARK;

    updatedIconFileName.CreateL( KMaxFileName );

    updatedIconSkinId.CreateL( KMaxFileName );

    updatedIconAppId.CreateL( KMaxUidName );

    TRAPD(err,