#include "../yourlsclient.h"

#include <QJsonDocument>
#include <QJsonObject>
#include <QSignalSpy>
#include <QTcpServer>
#include <QTcpSocket>
#include <QTest>
#include <QUrlQuery>

class YourlsClientTest : public QObject {
  Q_OBJECT
private slots:
  void responses_data() {
    QTest::addColumn<int> ("httpStatus");
    QTest::addColumn<QString> ("status");
    QTest::addColumn<QString> ("code");
    QTest::addColumn<QString> ("originalUrl");
    QTest::addColumn<QString> ("shortUrl");
    QTest::addColumn<bool> ("success");
    const QString source = QStringLiteral ("https://git.himro.ca/explore/repos?q=a%26b&next=x+y#section");
    const QString shortened = QStringLiteral ("https://cut.himro.ca/h");
    QTest::newRow ("created") << 200 << "success" << "" << source << shortened << true;
    QTest::newRow ("duplicate") << 400 << "fail" << "error:url" << source << shortened << true;
    QTest::newRow ("different-url") << 400 << "fail" << "error:url" << "https://example.com" << shortened << false;
    QTest::newRow ("missing-short-url") << 400 << "fail" << "error:url" << source << "" << false;
    QTest::newRow ("relative-short-url") << 400 << "fail" << "error:url" << source << "/h" << false;
    QTest::newRow ("other-error") << 400 << "fail" << "error:auth" << source << shortened << false;
    QTest::newRow ("server-error") << 500 << "fail" << "error:url" << source << shortened << false;
  }

  void responses() {
    QFETCH (int, httpStatus);
    QFETCH (QString, status);
    QFETCH (QString, code);
    QFETCH (QString, originalUrl);
    QFETCH (QString, shortUrl);
    QFETCH (bool, success);
    QTcpServer server;
    QVERIFY (server.listen (QHostAddress::LocalHost));
    QByteArray requestData;
    const QByteArray body = QJsonDocument (QJsonObject {
      {"status", status}, {"code", code}, {"shorturl", shortUrl},
      {"url", QJsonObject {{"url", originalUrl}}}
    }).toJson (QJsonDocument::Compact);
    connect (&server, &QTcpServer::newConnection, this, [&]() {
      auto *socket = server.nextPendingConnection();
      connect (socket, &QTcpSocket::readyRead, &server, [&, socket]() {
        requestData += socket->readAll();
        if (!requestData.contains ("\r\n\r\n"))
          return;
        socket->write ("HTTP/1.1 " + QByteArray::number (httpStatus)
                       + " Result\r\nContent-Type: application/json\r\nContent-Length: "
                       + QByteArray::number (body.size()) + "\r\nConnection: close\r\n\r\n" + body);
        socket->disconnectFromHost();
      });
    });
    YourlsClient client;
    QSignalSpy succeeded (&client, &YourlsClient::shortenSucceeded);
    QSignalSpy failed (&client, &YourlsClient::shortenFailed);
    const QUrl source (QStringLiteral ("https://git.himro.ca/explore/repos?q=a%26b&next=x+y#section"));
    client.shortenUrl ({QStringLiteral ("http://127.0.0.1:%1/api").arg (server.serverPort()),
                       QStringLiteral ("test-signature")}, source);
    QTRY_COMPARE (succeeded.count() + failed.count(), 1);
    QCOMPARE (succeeded.count(), success ? 1 : 0);
    if (success)
      QCOMPARE (succeeded.first().first().toUrl(), QUrl (shortUrl));
    const QByteArray target = requestData.split (' ').value (1);
    const QUrlQuery query (QUrl::fromEncoded (target));
    // Decode the HTTP query once, as the server does, preserving URL escapes.
    const QString encodedValue = query.queryItemValue (QStringLiteral ("url"), QUrl::FullyEncoded);
    QCOMPARE (QUrl::fromPercentEncoding (encodedValue.toLatin1()), source.toString (QUrl::FullyEncoded));
  }
};

QTEST_GUILESS_MAIN (YourlsClientTest)
#include "yourlsclient_test.moc"