// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

#include <iostream>

#include "gtest/gtest.h"
#include "http/core/routes.h"

namespace onnxruntime {
namespace server {
namespace test {

static const std::string predict_regex = R"(/(?:v1/models/([^/:]+)(?:/versions/(\d+))?:(classify|regress|predict)))";
using test_data = std::tuple<http::verb, std::string, std::string, std::string, std::string, http::status>;

void do_something(const std::string& name, const std::string& version,
                  const std::string& action, HttpContext& context) {
  auto noop = name + version + action + context.request.body();
}

void run_route(const std::string& pattern, http::verb method, const std::vector<test_data>& data, bool does_validate_data);

TEST(HttpRouteTests, RegisterTest) {
  Routes routes;
  EXPECT_TRUE(routes.RegisterController(http::verb::post, predict_regex, do_something));

  auto status_regex = R"(/v1/models(?:/([^/:]+))?(?:/versions/(\d+))?(?:\/(metadata))?)";
  EXPECT_TRUE(routes.RegisterController(http::verb::get, status_regex, do_something));
}

TEST(HttpRouteTests, PostRouteTest) {

  std::vector<test_data> actions{
      std::make_tuple(http::verb::post, "/v1/models/abc/versions/23:predict", "abc", "23", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/abc:predict", "abc", "", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/models/versions/45:predict", "models", "45", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/??$$%%@@$^^/versions/45:predict", "??$$%%@@$^^", "45", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/versions/versions/45:predict", "versions", "45", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/versions:predict", "versions", "", "predict", http::status::ok),
      std::make_tuple(http::verb::post, "/v1/models/default:predict", "default", "", "predict", http::status::ok)
      };


  run_route(predict_regex, http::verb::post, actions, true);
}

TEST(HttpRouteTests, PostRouteInvalidURLTest) {

  std::vector<test_data> actions{
      std::make_tuple(http::verb::post, "", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models:bar", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/abc/versions", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/abc/versions:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/a:bc/versions:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/abc/versions/2.0:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/models/abc/versions/2:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/versions/2:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/foo/versions/:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "v1/models/foo/versions/12:predict", "", "", "", http::status::not_found),
      std::make_tuple(http::verb::post, "/v1/models/abc/versions/23:foo", "", "", "", http::status::not_found)
      };

  run_route(predict_regex, http::verb::post, actions, false);
}

// These tests are because we currently only support POST and GET
// Some HTTP methods should be removed from test data if we support more (e.g. PUT)
TEST(HttpRouteTests, PostRouteInvalidMethodTest) {

  std::vector<test_data> actions{
      std::make_tuple(http::verb::get, "/v1/models/abc/versions/23:predict", "abc", "23", "predict", http::status::method_not_allowed),
      std::make_tuple(http::verb::put, "/v1/models", "", "", "", http::status::method_not_allowed),
      std::make_tuple(http::verb::delete_, "/v1/models", "", "", "", http::status::method_not_allowed),
      std::make_tuple(http::verb::head, "/v1/models", "", "", "", http::status::method_not_allowed)
      };

  run_route(predict_regex, http::verb::post, actions, false);
}

TEST(HttpRouteTests, PostRouteSpecialMethodTest){
  std::vector<test_data> actions{
    std::make_tuple(http::verb::post, "/score", "", "", "", http::status::ok)
  };

  run_route(R"(/score()()())", http::verb::post, actions, true);
}

void run_route(const std::string& pattern, http::verb method, const std::vector<test_data>& data, bool does_validate_data) {
  Routes routes;
  EXPECT_TRUE(routes.RegisterController(method, pattern, do_something));

  for (const auto& i : data) {
    http::verb test_method;
    std::string url_string;
    std::string name;
    std::string version;
    std::string action;
    HandlerFn fn;

    std::string expected_name;
    std::string expected_version;
    std::string expected_action;
    http::status expected_status;

    std::tie(test_method, url_string, expected_name, expected_version, expected_action, expected_status) = i;
    EXPECT_EQ(expected_status, routes.ParseUrl(test_method, url_string, name, version, action, fn)) << "On route " << url_string;
    if (does_validate_data) {
      EXPECT_EQ(name, expected_name) << "On route " << url_string;
      EXPECT_EQ(version, expected_version) << "On route " << url_string;
      EXPECT_EQ(action, expected_action) << "On route " << url_string;
    }
  }
}

}  // namespace test
}  // namespace server
}  // namespace onnxruntime