#include "rpc/Errors.hpp"
#include "util/NameGenerator.hpp"
#include "util/Taggable.hpp"
#include "util/config/ConfigDefinition.hpp"
#include "util/config/ConfigValue.hpp"
#include "util/config/Types.hpp"
#include "web/impl/ErrorHandling.hpp"
#include "web/interface/ConnectionBaseMock.hpp"

#include <boost/beast/http/status.hpp>
#include <boost/json/object.hpp>
#include <boost/json/serialize.hpp>
#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include <memory>
#include <optional>
#include <string>

using namespace web::impl;
using namespace web;
using namespace util::config;

struct ErrorHandlingTests : public virtual ::testing::Test {
protected:
    util::TagDecoratorFactory tagFactory_{ClioConfigDefinition{
        {"log.tag_style", ConfigValue{ConfigType::String}.defaultValue("uint")},
    }};
    std::string const clientIp_ = "some ip";
    ConnectionBaseStrictMockPtr connection_ =
        std::make_shared<testing::StrictMock<ConnectionBaseMock>>(tagFactory_, clientIp_);
};

struct ErrorHandlingComposeErrorTestBundle {
    std::string testName;
    bool connectionUpgraded;
    std::optional<boost::json::object> request;
    boost::json::object expectedResult;
};

struct ErrorHandlingComposeErrorTest
    : ErrorHandlingTests,
      testing::WithParamInterface<ErrorHandlingComposeErrorTestBundle> {};

TEST_P(ErrorHandlingComposeErrorTest, composeError)
{
    connection_->upgraded = GetParam().connectionUpgraded;
    ErrorHelper const errorHelper{connection_, GetParam().request};
    auto const result = errorHelper.composeError(rpc::RippledError::rpcNOT_READY);
    EXPECT_EQ(boost::json::serialize(result), boost::json::serialize(GetParam().expectedResult));
}

INSTANTIATE_TEST_CASE_P(
    ErrorHandlingComposeErrorTestGroup,
    ErrorHandlingComposeErrorTest,
    testing::ValuesIn(
        {ErrorHandlingComposeErrorTestBundle{
             "NoRequest_UpgradedConnection",
             true,
             std::nullopt,
             {{"error", "notReady"},
              {"error_code", 13},
              {"error_message", "Not ready to handle this request."},
              {"status", "error"},
              {"type", "response"}}
         },
         ErrorHandlingComposeErrorTestBundle{
             "NoRequest_NotUpgradedConnection",
             false,
             std::nullopt,
             {{"result",
               {{"error", "notReady"},
                {"error_code", 13},
                {"error_message", "Not ready to handle this request."},
                {"status", "error"},
                {"type", "response"}}}}
         },
         ErrorHandlingComposeErrorTestBundle{
             "Request_UpgradedConnection",
             true,
             boost::json::object{{"id", 1}, {"api_version", 2}},
             {{"error", "notReady"},
              {"error_code", 13},
              {"error_message", "Not ready to handle this request."},
              {"status", "error"},
              {"type", "response"},
              {"id", 1},
              {"api_version", 2},
              {"request", {{"id", 1}, {"api_version", 2}}}}
         },
         ErrorHandlingComposeErrorTestBundle{
             "Request_NotUpgradedConnection",
             false,
             boost::json::object{{"id", 1}, {"api_version", 2}},
             {{"result",
               {{"error", "notReady"},
                {"error_code", 13},
                {"error_message", "Not ready to handle this request."},
                {"status", "error"},
                {"type", "response"},
                {"id", 1},
                {"request", {{"id", 1}, {"api_version", 2}}}}}}
         }}
    ),
    tests::util::kNAME_GENERATOR
);

struct ErrorHandlingSendErrorTestBundle {
    std::string testName;
    bool connectionUpgraded;
    rpc::Status status;
    std::string expectedMessage;
    boost::beast::http::status expectedStatus;
};

struct ErrorHandlingSendErrorTest : ErrorHandlingTests,
                                    testing::WithParamInterface<ErrorHandlingSendErrorTestBundle> {
};

TEST_P(ErrorHandlingSendErrorTest, sendError)
{
    connection_->upgraded = GetParam().connectionUpgraded;
    ErrorHelper const errorHelper{connection_};

    EXPECT_CALL(
        *connection_, send(std::string{GetParam().expectedMessage}, GetParam().expectedStatus)
    );
    errorHelper.sendError(GetParam().status);
}

INSTANTIATE_TEST_CASE_P(
    ErrorHandlingSendErrorTestGroup,
    ErrorHandlingSendErrorTest,
    testing::ValuesIn({
        ErrorHandlingSendErrorTestBundle{
            "UpgradedConnection",
            true,
            rpc::Status{rpc::RippledError::rpcTOO_BUSY},
            R"JSON({"error":"tooBusy","error_code":9,"error_message":"The server is too busy to help you now.","status":"error","type":"response"})JSON",
            boost::beast::http::status::ok
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_InvalidApiVersion",
            false,
            rpc::Status{rpc::ClioError::RpcInvalidApiVersion},
            "invalid_API_version",
            boost::beast::http::status::bad_request
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_CommandIsMissing",
            false,
            rpc::Status{rpc::ClioError::RpcCommandIsMissing},
            "Null method",
            boost::beast::http::status::bad_request
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_CommandIsEmpty",
            false,
            rpc::Status{rpc::ClioError::RpcCommandIsEmpty},
            "method is empty",
            boost::beast::http::status::bad_request
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_CommandNotString",
            false,
            rpc::Status{rpc::ClioError::RpcCommandNotString},
            "method is not string",
            boost::beast::http::status::bad_request
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_ParamsUnparsable",
            false,
            rpc::Status{rpc::ClioError::RpcParamsUnparsable},
            "params unparsable",
            boost::beast::http::status::bad_request
        },
        ErrorHandlingSendErrorTestBundle{
            "NotUpgradedConnection_RippledError",
            false,
            rpc::Status{rpc::RippledError::rpcTOO_BUSY},
            R"JSON({"result":{"error":"tooBusy","error_code":9,"error_message":"The server is too busy to help you now.","status":"error","type":"response"}})JSON",
            boost::beast::http::status::bad_request
        },
    }),
    tests::util::kNAME_GENERATOR
);

TEST_F(ErrorHandlingTests, sendInternalError)
{
    ErrorHelper const errorHelper{connection_};

    EXPECT_CALL(
        *connection_,
        send(
            std::string{
                R"JSON({"result":{"error":"internal","error_code":73,"error_message":"Internal error.","status":"error","type":"response"}})JSON"
            },
            boost::beast::http::status::internal_server_error
        )
    );
    errorHelper.sendInternalError();
}

TEST_F(ErrorHandlingTests, sendNotReadyError)
{
    ErrorHelper const errorHelper{connection_};
    EXPECT_CALL(
        *connection_,
        send(
            std::string{
                R"JSON({"result":{"error":"notReady","error_code":13,"error_message":"Not ready to handle this request.","status":"error","type":"response"}})JSON"
            },
            boost::beast::http::status::ok
        )
    );
    errorHelper.sendNotReadyError();
}

TEST_F(ErrorHandlingTests, sendTooBusyError_UpgradedConnection)
{
    connection_->upgraded = true;
    ErrorHelper const errorHelper{connection_};
    EXPECT_CALL(
        *connection_,
        send(
            std::string{
                R"JSON({"error":"tooBusy","error_code":9,"error_message":"The server is too busy to help you now.","status":"error","type":"response"})JSON"
            },
            boost::beast::http::status::ok
        )
    );
    errorHelper.sendTooBusyError();
}

TEST_F(ErrorHandlingTests, sendTooBusyError_NotUpgradedConnection)
{
    connection_->upgraded = false;
    ErrorHelper const errorHelper{connection_};
    EXPECT_CALL(
        *connection_,
        send(
            std::string{
                R"JSON({"error":"tooBusy","error_code":9,"error_message":"The server is too busy to help you now.","status":"error","type":"response"})JSON"
            },
            boost::beast::http::status::service_unavailable
        )
    );
    errorHelper.sendTooBusyError();
}

TEST_F(ErrorHandlingTests, sendJsonParsingError_UpgradedConnection)
{
    connection_->upgraded = true;
    ErrorHelper const errorHelper{connection_};
    EXPECT_CALL(
        *connection_,
        send(
            std::string{
                R"JSON({"error":"badSyntax","error_code":1,"error_message":"Syntax error.","status":"error","type":"response"})JSON"
            },
            boost::beast::http::status::ok
        )
    );
    errorHelper.sendJsonParsingError();
}

TEST_F(ErrorHandlingTests, sendJsonParsingError_NotUpgradedConnection)
{
    connection_->upgraded = false;
    ErrorHelper const errorHelper{connection_};
    EXPECT_CALL(
        *connection_,
        send(
            std::string{"Unable to parse JSON from the request"},
            boost::beast::http::status::bad_request
        )
    );
    errorHelper.sendJsonParsingError();
}