/*
* 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:
*
*/
/*
 * castoragetestremove.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"


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

// ---------------------------------------------------------------------------
//
// ---------------------------------------------------------------------------
//
void CCAStorageTest::testRemoveOneEntryWithIcon()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;
//    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewL(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         CleanupStack::PushL(query);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromIconTable);
         query->PrepareL();
         TInt countIconsBeforeRemove(0);
         query->ExecuteL(countIconsBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // etry with id = 3 has icon
         TInt entryId = 3;
         entriesToRemoveArray.AppendL(entryId);
         mStorage->RemoveL(entriesToRemoveArray);


         // Check result after remove
         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesAfterRemove(0);
         query->ExecuteL(countEntriesAfterRemove);
         CleanupStack::PopAndDestroy(query);

    if ((countEntriesBeforeRemove - countEntriesAfterRemove) == entriesToRemoveArray.Count()) {
        err = KErrNone;
    }

query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
query->SetQueryL(KSQLGetIdCountFromIconTable);
query->PrepareL();
TInt countIconsAfterRemove(0);
query->ExecuteL(countIconsAfterRemove);
CleanupStack::PopAndDestroy(query);

if (!err) {
    if ((countIconsBeforeRemove - countIconsAfterRemove) == 1) {
            err = KErrNone;
        } else {
            err = KErrGeneral;
        }
    }

    CleanupStack::PopAndDestroy(&entriesToRemoveArray);
//        );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}

void CCAStorageTest::testRemoveOneEntryWhenIconIsUseByOtherEntry()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;
    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromIconTable);
         query->PrepareL();
         TInt countIconsBeforeRemove(0);
         query->ExecuteL(countIconsBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // etry with id = 20 has same icon as entry with id = 21
         entriesToRemoveArray.AppendL(20);
         mStorage->RemoveL(entriesToRemoveArray);


         // Check result after remove
         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesAfterRemove(0);
         query->ExecuteL(countEntriesAfterRemove);
         CleanupStack::PopAndDestroy(query);

    if ((countEntriesBeforeRemove - countEntriesAfterRemove) == entriesToRemoveArray.Count()) {
    err = KErrNone;
}

query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
query->SetQueryL(KSQLGetIdCountFromIconTable);
query->PrepareL();
TInt countIconsAfterRemove(0);
query->ExecuteL(countIconsAfterRemove);
CleanupStack::PopAndDestroy(query);

if (countIconsBeforeRemove == countIconsAfterRemove) {
    err = KErrNone;
} else {
    err = KErrGeneral;
}

CleanupStack::PopAndDestroy(&entriesToRemoveArray);
    );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}

void CCAStorageTest::testRemoveTwoEntriesWithSameIcon()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;
    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromIconTable);
         query->PrepareL();
         TInt countIconsBeforeRemove(0);
         query->ExecuteL(countIconsBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // both entries with same icon
         entriesToRemoveArray.AppendL(19);
         entriesToRemoveArray.AppendL(20);
         mStorage->RemoveL(entriesToRemoveArray);

         // Check result after remove
         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesAfterRemove(0);
         query->ExecuteL(countEntriesAfterRemove);
         CleanupStack::PopAndDestroy(query);

    if ((countEntriesBeforeRemove - countEntriesAfterRemove) == entriesToRemoveArray.Count()) {
    err = KErrNone;
}

query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
query->SetQueryL(KSQLGetIdCountFromIconTable);
query->PrepareL();
TInt countIconsAfterRemove(0);
query->ExecuteL(countIconsAfterRemove);
CleanupStack::PopAndDestroy(query);

if (countIconsBeforeRemove - countIconsAfterRemove == 1) {
    err = KErrNone;
} else {
    err = KErrGeneral;
}

CleanupStack::PopAndDestroy(&entriesToRemoveArray);
    );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}

void CCAStorageTest::testRemoveOneEntryWithoutIcon()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;
    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // id = 12 app without icon
         TInt entryId = 12;
         entriesToRemoveArray.AppendL(entryId);
         mStorage->RemoveL(entriesToRemoveArray);

         // Check result after remove
         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesAfterRemove(0);
         query->ExecuteL(countEntriesAfterRemove);
         CleanupStack::PopAndDestroy(query);

    if ((countEntriesBeforeRemove - countEntriesAfterRemove) == entriesToRemoveArray.Count()) {
    err = KErrNone;
}

CleanupStack::PopAndDestroy(&entriesToRemoveArray);
    );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}

void CCAStorageTest::testRemove5Entries()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;

    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // 5 entries
         entriesToRemoveArray.AppendL(4);
         entriesToRemoveArray.AppendL(5);
         entriesToRemoveArray.AppendL(10);
         entriesToRemoveArray.AppendL(13);
         entriesToRemoveArray.AppendL(14);

         mStorage->RemoveL(entriesToRemoveArray);

         // Check result after remove
         query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesAfterRemove(0);
         query->ExecuteL(countEntriesAfterRemove);
         CleanupStack::PopAndDestroy(query);

    if ((countEntriesBeforeRemove - countEntriesAfterRemove) == entriesToRemoveArray.Count()) {
    err = KErrNone;
}

CleanupStack::PopAndDestroy(&entriesToRemoveArray);
    );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}

void CCAStorageTest::testRemoveEntriesWithIdOutOFBound()
{
    TInt err(KErrGeneral);
    __UHEAP_MARK;
    TRAP(err,
         // Check database before remove
         CCaSqlQuery *query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
         query->SetQueryL(KSQLGetIdCountFromEntryTable);
         query->PrepareL();
         TInt countEntriesBeforeRemove(0);
         query->ExecuteL(countEntriesBeforeRemove);
         CleanupStack::PopAndDestroy(query);

         // Remove from database
         RArray<TInt> entriesToRemoveArray;
         CleanupClosePushL(entriesToRemoveArray);
         // 5 entries
         entriesToRemoveArray.AppendL(4);
         entriesToRemoveArray.AppendL(5);
         entriesToRemoveArray.AppendL(56); // Id out of bound
         entriesToRemoveArray.AppendL(13);
         entriesToRemoveArray.AppendL(87); // Id out of bound

         // shold leave with KErrNotFound
         TRAPD(err2, mStorage->RemoveL(entriesToRemoveArray));
    if (err2 == KErrNotFound) {
    err = KErrNone;
} else {
    err = KErrGeneral;
}


// Check result after remove
query = CCaSqlQuery::NewLC(((CCaSqLiteStorage *)mStorage)->iSqlDb);
query->SetQueryL(KSQLGetIdCountFromEntryTable);
query->PrepareL();
TInt countEntriesAfterRemove(0);
query->ExecuteL(countEntriesAfterRemove);
CleanupStack::PopAndDestroy(query);

if (countEntriesBeforeRemove == countEntriesAfterRemove) {
    err = KErrNone;
} else {
    err = KErrGeneral;
}

CleanupStack::PopAndDestroy(&entriesToRemoveArray);
    );
    __UHEAP_MARKEND;
    QCOMPARE(err, KErrNone);
}