// Copyright (c) 2003-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:
//

#include "ctestclienthttpget.h"
#include <httpstringconstants.h>
#include "mlocaltestserverobserver.h"
#include "chttpclienttestparams.h"
#include <tinternetdate.h>
_LIT8(KAcceptHeader, "Accept");
_LIT8(KAcceptValue, "text/xml,application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5");
_LIT(KTestHttpClientAPITestTitle, "HTTP client API Unit Test Harness");
_LIT(KTransferProgress, "File size: %d kb , Downloaded so far: %d kb");

CTestClientHttpSimpleGet* CTestClientHttpSimpleGet::NewL(CHttpClientTestParams& aParams, MLocalTestServerObserver& aObserver, CHttpService& aClient, const TDesC8& aUri, CHTTPTestUtils* aUtils)
    {
    CTestClientHttpSimpleGet* self = new(ELeave) CTestClientHttpSimpleGet(aParams, aObserver, aUtils);
    CleanupStack::PushL(self);
    self->ConstructL(aClient, aUri);
    CleanupStack::Pop(); // self
    return self;
    }  

CTestClientHttpSimpleGet::~CTestClientHttpSimpleGet()
    {
    }

CTestClientHttpSimpleGet::CTestClientHttpSimpleGet(CHttpClientTestParams& aParams, MLocalTestServerObserver& aObserver, CHTTPTestUtils* aUtils)
: iObserver(aObserver),
iParams(aParams),
iTestUtils(*aUtils)
    {
    
    }

void CTestClientHttpSimpleGet::ConstructL(CHttpService& aClient, const TDesC8& aUri)
    {
    TPtrC8 method;
    aClient.String(HTTP::EGET, method);
    CHttpClientTransaction::Create(aClient, method, aUri);   
    }

// From CHTTPClientTransaction
void CTestClientHttpSimpleGet::OnResponseHeaders()
    {
    // Check whether the status code matches
    if(iParams.StatusCode() != StatusCode())
        {
        iObserver.EndTest(KErrCorrupt);
        }
    
    // IF we are receiving a file. just sink the content to the file.
    // We will not check 
    if(iParams.IsFileSending())
        {
        CHttpClientTransaction::SetContentSink(iParams.SinkFile());
        return;
        }
    else
        {
        CHttpClientTransaction::SetContentSink(*this);
        }    
    TInt i = 0;
    const RPointerArray<CHeaderInfo>& hdrInfos = iParams.HeaderInfos();
    
    THttpHeaderIterator iter = ResponseHeaders();
    while(!iter.AtEnd())
        {
        CHeaderInfo* info = iParams.FindHeaderInfo(iter.Name());
        if(info == NULL)
            {
            iObserver.EndTest(KErrCorrupt);
            return;
            }
        
        TInt j = 0;
        THttpHeaderElementIterator elementIter = iter.Elements();
        while(!elementIter.AtEnd())
            {
            CHeaderFieldInfo* fieldInfo = info->iFieldInfos[j];
            THttpHeaderValueVariant value = elementIter.Value();
            if(!iParams.CheckVariantValue(value, *fieldInfo->iFieldValue))
                {
                iObserver.EndTest(KErrCorrupt);
                return;
                }
            
            // Check the params
            THttpHeaderParamIterator paramIter = elementIter.Parameter();
            TInt k = 0;
            while(!paramIter.AtEnd())
                {
                CHeaderFieldParamInfo* paramInfo = fieldInfo->iFieldParamInfos[k];
                // Check the param name and value
                if(paramInfo->iParamName->CompareF(paramIter.Name()) != 0)
                    {
                    iObserver.EndTest(KErrCorrupt);
                    return;
                    }
                THttpHeaderValueVariant value = paramIter.Value();
                if(!iParams.CheckVariantValue(value, *paramInfo->iParamValue) != 0)
                    {
                    iObserver.EndTest(KErrCorrupt);
                    return;
                    }                
                ++k;
                ++paramIter;                
                }
            ++j;
            ++elementIter;
            }
        TInt x = 0;
        THttpHeaderParamIterator hdrParamIter = iter.Parameters();
        while(!hdrParamIter.AtEnd())
            {
            CHeaderFieldParamInfo* paramInfo = info->iParamInfos[x];
            if(paramInfo->iParamName->CompareF(hdrParamIter.Name()) != 0)
                {
                iObserver.EndTest(KErrCorrupt);
                return;
                }
            THttpHeaderValueVariant value = hdrParamIter.Value();
            if(!iParams.CheckVariantValue(value, *paramInfo->iParamValue) != 0)
                {
                iObserver.EndTest(KErrCorrupt);
                return;
                }                
            ++x;
            ++hdrParamIter;
            }
        ++iter;
        ++i;
        }
    }

void CTestClientHttpSimpleGet::OnCompletion()
    {
    if(iParams.IsFileSending())
        {
        iParams.SinkFile().Close(); // Close the file
        TBool ret = ETrue;
        TRAPD(err, ret = iParams.MatchFileContent());
        if(err == KErrNone && ret)
            iObserver.EndTest(KErrNone);
        else
            iObserver.EndTest(KErrCorrupt);
        }
    else
        {
        iObserver.EndTest(KErrNone);
        }
    }

void CTestClientHttpSimpleGet::OnError(TInt aError)
    {
    if(aError == iParams.ExpectedError())
        iObserver.EndTest(KErrNone);
    else
        iObserver.EndTest(aError);    
    }

void CTestClientHttpSimpleGet::OnTransferProgress(TInt aTotal, TInt aTransffered)
    {
    if(iParams.TransferProgress())
        {
        iTestUtils.LogIt(KTransferProgress(),aTotal,aTransffered);
        }    
    }

TBool CTestClientHttpSimpleGet::OnRedirection()
    {
    if(iParams.IsRedirecting())
        return ETrue;
    else
        return EFalse;
    }

void CTestClientHttpSimpleGet::OnData(THttpContentSinkOp& aData)
    {
    if(iParams.ResponseDynamicAndStatic())
        {
        aData.Store();
        }
    else
        {
        TPtrC8 ptr;
        aData.GetData(ptr);
        if(ptr.CompareF(iParams.ResponseBody()) != 0)
            {
            iObserver.EndTest(KErrCorrupt);
            return;
            }
        aData.Release();
        }
    }


CTestClientHttpOnlineGet* CTestClientHttpOnlineGet::NewL(CHttpClientTestParams& aParams, MLocalTestServerObserver& aObserver, CHttpService& aClient, const TDesC8& aUri, CHTTPTestUtils* aUtils)
    {
    CTestClientHttpOnlineGet* self = new(ELeave) CTestClientHttpOnlineGet(aParams, aObserver, aUtils);
     CleanupStack::PushL(self);
     self->ConstructL(aClient, aUri);
     CleanupStack::Pop(); // self
     return self;    
    }

CTestClientHttpOnlineGet::CTestClientHttpOnlineGet(CHttpClientTestParams& aParams, MLocalTestServerObserver& aObserver, CHTTPTestUtils* aUtils)
: CTestClientHttpSimpleGet(aParams, aObserver, aUtils)
    {
    
    }

void CTestClientHttpOnlineGet::ConstructL(CHttpService& aClient, const TDesC8& aUri)
    {
    CTestClientHttpSimpleGet::ConstructL(aClient, aUri);
    _LIT8(KLocalHost, "127.0.0.1");
    if(aUri.FindF(KLocalHost()) == KErrNotFound)
        {
        CHttpClientTransaction::AddCustomRequestHeader(KAcceptHeader(), KAcceptValue());
        THttpHeaderValueVariant variant;    
        variant.SetInt(300);
        CHttpClientTransaction::AddRequestHeader(HTTP::EKeepAlive, variant);
        _LIT8(KEnUs, "en-us");
        variant.SetStr(KEnUs);
        CHttpClientTransaction::AddRequestHeader(HTTP::EAcceptLanguage, variant);
        _LIT8(KEn, "en");
        _LIT8(KQVal, "0.9");
        
        //The following functions SetDateTime() and DateTime() have been used to avoid the code coverage issues.
        TDateTime dateTime(2009, EDecember, 0, 0, 0, 0, 0);
        variant.SetDateTime(dateTime);
        
        TInternetDate internetDate(variant.DateTime());
    
        TLex8 lex(KQVal);
        TReal64 value = 0.;
        lex.Val(value);             
        THTTPHdrVal::TQConv q(value);
        TInt val2 = q;
        THttpHeaderValueVariant variant2(val2);
        variant.SetStr(KEn);
        CHttpClientTransaction::AddRequestHeader(HTTP::EAcceptLanguage, variant, HTTP::EQ, variant2);
        
        //The following two temporary objects have been created to avoid the code coverage issues.
        THttpContentSinkOp httpContentSinkOp;
        THttpContentSourceOp httpContentSourceOp;
        // Disable the pipelining
        DisablePipelining();
        }
    else
        {
        _LIT8(KAcceptAll, "*/*");
        CHttpClientTransaction::AddCustomRequestHeader(KAcceptHeader(), KAcceptAll());        
        }
    }

CTestClientHttpOnlineGet::~CTestClientHttpOnlineGet()
    {
    
    }

void CTestClientHttpOnlineGet::OnResponseHeaders()
    {
    // Check whether the status code matches
     if(iParams.StatusCode() != StatusCode())
         {
         iObserver.EndTest(KErrCorrupt);
         return;
         }
     _LIT8(KStatusText, "Ok");
     const TDesC8& statusText = StatusText();
     
     if(statusText.CompareF(KStatusText) != 0)
         {
         // we do not fail here. But comparison is failed. print that.
         }
     
     // Iterate the response headers
     THttpHeaderIterator iter;
     iter = ResponseHeaders();
     while(!iter.AtEnd())
         {
         iter++;
         }
    
     THttpHeaderElementIterator it;
     it = ResponseHeader(HTTP::ECacheControl);
     while(!it.AtEnd())
         {
         it++;
         }
     _LIT8(KContentType, "Content-Type");
     it = ResponseHeader(KContentType());
     while(!it.AtEnd())
         {
         THttpHeaderParamIterator it2;
         it2 = it.Parameter();
         while(!it2.AtEnd())
             {
             it2++;
             }
         it++;
         }     
     CHttpClientTransaction::SetContentSink(*this);
    }

void CTestClientHttpOnlineGet::OnCompletion()
    {
    iObserver.EndTest(KErrNone);
    }
 
void CTestClientHttpOnlineGet::OnData(THttpContentSinkOp& aData)
    {
    // Just release the data
    aData.Release();
    }