/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the test suite of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <qtest.h>

#include <QDeclarativeEngine>

#include <QDeclarativeComponent>

#include <QDebug>

#include <QNetworkCookieJar>

#include "testhttpserver.h"

#define SERVER_PORT 14445

class tst_qdeclarativexmlhttprequest : public QObject

{

    Q_OBJECT

public:

    tst_qdeclarativexmlhttprequest() {}

private slots:

    void domExceptionCodes();

    void callbackException();

    void callbackException_data();

    void staticStateValues();

    void instanceStateValues();

    void constructor();

    void defaultState();

    void open();

    void open_data();

    void open_invalid_method();

    void open_sync();

    void open_arg_count();

    void setRequestHeader();

    void setRequestHeader_unsent();

    void setRequestHeader_illegalName_data();

    void setRequestHeader_illegalName();

    void setRequestHeader_sent();

    void setRequestHeader_args();

    void send_unsent();

    void send_alreadySent();

    void send_ignoreData();

    void send_withdata();

    void send_withdata_data();

    void abort();

    void abort_unsent();

    void abort_opened();

    void getResponseHeader();

    void getResponseHeader_unsent();

    void getResponseHeader_sent();

    void getResponseHeader_args();

    void getAllResponseHeaders();

    void getAllResponseHeaders_unsent();

    void getAllResponseHeaders_sent();

    void getAllResponseHeaders_args();

    void status();

    void status_data();

    void statusText();

    void statusText_data();

    void responseText();

    void responseText_data();

    void responseXML_invalid();

    void invalidMethodUsage();

    void redirects();

    void nonUtf8();

    // Attributes

    void document();

    void element();

    void attr();

    void text();

    void cdata();

    // Crashes

    // void outstanding_request_at_shutdown();

    // void network_errors()

    // void readyState()

private:

    QDeclarativeEngine engine;

};

inline QUrl TEST_FILE(const QString &filename)

{

    return QUrl::fromLocalFile(QLatin1String(SRCDIR) + QLatin1String("/data/") + filename);

}

// Test that the dom exception codes are correct

void tst_qdeclarativexmlhttprequest::domExceptionCodes()

{

    QDeclarativeComponent component(&engine, TEST_FILE("domExceptionCodes.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("index_size_err").toInt(), 1);

    QCOMPARE(object->property("domstring_size_err").toInt(), 2);

    QCOMPARE(object->property("hierarchy_request_err").toInt(), 3);

    QCOMPARE(object->property("wrong_document_err").toInt(), 4);

    QCOMPARE(object->property("invalid_character_err").toInt(), 5);

    QCOMPARE(object->property("no_data_allowed_err").toInt(), 6);

    QCOMPARE(object->property("no_modification_allowed_err").toInt(), 7);

    QCOMPARE(object->property("not_found_err").toInt(), 8);

    QCOMPARE(object->property("not_supported_err").toInt(), 9);

    QCOMPARE(object->property("inuse_attribute_err").toInt(), 10);

    QCOMPARE(object->property("invalid_state_err").toInt(), 11);

    QCOMPARE(object->property("syntax_err").toInt(), 12);

    QCOMPARE(object->property("invalid_modification_err").toInt(), 13);

    QCOMPARE(object->property("namespace_err").toInt(), 14);

    QCOMPARE(object->property("invalid_access_err").toInt(), 15);

    QCOMPARE(object->property("validation_err").toInt(), 16);

    QCOMPARE(object->property("type_mismatch_err").toInt(), 17);

    delete object;

}

#define TRY_WAIT(expr) \

    do { \

        for (int ii = 0; ii < 6; ++ii) { \

            if ((expr)) break; \

            QTest::qWait(50); \

        } \

        QVERIFY((expr)); \

    } while (false) 

void tst_qdeclarativexmlhttprequest::callbackException_data()

{

    QTest::addColumn<QString>("which");

    QTest::addColumn<int>("line");

    QTest::newRow("on-opened") << "1" << 15;

    QTest::newRow("on-loading") << "3" << 15;

    QTest::newRow("on-done") << "4" << 15;

}

void tst_qdeclarativexmlhttprequest::callbackException()

{

    // Test exception reporting for exceptions thrown at various points.

    QFETCH(QString, which);

    QFETCH(int, line);

    QString expect = TEST_FILE("callbackException.qml").toString() + ":"+QString::number(line)+": Error: Exception from Callback";

    QTest::ignoreMessage(QtWarningMsg, expect.toLatin1());

    QDeclarativeComponent component(&engine, TEST_FILE("callbackException.qml"));

    QObject *object = component.beginCreate(engine.rootContext());

    QVERIFY(object != 0);

    object->setProperty("url", "testdocument.html");

    object->setProperty("which", which);

    component.completeCreate();

    TRY_WAIT(object->property("threw").toBool() == true);

    delete object;

}

// Test that the state value properties on the XMLHttpRequest constructor have the correct values.

// ### WebKit does not do this, but it seems to fit the standard and QML better

void tst_qdeclarativexmlhttprequest::staticStateValues()

{

    QDeclarativeComponent component(&engine, TEST_FILE("staticStateValues.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("unsent").toInt(), 0);

    QCOMPARE(object->property("opened").toInt(), 1);

    QCOMPARE(object->property("headers_received").toInt(), 2);

    QCOMPARE(object->property("loading").toInt(), 3);

    QCOMPARE(object->property("done").toInt(), 4);

    delete object;

}

// Test that the state value properties on instances have the correct values.

void tst_qdeclarativexmlhttprequest::instanceStateValues()

{

    QDeclarativeComponent component(&engine, TEST_FILE("instanceStateValues.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("unsent").toInt(), 0);

    QCOMPARE(object->property("opened").toInt(), 1);

    QCOMPARE(object->property("headers_received").toInt(), 2);

    QCOMPARE(object->property("loading").toInt(), 3);

    QCOMPARE(object->property("done").toInt(), 4);

    delete object;

}

// Test calling constructor 

void tst_qdeclarativexmlhttprequest::constructor()

{

    QDeclarativeComponent component(&engine, TEST_FILE("constructor.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("calledAsConstructor").toBool(), true);

    QCOMPARE(object->property("calledAsFunction").toBool(), true);

    delete object;

}

// Test that all the properties are set correctly before any request is sent

void tst_qdeclarativexmlhttprequest::defaultState()

{

    QDeclarativeComponent component(&engine, TEST_FILE("defaultState.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("readState").toInt(), 0);

    QCOMPARE(object->property("statusIsException").toBool(), true);

    QCOMPARE(object->property("statusTextIsException").toBool(), true);

    QCOMPARE(object->property("responseText").toString(), QString());

    QCOMPARE(object->property("responseXMLIsNull").toBool(), true);

    delete object;

}

// Test valid XMLHttpRequest.open() calls

void tst_qdeclarativexmlhttprequest::open()

{

    QFETCH(QUrl, qmlFile);

    QFETCH(QString, url);

    QFETCH(bool, remote);

    TestHTTPServer *server = 0;

    if (remote) {

        server = new TestHTTPServer(SERVER_PORT);

        QVERIFY(server->isValid());

        QVERIFY(server->wait(TEST_FILE("open_network.expect"), 

                             TEST_FILE("open_network.reply"), 

                             TEST_FILE("testdocument.html")));

    }

    QDeclarativeComponent component(&engine, qmlFile);

    QObject *object = component.beginCreate(engine.rootContext());

    QVERIFY(object != 0);

    object->setProperty("url", url);

    component.completeCreate();

    QCOMPARE(object->property("readyState").toBool(), true);

    QCOMPARE(object->property("openedState").toBool(), true);

    QCOMPARE(object->property("status").toBool(), true);

    QCOMPARE(object->property("statusText").toBool(), true);

    QCOMPARE(object->property("responseText").toBool(), true);

    QCOMPARE(object->property("responseXML").toBool(), true);

    TRY_WAIT(object->property("dataOK").toBool() == true);

    delete server;

    delete object;

}

void tst_qdeclarativexmlhttprequest::open_data()

{

    QTest::addColumn<QUrl>("qmlFile");

    QTest::addColumn<QString>("url");

    QTest::addColumn<bool>("remote");

    QTest::newRow("Relative url)") << TEST_FILE("open.qml") << "testdocument.html" << false;

    QTest::newRow("Absolute url)") << TEST_FILE("open.qml") << TEST_FILE("testdocument.html").toString() << false;

    QTest::newRow("Absolute network url)") << TEST_FILE("open.qml") << "http://127.0.0.1:14445/testdocument.html" << true;

    // ### Check that the username/password were sent to the server

    QTest::newRow("User/pass") << TEST_FILE("open_user.qml") << "http://127.0.0.1:14445/testdocument.html" << true;

}

// Test that calling XMLHttpRequest.open() with an invalid method raises an exception

void tst_qdeclarativexmlhttprequest::open_invalid_method()

{

    QDeclarativeComponent component(&engine, TEST_FILE("open_invalid_method.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("exceptionThrown").toBool(), true);

    delete object;

}

// Test that calling XMLHttpRequest.open() with sync raises an exception

void tst_qdeclarativexmlhttprequest::open_sync()

{

    QDeclarativeComponent component(&engine, TEST_FILE("open_sync.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("exceptionThrown").toBool(), true);

    delete object;

}

// Calling with incorrect arg count raises an exception

void tst_qdeclarativexmlhttprequest::open_arg_count()

{

    {

        QDeclarativeComponent component(&engine, TEST_FILE("open_arg_count.1.qml"));

        QObject *object = component.create();

        QVERIFY(object != 0);

        QCOMPARE(object->property("exceptionThrown").toBool(), true);

        delete object;

    }

    {

        QDeclarativeComponent component(&engine, TEST_FILE("open_arg_count.2.qml"));

        QObject *object = component.create();

        QVERIFY(object != 0);

        QCOMPARE(object->property("exceptionThrown").toBool(), true);

        delete object;

    }

}

// Test valid setRequestHeader() calls

void tst_qdeclarativexmlhttprequest::setRequestHeader()

{

    TestHTTPServer server(SERVER_PORT);

    QVERIFY(server.isValid());

    QVERIFY(server.wait(TEST_FILE("setRequestHeader.expect"), 

                        TEST_FILE("setRequestHeader.reply"), 

                        TEST_FILE("testdocument.html")));

    QDeclarativeComponent component(&engine, TEST_FILE("setRequestHeader.qml"));

    QObject *object = component.beginCreate(engine.rootContext());

    QVERIFY(object != 0);

    object->setProperty("url", "http://127.0.0.1:14445/testdocument.html");

    component.completeCreate();

    TRY_WAIT(object->property("dataOK").toBool() == true);

    delete object;

}

// Test setting headers before open() throws exception

void tst_qdeclarativexmlhttprequest::setRequestHeader_unsent()

{

    QDeclarativeComponent component(&engine, TEST_FILE("setRequestHeader_unsent.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("test").toBool(), true);

    delete object;

}

void tst_qdeclarativexmlhttprequest::setRequestHeader_illegalName_data()

{

    QTest::addColumn<QString>("name");

    QTest::newRow("Accept-Charset") << "AccePT-CHArset";

    QTest::newRow("Accept-Encoding") << "AccEpt-EnCOding";

    QTest::newRow("Connection") << "ConnECtion";

    QTest::newRow("Content-Length") << "ContEnt-LenGth";

    QTest::newRow("Cookie") << "CookIe";

    QTest::newRow("Cookie2") << "CoOkie2";

    QTest::newRow("Content-Transfer-Encoding") << "ConteNT-tRANSFER-eNCOding";

    QTest::newRow("Date") << "DaTE";

    QTest::newRow("Expect") << "ExPect";

    QTest::newRow("Host") << "HoST";

    QTest::newRow("Keep-Alive") << "KEEP-aLive";

    QTest::newRow("Referer") << "ReferEr";

    QTest::newRow("TE") << "Te";

    QTest::newRow("Trailer") << "TraILEr";

    QTest::newRow("Transfer-Encoding") << "tRANsfer-Encoding";

    QTest::newRow("Upgrade") << "UpgrADe";

    QTest::newRow("User-Agent") << "uSEr-Agent";

    QTest::newRow("Via") << "vIa";

    QTest::newRow("Proxy-") << "ProXy-";

    QTest::newRow("Sec-") << "SeC-";

    QTest::newRow("Proxy-*") << "Proxy-BLAH";

    QTest::newRow("Sec-*") << "Sec-F";

}

// Tests that using illegal header names has no effect

void tst_qdeclarativexmlhttprequest::setRequestHeader_illegalName()

{

    QFETCH(QString, name);

    TestHTTPServer server(SERVER_PORT);

    QVERIFY(server.isValid());

    QVERIFY(server.wait(TEST_FILE("open_network.expect"), 

                        TEST_FILE("open_network.reply"), 

                        TEST_FILE("testdocument.html")));

    QDeclarativeComponent component(&engine, TEST_FILE("setRequestHeader_illegalName.qml"));

    QObject *object = component.beginCreate(engine.rootContext());

    QVERIFY(object != 0);

    object->setProperty("url", "http://127.0.0.1:14445/testdocument.html");

    object->setProperty("header", name);

    component.completeCreate();

    QCOMPARE(object->property("readyState").toBool(), true);

    QCOMPARE(object->property("openedState").toBool(), true);

    QCOMPARE(object->property("status").toBool(), true);

    QCOMPARE(object->property("statusText").toBool(), true);

    QCOMPARE(object->property("responseText").toBool(), true);

    QCOMPARE(object->property("responseXML").toBool(), true);

    TRY_WAIT(object->property("dataOK").toBool() == true);

    delete object;

}

// Test that attempting to set a header after a request is sent throws an exception

void tst_qdeclarativexmlhttprequest::setRequestHeader_sent()

{

    TestHTTPServer server(SERVER_PORT);

    QVERIFY(server.isValid());

    QVERIFY(server.wait(TEST_FILE("open_network.expect"), 

                        TEST_FILE("open_network.reply"), 

                        TEST_FILE("testdocument.html")));

    QDeclarativeComponent component(&engine, TEST_FILE("setRequestHeader_sent.qml"));

    QObject *object = component.beginCreate(engine.rootContext());

    QVERIFY(object != 0);

    object->setProperty("url", "http://127.0.0.1:14445/testdocument.html");

    component.completeCreate();

    QCOMPARE(object->property("test").toBool(), true);

    TRY_WAIT(object->property("dataOK").toBool() == true);

    delete object;

}

// Invalid arg count throws exception

void tst_qdeclarativexmlhttprequest::setRequestHeader_args()

{

    QDeclarativeComponent component(&engine, TEST_FILE("setRequestHeader_args.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("exceptionThrown").toBool(), true);

    delete object;

}

// Test that calling send() in UNSENT state throws an exception

void tst_qdeclarativexmlhttprequest::send_unsent()

{

    QDeclarativeComponent component(&engine, TEST_FILE("send_unsent.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("test").toBool(), true);

    delete object;

}

// Test attempting to resend a sent request throws an exception

void tst_qdeclarativexmlhttprequest::send_alreadySent()

{

    QDeclarativeComponent component(&engine, TEST_FILE("send_alreadySent.qml"));

    QObject *object = component.create();

    QVERIFY(object != 0);

    QCOMPARE(object->property("test").toBool(), true);

    TRY_WAIT(object->property("dataOK").toBool() == true);

    delete object;

}

// Test that send for a GET or HEAD ignores data

void tst_qdeclarativexmlhttprequest::send_ignoreData()

{

    {

        TestHTTPServer server(SERVER_PORT);

        QVERIFY(server.isValid());

        QVERIFY(server.wait(TEST_FILE("send_ignoreData_GET.expect"), 

                            TEST_FILE("send_ignoreData.reply"), 

                            TEST_FILE("testdocument.html")));

        QDeclarativeComponent component(&engine, TEST_FILE("send_ignoreData.qml"));

        QObject *object = component.beginCreate(engine.rootContext());

        QVERIFY(object != 0);

        object->setProperty("reqType", "GET");

        object->setProperty("url", "http://127.0.0.1:14445/testdocument.html");

        component.completeCreate();

        TRY_WAIT(object->property("dataOK").toBool() == true);

        delete object;

    }

    {

        TestHTTPServer server(SERVER_PORT);

        QVERIFY(server.isValid());

        QVERIFY(server.wait(TEST_FILE("send_ignoreData_PUT.expect"),