/*

    Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies)

    Copyright (C) 2009 Girish Ramakrishnan <girish@forwardbias.in>

    Copyright (C) 2010 Holger Hans Peter Freyther

    This library is free software; you can redistribute it and/or

    modify it under the terms of the GNU Library General Public

    License as published by the Free Software Foundation; either

    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License

    along with this library; see the file COPYING.LIB.  If not, write to

    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

    Boston, MA 02110-1301, USA.

*/

#include "../util.h"

#include "../WebCoreSupport/DumpRenderTreeSupportQt.h"

#include <QDir>

#include <QGraphicsWidget>

#include <QLineEdit>

#include <QLocale>

#include <QMenu>

#include <QPushButton>

#include <QStyle>

#include <QtTest/QtTest>

#include <QTextCharFormat>

#include <qgraphicsscene.h>

#include <qgraphicsview.h>

#include <qgraphicswebview.h>

#include <qnetworkrequest.h>

#include <qwebdatabase.h>

#include <qwebelement.h>

#include <qwebframe.h>

#include <qwebhistory.h>

#include <qwebpage.h>

#include <qwebsecurityorigin.h>

#include <qwebview.h>

class EventSpy : public QObject, public QList<QEvent::Type>

{

    Q_OBJECT

public:

    EventSpy(QObject* objectToSpy)

    {

        objectToSpy->installEventFilter(this);

    }

    virtual bool eventFilter(QObject* receiver, QEvent* event)

    {

        append(event->type());

        return false;

    }

};

class tst_QWebPage : public QObject

{

    Q_OBJECT

public:

    tst_QWebPage();

    virtual ~tst_QWebPage();

public slots:

    void init();

    void cleanup();

    void cleanupFiles();

private slots:

    void initTestCase();

    void cleanupTestCase();

    void acceptNavigationRequest();

    void infiniteLoopJS();

    void geolocationRequestJS();

    void loadFinished();

    void acceptNavigationRequestWithNewWindow();

    void userStyleSheet();

    void modified();

    void contextMenuCrash();

    void database();

    void createPluginWithPluginsEnabled();

    void createPluginWithPluginsDisabled();

    void destroyPlugin_data();

    void destroyPlugin();

    void createViewlessPlugin_data();

    void createViewlessPlugin();

    void multiplePageGroupsAndLocalStorage();

    void cursorMovements();

    void textSelection();

    void textEditing();

    void backActionUpdate();

    void frameAt();

    void requestCache();

    void protectBindingsRuntimeObjectsFromCollector();

    void localURLSchemes();

    void testOptionalJSObjects();

    void testEnablePersistentStorage();

    void consoleOutput();

    void inputMethods_data();

    void inputMethods();

    void inputMethodsTextFormat_data();

    void inputMethodsTextFormat();

    void defaultTextEncoding();

    void errorPageExtension();

    void errorPageExtensionInIFrames();

    void errorPageExtensionInFrameset();

    void userAgentApplicationName();

    void userAgentLocaleChange();

    void viewModes();

    void crashTests_LazyInitializationOfMainFrame();

    void screenshot_data();

    void screenshot();

    void originatingObjectInNetworkRequests();

    void testJSPrompt();

    void showModalDialog();

    void testStopScheduledPageRefresh();

    void findText();

private:

    QWebView* m_view;

    QWebPage* m_page;

};

tst_QWebPage::tst_QWebPage()

{

}

tst_QWebPage::~tst_QWebPage()

{

}

void tst_QWebPage::init()

{

    m_view = new QWebView();

    m_page = m_view->page();

}

void tst_QWebPage::cleanup()

{

    delete m_view;

}

void tst_QWebPage::cleanupFiles()

{

    QFile::remove("Databases.db");

    QDir::current().rmdir("http_www.myexample.com_0");

    QFile::remove("http_www.myexample.com_0.localstorage");

}

void tst_QWebPage::initTestCase()

{

    cleanupFiles(); // In case there are old files from previous runs

}

void tst_QWebPage::cleanupTestCase()

{

    cleanupFiles(); // Be nice

}

class NavigationRequestOverride : public QWebPage

{

public:

    NavigationRequestOverride(QWebView* parent, bool initialValue) : QWebPage(parent), m_acceptNavigationRequest(initialValue) {}

    bool m_acceptNavigationRequest;

protected:

    virtual bool acceptNavigationRequest(QWebFrame* frame, const QNetworkRequest &request, QWebPage::NavigationType type) {

        Q_UNUSED(frame);

        Q_UNUSED(request);

        Q_UNUSED(type);

        return m_acceptNavigationRequest;

    }

};

void tst_QWebPage::acceptNavigationRequest()

{

    QSignalSpy loadSpy(m_view, SIGNAL(loadFinished(bool)));

    NavigationRequestOverride* newPage = new NavigationRequestOverride(m_view, false);

    m_view->setPage(newPage);

    m_view->setHtml(QString("<html><body><form name='tstform' action='data:text/html,foo'method='get'>"

                            "<input type='text'><input type='submit'></form></body></html>"), QUrl());

    QTRY_COMPARE(loadSpy.count(), 1);

    m_view->page()->mainFrame()->evaluateJavaScript("tstform.submit();");

    newPage->m_acceptNavigationRequest = true;

    m_view->page()->mainFrame()->evaluateJavaScript("tstform.submit();");

    QTRY_COMPARE(loadSpy.count(), 2);

    QCOMPARE(m_view->page()->mainFrame()->toPlainText(), QString("foo?"));

    // Restore default page

    m_view->setPage(0);

}

class JSTestPage : public QWebPage

{

Q_OBJECT

public:

    JSTestPage(QObject* parent = 0)

    : QWebPage(parent) {}

public slots:

    bool shouldInterruptJavaScript() {

        return true;

    }

    bool allowGeolocationRequest(QWebFrame *frame) 

    {

        return m_allowGeolocation;

    }

public:

    void setGeolocationPermission(bool allow) 

    {

        m_allowGeolocation = allow;

    }

private: 

    bool m_allowGeolocation;

};

void tst_QWebPage::infiniteLoopJS()

{

#ifdef Q_WS_MAEMO_5

    QSKIP("Test never terminates on Maemo 5 : https://bugs.webkit.org/show_bug.cgi?id=38538", SkipAll);

#endif

    JSTestPage* newPage = new JSTestPage(m_view);

    m_view->setPage(newPage);

    m_view->setHtml(QString("<html><body>test</body></html>"), QUrl());

    m_view->page()->mainFrame()->evaluateJavaScript("var run = true;var a = 1;while(run){a++;}");

    delete newPage;

}

void tst_QWebPage::geolocationRequestJS()

{

    JSTestPage* newPage = new JSTestPage(m_view);

    newPage->setGeolocationPermission(false);

    m_view->setPage(newPage);

    m_view->setHtml(QString("<html><body>test</body></html>"), QUrl());

    m_view->page()->mainFrame()->evaluateJavaScript("var errorCode = 0; function error(err) { errorCode = err.code; } function success(pos) { } navigator.geolocation.getCurrentPosition(success, error)");

    QTest::qWait(2000);

    QVariant empty = m_view->page()->mainFrame()->evaluateJavaScript("errorCode");

    QVERIFY(empty.type() == QVariant::Double && empty.toInt() != 0);

    newPage->setGeolocationPermission(true);

    m_view->page()->mainFrame()->evaluateJavaScript("errorCode = 0; navigator.geolocation.getCurrentPosition(success, error);");

    empty = m_view->page()->mainFrame()->evaluateJavaScript("errorCode");

    //http://dev.w3.org/geo/api/spec-source.html#position

    //PositionError: const unsigned short PERMISSION_DENIED = 1;

    QVERIFY(empty.type() == QVariant::Double && empty.toInt() != 1);

    delete newPage;

}

void tst_QWebPage::loadFinished()

{

    qRegisterMetaType<QWebFrame*>("QWebFrame*");

    qRegisterMetaType<QNetworkRequest*>("QNetworkRequest*");

    QSignalSpy spyLoadStarted(m_view, SIGNAL(loadStarted()));

    QSignalSpy spyLoadFinished(m_view, SIGNAL(loadFinished(bool)));

    m_view->setHtml(QString("data:text/html,<frameset cols=\"25%,75%\"><frame src=\"data:text/html,"

                            "<head><meta http-equiv='refresh' content='1'></head>foo \">"

                            "<frame src=\"data:text/html,bar\"></frameset>"), QUrl());

    QTRY_COMPARE(spyLoadFinished.count(), 1);

    QTRY_VERIFY(spyLoadStarted.count() > 1);

    QTRY_VERIFY(spyLoadFinished.count() > 1);

    spyLoadFinished.clear();

    m_view->setHtml(QString("data:text/html,<frameset cols=\"25%,75%\"><frame src=\"data:text/html,"

                            "foo \"><frame src=\"data:text/html,bar\"></frameset>"), QUrl());

    QTRY_COMPARE(spyLoadFinished.count(), 1);

    QCOMPARE(spyLoadFinished.count(), 1);

}

class ConsolePage : public QWebPage

{

public:

    ConsolePage(QObject* parent = 0) : QWebPage(parent) {}

    virtual void javaScriptConsoleMessage(const QString& message, int lineNumber, const QString& sourceID)

    {

        messages.append(message);

        lineNumbers.append(lineNumber);

        sourceIDs.append(sourceID);

    }

    QStringList messages;

    QList<int> lineNumbers;

    QStringList sourceIDs;

};

void tst_QWebPage::consoleOutput()

{

    ConsolePage page;

    page.mainFrame()->evaluateJavaScript("this is not valid JavaScript");

    QCOMPARE(page.messages.count(), 1);

    QCOMPARE(page.lineNumbers.at(0), 1);

}

class TestPage : public QWebPage

{

public:

    TestPage(QObject* parent = 0) : QWebPage(parent) {}

    struct Navigation {

        QPointer<QWebFrame> frame;

        QNetworkRequest request;

        NavigationType type;

    };

    QList<Navigation> navigations;

    QList<QWebPage*> createdWindows;

    virtual bool acceptNavigationRequest(QWebFrame* frame, const QNetworkRequest &request, NavigationType type) {

        Navigation n;

        n.frame = frame;

        n.request = request;

        n.type = type;

        navigations.append(n);

        return true;

    }

    virtual QWebPage* createWindow(WebWindowType) {

        QWebPage* page = new TestPage(this);

        createdWindows.append(page);

        return page;

    }

};

void tst_QWebPage::acceptNavigationRequestWithNewWindow()

{

    TestPage* page = new TestPage(m_view);

    page->settings()->setAttribute(QWebSettings::LinksIncludedInFocusChain, true);

    m_page = page;

    m_view->setPage(m_page);

    m_view->setUrl(QString("data:text/html,<a href=\"data:text/html,Reached\" target=\"_blank\">Click me</a>"));

    QVERIFY(::waitForSignal(m_view, SIGNAL(loadFinished(bool))));

    QFocusEvent fe(QEvent::FocusIn);

    m_page->event(&fe);

    QVERIFY(m_page->focusNextPrevChild(/*next*/ true));

    QKeyEvent keyEnter(QEvent::KeyPress, Qt::Key_Enter, Qt::NoModifier);

    m_page->event(&keyEnter);

    QCOMPARE(page->navigations.count(), 2);

    TestPage::Navigation n = page->navigations.at(1);

    QVERIFY(n.frame.isNull());

    QCOMPARE(n.request.url().toString(), QString("data:text/html,Reached"));

    QVERIFY(n.type == QWebPage::NavigationTypeLinkClicked);

    QCOMPARE(page->createdWindows.count(), 1);

}

class TestNetworkManager : public QNetworkAccessManager

{

public:

    TestNetworkManager(QObject* parent) : QNetworkAccessManager(parent) {}

    QList<QUrl> requestedUrls;

    QList<QNetworkRequest> requests;

protected:

    virtual QNetworkReply* createRequest(Operation op, const QNetworkRequest &request, QIODevice* outgoingData) {

        requests.append(request);

        requestedUrls.append(request.url());

        return QNetworkAccessManager::createRequest(op, request, outgoingData);

    }

};

void tst_QWebPage::userStyleSheet()

{

    TestNetworkManager* networkManager = new TestNetworkManager(m_page);

    m_page->setNetworkAccessManager(networkManager);

    networkManager->requestedUrls.clear();

    m_page->settings()->setUserStyleSheetUrl(QUrl("data:text/css;charset=utf-8;base64,"

            + QByteArray("p { background-image: url('http://does.not/exist.png');}").toBase64()));

    m_view->setHtml("<p>hello world</p>");

    QVERIFY(::waitForSignal(m_view, SIGNAL(loadFinished(bool))));

    QVERIFY(networkManager->requestedUrls.count() >= 1);

    QCOMPARE(networkManager->requestedUrls.at(0), QUrl("http://does.not/exist.png"));

}

void tst_QWebPage::viewModes()

{

    m_view->setHtml("<body></body>");

    m_page->setProperty("_q_viewMode", "minimized");

    QVariant empty = m_page->mainFrame()->evaluateJavaScript("window.styleMedia.matchMedium(\"(-webkit-view-mode)\")");

    QVERIFY(empty.type() == QVariant::Bool && empty.toBool());

    QVariant minimized = m_page->mainFrame()->evaluateJavaScript("window.styleMedia.matchMedium(\"(-webkit-view-mode: minimized)\")");

    QVERIFY(minimized.type() == QVariant::Bool && minimized.toBool());

    QVariant maximized = m_page->mainFrame()->evaluateJavaScript("window.styleMedia.matchMedium(\"(-webkit-view-mode: maximized)\")");

    QVERIFY(maximized.type() == QVariant::Bool && !maximized.toBool());

}

void tst_QWebPage::modified()

{

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body>blub"));

    QVERIFY(::waitForSignal(m_view, SIGNAL(loadFinished(bool))));

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body id=foo contenteditable>blah"));

    QVERIFY(::waitForSignal(m_view, SIGNAL(loadFinished(bool))));

    QVERIFY(!m_page->isModified());

//    m_page->mainFrame()->evaluateJavaScript("alert(document.getElementById('foo'))");

    m_page->mainFrame()->evaluateJavaScript("document.getElementById('foo').focus()");

    m_page->mainFrame()->evaluateJavaScript("document.execCommand('InsertText', true, 'Test');");

    QVERIFY(m_page->isModified());

    m_page->mainFrame()->evaluateJavaScript("document.execCommand('Undo', true);");

    QVERIFY(!m_page->isModified());

    m_page->mainFrame()->evaluateJavaScript("document.execCommand('Redo', true);");

    QVERIFY(m_page->isModified());

    QVERIFY(m_page->history()->canGoBack());

    QVERIFY(!m_page->history()->canGoForward());

    QCOMPARE(m_page->history()->count(), 2);

    QVERIFY(m_page->history()->backItem().isValid());

    QVERIFY(!m_page->history()->forwardItem().isValid());

    m_page->history()->back();

    QVERIFY(::waitForSignal(m_view, SIGNAL(loadFinished(bool))));

    QVERIFY(!m_page->history()->canGoBack());

    QVERIFY(m_page->history()->canGoForward());

    QVERIFY(!m_page->isModified());

    QVERIFY(m_page->history()->currentItemIndex() == 0);

    m_page->history()->setMaximumItemCount(3);

    QVERIFY(m_page->history()->maximumItemCount() == 3);

    QVariant variant("string test");

    m_page->history()->currentItem().setUserData(variant);

    QVERIFY(m_page->history()->currentItem().userData().toString() == "string test");

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body>This is second page"));

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body>This is third page"));

    QVERIFY(m_page->history()->count() == 2);

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body>This is fourth page"));

    QVERIFY(m_page->history()->count() == 2);

    m_page->mainFrame()->setUrl(QUrl("data:text/html,<body>This is fifth page"));

    QVERIFY(::waitForSignal(m_page, SIGNAL(saveFrameStateRequested(QWebFrame*,QWebHistoryItem*))));

}

void tst_QWebPage::contextMenuCrash()

{

    QWebView view;

    view.setHtml("<p>test");

    view.page()->updatePositionDependentActions(QPoint(0, 0));

    QMenu* contextMenu = 0;

    foreach (QObject* child, view.children()) {

        contextMenu = qobject_cast<QMenu*>(child);

        if (contextMenu)

            break;

    }

    QVERIFY(contextMenu);

    delete contextMenu;

}

void tst_QWebPage::database()

{

    QString path = QDir::currentPath();

    m_page->settings()->setOfflineStoragePath(path);

    QVERIFY(m_page->settings()->offlineStoragePath() == path);

    QWebSettings::setOfflineStorageDefaultQuota(1024 * 1024);

    QVERIFY(QWebSettings::offlineStorageDefaultQuota() == 1024 * 1024);

    m_page->settings()->setAttribute(QWebSettings::LocalStorageEnabled, true);

    m_page->settings()->setAttribute(QWebSettings::OfflineStorageDatabaseEnabled, true);

    QString dbFileName = path + "Databases.db";

    if (QFile::exists(dbFileName))

        QFile::remove(dbFileName);

    qRegisterMetaType<QWebFrame*>("QWebFrame*");

    QSignalSpy spy(m_page, SIGNAL(databaseQuotaExceeded(QWebFrame*,QString)));

    m_view->setHtml(QString("<html><head><script>var db; db=openDatabase('testdb', '1.0', 'test database API', 50000); </script></head><body><div></div></body></html>"), QUrl("http://www.myexample.com"));

    QTRY_COMPARE(spy.count(), 1);

    m_page->mainFrame()->evaluateJavaScript("var db2; db2=openDatabase('testdb', '1.0', 'test database API', 50000);");

    QTRY_COMPARE(spy.count(),1);

    m_page->mainFrame()->evaluateJavaScript("localStorage.test='This is a test for local storage';");

    m_view->setHtml(QString("<html><body id='b'>text</body></html>"), QUrl("http://www.myexample.com"));

    QVariant s1 = m_page->mainFrame()->evaluateJavaScript("localStorage.test");

    QCOMPARE(s1.toString(), QString("This is a test for local storage"));

    m_page->mainFrame()->evaluateJavaScript("sessionStorage.test='This is a test for session storage';");

    m_view->setHtml(QString("<html><body id='b'>text</body></html>"), QUrl("http://www.myexample.com"));

    QVariant s2 = m_page->mainFrame()->evaluateJavaScript("sessionStorage.test");

    QCOMPARE(s2.toString(), QString("This is a test for session storage"));

    m_view->setHtml(QString("<html><head></head><body><div></div></body></html>"), QUrl("http://www.myexample.com"));

    m_page->mainFrame()->evaluateJavaScript("var db3; db3=openDatabase('testdb', '1.0', 'test database API', 50000);db3.transaction(function(tx) { tx.executeSql('CREATE TABLE IF NOT EXISTS Test (text TEXT)', []); }, function(tx, result) { }, function(tx, error) { });");