/*

* 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:  Source code file for contact adapter's datastore unit tests.

*

*/

#include "CNSmlDummyDataStore_Test.h"

#include "CNSmlAdapterTestUtilities.h"

#include <EUnitMacros.h>

#include <EUnitDecorators.h>

#include <cntdb.h>

#include <cntitem.h>

#include <cntfield.h>

#include <cntfldst.h>

#include <s32mem.h>

#include <BAUTILS.H>

#include <SmlDataFormat.h>

#include <CVPbkContactViewDefinition.h>

#include <CVPbkContactStoreUriArray.h>

#include <tvpbkcontactstoreuriptr.h>

#include <VPbkContactStoreUris.h>

#include "nsmldummydatastore.h"

#include "cnsmldummydatastorewaiter.h"

namespace

    {

    // "abld test export" will export test data.

    _LIT( KVCardStandard, "c:\\Data\\TestData\\standard.vcf" );

    _LIT( KVCardBasic, "c:\\Data\\TestData\\basic.vcf" );

    _LIT( KVCardBasicInvalid, "c:\\Data\\TestData\\basicInvalid.vcf" );

    _LIT( KVCardMultiOccurence, "c:\\Data\\TestData\\multioccurence.vcf" );

    }

// - Construction -----------------------------------------------------------

CNSmlDummyDataStore_Test* CNSmlDummyDataStore_Test::NewL()

    {

    CNSmlDummyDataStore_Test* self = CNSmlDummyDataStore_Test::NewLC();

    CleanupStack::Pop( self );

    return self;

    }

CNSmlDummyDataStore_Test* CNSmlDummyDataStore_Test::NewLC()

    {

    CNSmlDummyDataStore_Test* self = new( ELeave ) CNSmlDummyDataStore_Test();

    CleanupStack::PushL( self );

    self->ConstructL();

    return self;

    }

CNSmlDummyDataStore_Test::~CNSmlDummyDataStore_Test()

    {

    iWriteStream.Close();

    iFsSession.Close();

    delete iCNSmlDummyDataStore;

    delete iTestUtilities;

    delete iContactDatabase;

    delete iIdArray;

    delete iStoreWaiter;

    }

CNSmlDummyDataStore_Test::CNSmlDummyDataStore_Test()

    {

    }

void CNSmlDummyDataStore_Test::ConstructL()

    {

    CEUnitTestSuiteClass::ConstructL();

    User::LeaveIfError( iFsSession.Connect() );

    }

// from MSmlSyncRelationship

TSmlSyncTaskKey CNSmlDummyDataStore_Test::SyncTaskKey() const

    {

    TSmlSyncTaskKey key = 0;

    return key;

    }

void CNSmlDummyDataStore_Test::OpenReadStreamLC(

        RReadStream& /*aReadStream*/, TUid /*aStreamUid*/)

    {

    }

void CNSmlDummyDataStore_Test::OpenWriteStreamLC(

        RWriteStream& aWriteStream, TUid /*aStreamUid*/ )

    {

    User::LeaveIfError(

            iWriteStream.Replace( iFsSession, _L("store.test"), EFileWrite  ) );

    aWriteStream = iWriteStream;

    CleanupClosePushL( aWriteStream );

    }

TBool CNSmlDummyDataStore_Test::IsStreamPresentL(TUid /*aStreamUid*/) const

    {

    return EFalse;

    }

// - Test methods -----------------------------------------------------------

// ++++++++++++ SETUP METHODS ++++++++++++

void CNSmlDummyDataStore_Test::SetupL(  )

    {

    iCNSmlDummyDataStore = CNSmlDummyDataStore::NewL();

    iTestUtilities = CNSmlAdapterTestUtilities::NewL();

    // Create contact manager

    CVPbkContactStoreUriArray* uriArray = CVPbkContactStoreUriArray::NewLC();

    uriArray->AppendL( TVPbkContactStoreUriPtr( VPbkContactStoreUris::DefaultCntDbUri() ) );

    iContactManager = CVPbkContactManager::NewL(*uriArray);

    CleanupStack::PopAndDestroy(uriArray);

    }

void CNSmlDummyDataStore_Test::SetupAndOpenStoreL(  )

    {

    SetupL();

    iStoreWaiter =

        CNSmlDummyDataStoreWaiter::NewL( *iCNSmlDummyDataStore );

    iStoreWaiter->OpenStoreL( *this );

    iContactDatabase = CContactDatabase::OpenL();

    }

void CNSmlDummyDataStore_Test::SetupPopulateDbAndOpenStoreL()

    {

    SetupAndPopulateContactDbL();

    EUNIT_ASSERT_NO_LEAVE(

    iCNSmlDummyDataStore = CNSmlDummyDataStore::NewL() );

    iStoreWaiter =

        CNSmlDummyDataStoreWaiter::NewL( *iCNSmlDummyDataStore );

    iStoreWaiter->OpenStoreL( *this );

    }

/**

 * Setup contact DB with the following contacts

 *   + Tim Tester (id index 0)

 *   + Teddy McTest (id index 1)

 *   + Tod TestHill (id index 2)

 *   + Simon Software (id index 3)

 *   + Cyd Effect (id index 4)

 *

 */

void CNSmlDummyDataStore_Test::SetupAndPopulateContactDbL()

    {

    if( !iContactDatabase )

        {

        iContactDatabase = CContactDatabase::OpenL();

        }

    if( !iTestUtilities )

        {

        iTestUtilities = CNSmlAdapterTestUtilities::NewL();

        }

    iIdArray = CContactIdArray::NewL();

    // Create test contacts

    iIdArray->AddL( AddContactL( _L( "Tim" ), _L( "Tester" ), _L( "+358501234567" ) ) );

    iIdArray->AddL( AddContactL( _L( "Teddy" ), _L("McTest" ), _L( "+358501112222" ) ) );

    iIdArray->AddL( AddContactL( _L( "Tod" ), _L( "TestHill" ), _L( "+358506666666" ) ) );

    iIdArray->AddL( AddContactL( _L( "Simon" ), _L( "Software" ), _L( "+3585011223333" ) ) );

    iIdArray->AddL( AddContactL( _L( "Cyd" ), _L( "Effect" ), _L( "+358503334444" ) ) );

    }

//// ++++++++++++ TEARDOWN METHODS ++++++++++++

//

void CNSmlDummyDataStore_Test::Teardown()

    {

    delete iCNSmlDummyDataStore;

    iCNSmlDummyDataStore = NULL;

    delete iTestUtilities;

    iTestUtilities = NULL;

    delete iContactManager;

    iContactManager = NULL;

    }

void CNSmlDummyDataStore_Test::TeardownAndCloseStore()

    {

    delete iStoreWaiter;

    iStoreWaiter = NULL;

    delete iContactDatabase;

    iContactDatabase = NULL;

    Teardown();

    }

void CNSmlDummyDataStore_Test::TeardownCloseAndEmptyContactDb()

    {

    delete iStoreWaiter;

    iStoreWaiter = NULL;

    delete iCNSmlDummyDataStore;

    iCNSmlDummyDataStore = NULL;

    TeardownAndEmptyContactDb();

    }

void CNSmlDummyDataStore_Test::TeardownAndEmptyContactDb()

    {

    if (iContactDatabase && iIdArray)

        {

        TRAP_IGNORE( iContactDatabase->DeleteContactsL( *iIdArray ) );

        }

    delete iIdArray;

    iIdArray = NULL;

    delete iContactDatabase;

    iContactDatabase = NULL;

    delete iTestUtilities;

    iTestUtilities = NULL;

    }

// ++++++++++++ UNIT TEST METHODS ++++++++++++

// ====================== CNSmlDummyDataStore ===========================

// MEMORY ALLOCATION FOR DATASTORE

void CNSmlDummyDataStore_Test::NewLL(  )

    {

    CNSmlDummyDataStore* dataStore = NULL;

    EUNIT_ASSERT_NO_LEAVE( dataStore = CNSmlDummyDataStore::NewL() );

    EUNIT_ASSERT( dataStore );

    delete dataStore;

    }

// DoOpenL()

void CNSmlDummyDataStore_Test::DoOpenLL()

    {

    CNSmlDummyDataStoreWaiter* storeWaiter =

    CNSmlDummyDataStoreWaiter::NewLC( *iCNSmlDummyDataStore );

    TRAPD( err, storeWaiter->OpenStoreL( *this ) );

    // Check for leave

    EUNIT_PRINT( _L( "Leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoOpenL() leave" );

    CleanupStack::PopAndDestroy( storeWaiter );

    }

// DoHasSyncHistory()

void CNSmlDummyDataStore_Test::DoHasSyncHistoryL()

    {

    TBool ret = iCNSmlDummyDataStore->DoHasSyncHistory();

    EUNIT_ASSERT_DESC( ret == EFalse, "DoHasSyncHistory returns faulty value");

    }

// DoModifiedItems(), DoCommitChangeInfoL()

void CNSmlDummyDataStore_Test::DoModifiedItemsL()

    {

    const MSmlDataItemUidSet& uiSet = iCNSmlDummyDataStore->DoModifiedItems();

    EUNIT_ASSERT_DESC( uiSet.ItemCount() == 0, "DoModifiedItems returns faulty uid set");

    TInt returnCode = KErrNone;

    TRAPD( err, returnCode = iStoreWaiter->DoCommitChangeInfoL( uiSet ) );

    // Check for leave

    EUNIT_PRINT( _L( "Leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitChangeInfoL() leave" );

    // Check for return code

    EUNIT_PRINT( _L( "Return code: %d" ), returnCode );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitChangeInfoL returns error");

    }

// DoResetChangeInfoL()

void CNSmlDummyDataStore_Test::DoResetChangeInfoLL()

    {

    // Temporarily remove ChangeFinders from Store to prevent leaves

    // when reseting ChangeFinders in code under test.

    CNSmlChangeFinder* cF = iCNSmlDummyDataStore->iChangeFinder;

    iCNSmlDummyDataStore->iChangeFinder = NULL;

    TInt returnCode = KErrNone;

    TRAPD( err, returnCode = iStoreWaiter->DoResetChangeInfoL() );

    // Return ChangeFinders

    iCNSmlDummyDataStore->iChangeFinder = cF;

    // Check for leave

    EUNIT_PRINT( _L( "Leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoResetChangeInfoL() leave" );

    // Check for return code

    EUNIT_PRINT( _L( "Return code: %d" ), returnCode );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoResetChangeInfoL returns error");

    }

// DoCommitChangeInfoL()

void CNSmlDummyDataStore_Test::DoCommitChangeInfoLL()

    {

    TInt returnCode = KErrNone;

    TRAPD( err, returnCode = iStoreWaiter->DoCommitChangeInfoL() );

    // Check for leave

    EUNIT_PRINT( _L( "Leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitChangeInfoL() leave" );

    // Check for return code

    EUNIT_PRINT( _L( "Return code: %d" ), returnCode );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitChangeInfoL returns error");

    }

// Item addition - "Normal" (non-batch) mode

//  (DoCreateItemL(), DoWriteItemL(), DoCommitItemL())

void CNSmlDummyDataStore_Test::AddSingleItemL()

    {

    TBuf8< 20 > type;

    TBuf8< 20 > ver = KNSmlvCard21Ver();

    TInt returnCode = KErrNone;

    iItemUid = KNullContactId;

    // Try creating contact with illegal MIME type

    EUNIT_PRINT( _L("Calling DoCreateItemL(), STEP 1") );

    TRAPD( err, returnCode = iStoreWaiter->DoCreateItemL(

        iItemUid, 0, type, ver ) );

    EUNIT_PRINT( _L( "DoCreateItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( returnCode == KErrNotSupported,

        "DoCreateItemL() accepts illegal MIME" );

    // Create legal contact

    type = KNSmlvCard21Name();

    EUNIT_PRINT( _L("Calling DoCreateItemL(), STEP 2") );

    TRAP( err, returnCode = iStoreWaiter->DoCreateItemL(

        iItemUid, 512, type, ver ) );

    EUNIT_PRINT( _L( "DoCreateItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCreateItemL returns error");

    // Perform item writing to data store

    HBufC8* vCard = iTestUtilities->ReadVCard8FromFileLC( KVCardBasic() );

    EUNIT_PRINT( _L("Calling DoWriteItemL(), STEP 3") );

    // Non-async call (no need for iStoreWaiter usage)

    TRAP( err, iCNSmlDummyDataStore->DoWriteItemL( *vCard ) );

    EUNIT_PRINT( _L( "DoWriteItemL(): leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    CleanupStack::PopAndDestroy( vCard );

    // Commit the contact to be created

    EUNIT_PRINT( _L("Calling DoCommmitItemL(), STEP 4") );

    TRAP( err, returnCode = iStoreWaiter->DoCommitItemL() );

    EUNIT_PRINT( _L( "DoCommitItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitItemL returns error");

    // Check the imported item

    EUNIT_ASSERT_DESC( iItemUid != KNullContactId, "Illegal UID for imported contact" );

    EUNIT_PRINT( _L("Item ID: %d"), iItemUid );

    // Print out the imported vCard

    CContactIdArray* idArr = CContactIdArray::NewLC();

    idArr->AddL( iItemUid );

    CBufFlat* buffer = CBufFlat::NewL( 1024 );

    CleanupStack::PushL( buffer );

    RBufWriteStream writeStream( *buffer );

    CleanupClosePushL( writeStream );

    TRAP( err, iContactDatabase->ExportSelectedContactsL( TUid::Uid(KVersitEntityUidVCard), *idArr,

        writeStream, CContactDatabase::EExcludeUid, Versit::EUTF8CharSet ) );

    EUNIT_ASSERT_EQUALS_DESC( err, KErrNone, "ExportSelectedContactsL leave" );

    CleanupStack::PopAndDestroy( &writeStream );

    EUNIT_PRINT( buffer->Ptr( 0 ) );

    // Do some asserts

    EUNIT_ASSERT( buffer->Ptr( 0 ).Find( _L8("N:Smith;John") ) != KErrNotFound );

    EUNIT_ASSERT( buffer->Ptr( 0 ).Find( _L8("TITLE:Boss") ) != KErrNotFound );

    CleanupStack::PopAndDestroy( 2, idArr );  // buffer, idArr

    // Do cleanup (remove added contact)

    CContactIdArray* tmpArray = CContactIdArray::NewLC();

    tmpArray->AddL( iItemUid );

    TRAP_IGNORE( iContactDatabase->DeleteContactsL( *tmpArray ) );

    CleanupStack::PopAndDestroy( tmpArray );

    }

// Item addition - Batch mode, empty batch

//  (DoBeginBatchL(), DoCommitBatchL())

void CNSmlDummyDataStore_Test::AddEmptyBatchL()

    {

    // DoBeginBatchL()

    TRAPD( err, iCNSmlDummyDataStore->DoBeginBatchL() );

    EUNIT_PRINT( _L( "DoBeginBatchL(): leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoBeginBatchL() leave" );

    // DoCommitBatchL()

    TInt returnCode = KErrNone;

    RArray< TInt > array;

    TRAP( err, returnCode = iStoreWaiter->DoCommitBatchL( array ) );

    EUNIT_PRINT( _L( "DoCommitBatchL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitBatchL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitBatchL returns error");

    }

// Item addition - Batch mode with two contacts

void CNSmlDummyDataStore_Test::AddTwoItemBatchL()

    {

    TBuf8<20> type = KNSmlvCard21Name();

    TBuf8<20> ver = KNSmlvCard21Ver();

    TInt returnCode = KErrNone;

    HBufC8* vCard = NULL;

    // DoBeginBatchL()

    EUNIT_PRINT( _L("Calling DoCreateItemL(), STEP 1") );

    TRAPD( err, iCNSmlDummyDataStore->DoBeginBatchL() );

    EUNIT_PRINT( _L( "DoBeginBatchL(): leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoBeginBatchL() leave" );

    // == == CONTACT ITEM NO 1 == ==

    vCard = iTestUtilities->ReadVCard8FromFileLC( KVCardBasic() );

    EUNIT_PRINT( _L("Calling DoCreateItemL(), STEP 2") );

    TRAP( err, returnCode = iStoreWaiter->DoCreateItemL(

        iItemUid, vCard->Size(), type, ver ) );

    EUNIT_PRINT( _L( "DoCreateItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCreateItemL returns error");

    // Perform item writing to data store

    EUNIT_PRINT( _L("Calling DoWriteItemL(), STEP 3") );

    // Non-async call (no need for iStoreWaiter usage)

    TRAP( err, iCNSmlDummyDataStore->DoWriteItemL( *vCard ) );

    EUNIT_PRINT( _L( "DoWriteItemL(): leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    CleanupStack::PopAndDestroy( vCard );

    vCard = NULL;

    // Commit the contact to be created

    EUNIT_PRINT( _L("Calling DoCommitItemL(), STEP 4") );

    TRAP( err, returnCode = iStoreWaiter->DoCommitItemL() );

    EUNIT_PRINT( _L( "DoCommitItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitItemL returns error");

    // == == CONTACT ITEM NO 2 == ==

    vCard = iTestUtilities->ReadVCard8FromFileLC( KVCardStandard() );

    EUNIT_PRINT( _L("Calling DoCreateItemL(), STEP 5") );

    TRAP( err, returnCode = iStoreWaiter->DoCreateItemL(

        iItemUid2, vCard->Size(), type, ver ) );

    EUNIT_PRINT( _L( "DoCreateItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCreateItemL returns error" );

    // Perform item writing to data store

    EUNIT_PRINT( _L("Calling DoWriteItemL(), STEP 6") );

    // Non-async call (no need for iStoreWaiter usage)

    TRAP( err, iCNSmlDummyDataStore->DoWriteItemL( *vCard ) );

    EUNIT_PRINT( _L( "DoWriteItemL(): leave code: %d" ), err );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCreateItemL() leave" );

    CleanupStack::PopAndDestroy( vCard );

    // Commit the contact to be created

    EUNIT_PRINT( _L("Calling DoCommitItemL(), STEP 7") );

    TRAP( err, returnCode = iStoreWaiter->DoCommitItemL() );

    EUNIT_PRINT( _L( "DoCommitItemL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitItemL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitItemL returns error" );

    // Commit batch

    RArray<TInt> array;

    EUNIT_PRINT( _L("Calling DoCommitBatchL(), STEP 8") );

    TRAP( err, returnCode = iStoreWaiter->DoCommitBatchL( array ) );

    EUNIT_PRINT( _L( "DoCommitBatchL(): leave code: %d, return code: %d" ),

        err, returnCode );

    EUNIT_ASSERT_DESC( err == KErrNone, "DoCommitBatchL() leave" );

    EUNIT_ASSERT_DESC( returnCode == KErrNone, "DoCommitBatchL returns error" );

    // Verify results

    for( TInt i = 0; i < array.Count(); i++ )

        {

        EUNIT_PRINT( _L( "Contact: %d, Error code: %d "), i, array[ i ] );

        EUNIT_ASSERT( array[ i ] == KErrNone );

        }

    EUNIT_ASSERT_DESC( array.Count() == 2, "Wrong number of contacts added" );

    EUNIT_PRINT( _L("Item 1 ID = %d"), iItemUid );

    EUNIT_PRINT( _L("Item 2 ID = %d"), iItemUid2 );

    // Cleanup

    array.Reset();

    array.Close();

    CContactIdArray* tmpArray = CContactIdArray::NewLC();

    tmpArray->AddL( iItemUid );

    tmpArray->AddL( iItemUid2 );

    TRAP_IGNORE( iContactDatabase->DeleteContactsL( *tmpArray ) );

    CleanupStack::PopAndDestroy( tmpArray );

    }

// Item addition - cancel Batch mode 

void CNSmlDummyDataStore_Test::CancelBatchL()

    {

    TBuf8<20> type = KNSmlvCard21Name();

    TBuf8<20> ver = KNSmlvCard21Ver();