/*

* Copyright (c) 2010 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: 

*

*/

#include "cpixboostertester.h"

#include "ccpixsearcher.h"

#include "CSearchDocument.h"

#include "CCPixIndexer.h"

#include "indevicecfg.h"

_LIT( KTestBaseAppClass, "test" );

_LIT( KTestQualifiedBaseAppClass, "@c:test" );

_LIT( KContentFieldName, "content" ); 

void CTestBoost::setUp()

    {

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

    iSession.DefineVolume(KTestQualifiedBaseAppClass, KTestBaseAppClass );

    iIndexer = CCPixIndexer::NewL(iSession);

    iIndexer->OpenDatabaseL(KTestQualifiedBaseAppClass);

    iSearcher = CCPixSearcher::NewL(iSession, _L( "" CONTENTS_FIELD ) );

    iSearcher->OpenDatabaseL(KTestBaseAppClass);

    }

void CTestBoost::tearDown()

    {

    delete iIndexer;

    delete iSearcher;

    iSession.Close();

    }

void CTestBoost::testNoBoost()

    {

    // 

    // This is the 'control' test. It is used to assert that 

    // without tinkering with boosts, the engine would return 

    // the result with most 'hits' 

    //

    iIndexer->ResetL(); 

    AddDocumentL( 0, _L( "hit hit hit" ), 1.0f );  // first result

    AddDocumentL( 1, _L( "hit hit miss" ), 1.0f ); // second result

    AddDocumentL( 2, _L( "hit miss miss" ), 1.0f );// last result

    if( iSearcher->SearchL( _L( "hit" ), KContentFieldName ) != 3 )

        User::Leave( KErrNotFound );

    AssertHitDocumentL( 0, 0 );  // id 1 is first hit

    AssertHitDocumentL( 1, 1 );  // id 2 is second hit

    AssertHitDocumentL( 2, 2 );  // id 3 is third hit

    }

void CTestBoost::testBoost() 

    {

    //

    // This is the actual boost test. It is used to show that 

    // changing boost value actually affects the rank. The 'normal'

    // document order is reversed with heavy boost usage. 

    //

    iIndexer->ResetL(); 

    AddDocumentL( 0, _L( "hit hit hit" ), 1.0f );    // last result

    AddDocumentL( 1, _L( "hit hit miss" ), 10.0f );  // second result

    AddDocumentL( 2, _L( "hit miss miss" ), 100.0f );// first result

    if( iSearcher->SearchL( _L( "hit" ), KContentFieldName ) != 3 )

        User::Leave( KErrNotFound );

    AssertHitDocumentL( 0, 2 ); // id 1 is third hit

    AssertHitDocumentL( 1, 1 ); // id 2 is second hit

    AssertHitDocumentL( 2, 0 ); // id 3 is first hit

    }

void CTestBoost::testFieldBoost() 

    {

    //

    // This is the actual boost test. It is used to show that 

    // changing boost value actually affects the rank. The 'normal'

    // document order is reversed with heavy boost usage. 

    //

    iIndexer->ResetL(); 

    AddDocumentWithFieldBoostL( 0, _L( "hit hit hit" ), 1.0f );      // last result

    AddDocumentWithFieldBoostL( 1, _L( "hit hit miss" ), 10.0f );  // second result

    AddDocumentWithFieldBoostL( 2, _L( "hit miss miss" ), 100.0f );// first result

    if( iSearcher->SearchL( _L( "hit" ), KContentFieldName ) != 3 )

        User::Leave( KErrNotFound );

    AssertHitDocumentL( 0, 2 ); // id 1 is third hit

    AssertHitDocumentL( 1, 1 ); // id 2 is second hit

    AssertHitDocumentL( 2, 0 ); // id 3 is first hit

    }

void CTestBoost::AssertHitDocumentL( TInt aIndex, TInt aId )

    {

    TBuf<32> buf;

    TInt err = KErrNone;

    buf.AppendNum( aId );

    CSearchDocument* doc = iSearcher->GetDocumentL( aIndex ); 

    CleanupStack::PushL( doc ); 

    if( doc->Id().Compare( buf ) )

        err = KErrNotFound;

    CleanupStack::PopAndDestroy( doc );

    User::LeaveIfError( err );

    }

void CTestBoost::AddDocumentL( TInt aId, const TDesC& aContent, TReal aBoost ) 

    {

    TBuf<32> buf; 

    buf.AppendNum( aId );

    CSearchDocument* doc = CSearchDocument::NewLC( buf, KTestBaseAppClass, aContent ); 

    doc->SetBoost( aBoost ); 

    doc->AddFieldL( KContentFieldName, aContent, CDocumentField::EStoreNo | CDocumentField::EIndexTokenized | CDocumentField::EAggregateYes );

    iIndexer->AddL( *doc ); 

    iIndexer->FlushL();

    CleanupStack::PopAndDestroy( doc ); 

    }

void CTestBoost::AddDocumentWithFieldBoostL( TInt aId, const TDesC& aContent, TReal aBoost ) 

    {

    TBuf<32> buf; 

    buf.AppendNum( aId );

    CSearchDocument* doc = CSearchDocument::NewLC( buf, KTestBaseAppClass, aContent ); 

    doc->AddFieldL( KContentFieldName, 

                    aContent, 

                    CDocumentField::EStoreNo | CDocumentField::EIndexTokenized | CDocumentField::EAggregateYes )

        .SetBoost( aBoost ); // Boost field instead of document

    iIndexer->AddL( *doc );

    iIndexer->FlushL();

    CleanupStack::PopAndDestroy( doc ); 

    }