////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2019 ArangoDB GmbH, Cologne, Germany
///
/// Licensed under the Apache License, Version 2.0 (the "License");
/// you may not use this file except in compliance with the License.
/// You may obtain a copy of the License at
///
///     http://www.apache.org/licenses/LICENSE-2.0
///
/// Unless required by applicable law or agreed to in writing, software
/// distributed under the License is distributed on an "AS IS" BASIS,
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
/// See the License for the specific language governing permissions and
/// limitations under the License.
///
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
///
/// @author Markus Pfeiffer
////////////////////////////////////////////////////////////////////////////////

#include "../Mocks/Servers.h"
#include "QueryHelper.h"
#include "gtest/gtest.h"

#include "Aql/Query.h"
#include "Aql/VelocyPackHelper.h"
#include "RestServer/QueryRegistryFeature.h"

#include <velocypack/Buffer.h>
#include <velocypack/Builder.h>
#include <velocypack/Iterator.h>
#include <velocypack/Slice.h>
#include <velocypack/velocypack-aliases.h>

#include "../IResearch/IResearchQueryCommon.h"

#include "Logger/LogMacros.h"

using namespace arangodb;
using namespace arangodb::aql;

namespace arangodb {
namespace tests {
namespace aql {

class InsertExecutorTest : public ::testing::Test {
 protected:
  mocks::MockAqlServer server;
  TRI_vocbase_t& vocbase;

  std::string const collectionName = "UnitTestCollection";
  std::string const checkQuery =
      "FOR i IN " + collectionName + " SORT i.value RETURN i.value";

 public:
  InsertExecutorTest() : vocbase(server.getSystemDatabase()) {}

  void SetUp() override {
    SCOPED_TRACE("SetUp");
    auto info = VPackParser::fromJson("{\"name\": \"" + collectionName + "\"}");
    auto collection = vocbase.createCollection(info->slice());
    ASSERT_NE(collection.get(), nullptr) << "Failed to create collection";
  }
};

class InsertExecutorTestCount : public InsertExecutorTest,
                                public ::testing::WithParamInterface<size_t> {
 public:
  size_t nDocs;
  std::string nDocsString;

 public:
  InsertExecutorTestCount()
      : nDocs(GetParam()), nDocsString(std::to_string(GetParam())) {}
};

class InsertExecutorTestCounts
    : public InsertExecutorTest,
      public ::testing::WithParamInterface<std::vector<size_t>> {};

TEST_F(InsertExecutorTest, basic) {
  std::string query = R"(INSERT { value: 1 } IN )" + collectionName;
  AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

  auto expected = VPackParser::fromJson(R"([1])");
  AssertQueryHasResult(vocbase, checkQuery, expected->slice());
}

TEST_F(InsertExecutorTest, insert_but_not_rev) {
  std::string const invalidRev = "IAmAnInvalidRev";
  std::string query = R"(INSERT { _key: "IAmAKey", _rev: ")" + invalidRev +
                      R"(" } IN )" + collectionName;
  AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

  auto const bindParameters = VPackParser::fromJson("{ }");

  auto queryResult =
      arangodb::tests::executeQuery(vocbase,
                                    "FOR d IN " + collectionName + " RETURN d",
                                    bindParameters);
  ASSERT_TRUE(queryResult.ok());
  auto slice = queryResult.data->slice();
  ASSERT_TRUE(slice.isArray());
  ASSERT_EQ(slice.length(), 1);

  auto doc = slice.at(0);
  ASSERT_NE(arangodb::basics::VelocyPackHelper::getStringValue(doc, "_rev", ""), invalidRev);
}

TEST_F(InsertExecutorTest, insert_ignore_error_default) {
  {
    std::string query =
        R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" + collectionName;
    AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());
  }

  {
    std::string query = R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" +
                        collectionName + R"( OPTIONS { ignoreErrors: false } )";
    AssertQueryFailsWith(vocbase, query, TRI_ERROR_ARANGO_UNIQUE_CONSTRAINT_VIOLATED);
  }
}

TEST_F(InsertExecutorTest, insert_ignore_error_true) {
  {
    std::string query =
        R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" + collectionName;
    AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());
  }

  {
    std::string query = R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" +
                        collectionName + R"( OPTIONS { ignoreErrors: true } )";
    AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

    auto expected = VPackParser::fromJson(R"([1])");
    AssertQueryHasResult(vocbase, checkQuery, expected->slice());
  }
}

TEST_F(InsertExecutorTest, insert_ignore_error_false) {
  {
    std::string query =
        R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" + collectionName;
    AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());
  }

  {
    std::string query = R"(INSERT { _key: "iAmADocumentDoWhatIsay", value: 1 } IN )" +
                        collectionName + R"( OPTIONS { ignoreErrors: false } )";
    AssertQueryFailsWith(vocbase, query, TRI_ERROR_ARANGO_UNIQUE_CONSTRAINT_VIOLATED);
  }
}

TEST_F(InsertExecutorTest, multi_insert_same_collection) {
  std::string query = "LET x = (INSERT { value: 15} IN " + collectionName +
                      ")" + "LET y = (INSERT {value: 16} IN " + collectionName +
                      ")" + " RETURN [x,y]";

  AssertQueryFailsWith(vocbase, query, TRI_ERROR_QUERY_ACCESS_AFTER_MODIFICATION);
}

TEST_P(InsertExecutorTestCount, insert_without_return) {
  std::string query = std::string("FOR i IN 1..") + std::to_string(GetParam()) +
                      " INSERT { value: i } INTO " + collectionName;

  AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

  VPackBuilder builder;
  builder.openArray();
  for (size_t i = 1; i <= GetParam(); i++) {
    builder.add(VPackValue(i));
  }
  builder.close();
  AssertQueryHasResult(vocbase, checkQuery, builder.slice());
}

TEST_P(InsertExecutorTestCount, insert_with_key_with_return) {
  auto const bindParameters = VPackParser::fromJson("{ }");

  std::string query = std::string("FOR i IN 1..") + std::to_string(GetParam()) +
                      " INSERT { _key: TO_STRING(i), value: i } INTO " +
                      collectionName + " RETURN NEW.value";
  auto result = arangodb::tests::executeQuery(vocbase, query, bindParameters);
  TRI_ASSERT(result.data->slice().isArray());
  TRI_ASSERT(result.data->slice().length() == GetParam());

  AssertQueryHasResult(vocbase, checkQuery, result.data->slice());
}

TEST_P(InsertExecutorTestCount, insert_with_key_without_return) {
  std::string query = std::string("FOR i IN 1..") + std::to_string(GetParam()) +
                      " INSERT { _key: TO_STRING(i), value: i } INTO " + collectionName;

  AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

  VPackBuilder builder;
  builder.openArray();
  for (size_t i = 1; i <= GetParam(); i++) {
    builder.add(VPackValue(i));
  }
  builder.close();
  AssertQueryHasResult(vocbase, checkQuery, builder.slice());
}

TEST_P(InsertExecutorTestCount, insert_with_return) {
  auto const bindParameters = VPackParser::fromJson("{ }");

  std::string query = std::string("FOR i IN 1..") + std::to_string(GetParam()) +
                      " INSERT { value: i } INTO " + collectionName +
                      " RETURN NEW.value";
  auto result = arangodb::tests::executeQuery(vocbase, query, bindParameters);
  TRI_ASSERT(result.data->slice().isArray());
  TRI_ASSERT(result.data->slice().length() == GetParam());

  AssertQueryHasResult(vocbase, checkQuery, result.data->slice());
}

INSTANTIATE_TEST_CASE_P(InsertExecutorTestInstance, InsertExecutorTestCount,
                        testing::Values(1, 100, 999, 1000, 1001));

TEST_P(InsertExecutorTestCounts, insert_multiple_without_return) {
  std::vector<size_t> insertedVals = {};
  std::vector<size_t> param = GetParam();
  for (auto i : param) {
    std::string query = std::string("FOR i IN 1..") + std::to_string(i) +
                        " INSERT { value: i } INTO " + collectionName;

    AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());

    for (size_t j = 1; j <= i; j++) {
      insertedVals.push_back(j);
    }
  }

  std::sort(std::begin(insertedVals), std::end(insertedVals));
  VPackBuilder builder;
  builder.openArray();
  for (auto i : insertedVals) {
    builder.add(VPackValue(i));
  }
  builder.close();
  AssertQueryHasResult(vocbase, checkQuery, builder.slice());
}

TEST_P(InsertExecutorTestCounts, insert_multiple_with_return) {
  std::vector<size_t> insertedVals = {};
  auto const bindParameters = VPackParser::fromJson("{ }");

  std::vector<size_t> param = GetParam();

  for (auto i : param) {
    std::string query = std::string("FOR i IN 1..") + std::to_string(i) +
                        " INSERT { value: i } INTO " + collectionName +
                        " RETURN NEW ";
    auto result = arangodb::tests::executeQuery(vocbase, query, bindParameters);
    ASSERT_TRUE(result.ok());
    for (size_t j = 1; j <= i; j++) {
      insertedVals.push_back(j);
    }
  }

  std::sort(std::begin(insertedVals), std::end(insertedVals));
  VPackBuilder builder;
  builder.openArray();
  for (auto i : insertedVals) {
    builder.add(VPackValue(i));
  }
  builder.close();
  AssertQueryHasResult(vocbase, checkQuery, builder.slice());
}

INSTANTIATE_TEST_CASE_P(
    InsertExecutorTestInstance, InsertExecutorTestCounts,
    testing::Values(std::vector<size_t>{1}, std::vector<size_t>{100},
                    std::vector<size_t>{999}, std::vector<size_t>{1000},
                    std::vector<size_t>{1001}, std::vector<size_t>{1, 100, 1000, 1000, 900},
                    std::vector<size_t>{10, 10, 10, 10, 10, 100, 100, 10, 100,
                                        1000, 1000, 900, 10, 100}));

// OLD is a keyword, but only sometimes. In particular in insert queries it
// isn't unless overwrite: true. Yes, really.
TEST_F(InsertExecutorTest, insert_return_old) {
  std::string query = std::string("FOR i IN 1..1 INSERT { value: i } INTO ") +
                      collectionName + " RETURN OLD";

  AssertQueryFailsWith(vocbase, query, TRI_ERROR_ARANGO_DATA_SOURCE_NOT_FOUND);
}

TEST_P(InsertExecutorTestCount, insert_with_key) {
  std::string query =
      std::string("FOR i IN 1.." + nDocsString +
                  " INSERT { _key: TO_STRING(i), value: i } INTO ") +
      collectionName + " SORT NEW.value RETURN NEW.value";

  VPackBuilder builder;
  builder.openArray();
  for (size_t i = 1; i <= nDocs; i++) {
    builder.add(VPackValue(i));
  }
  builder.close();

  AssertQueryHasResult(vocbase, query, builder.slice());
}

TEST_P(InsertExecutorTestCount, insert_with_key_and_overwrite) {
  // Write
  {
    std::string query =
        std::string("FOR i IN 1.." + nDocsString +
                    " INSERT { _key: TO_STRING(i), value: i } INTO ") +
        collectionName +
        " OPTIONS { overwrite: true } SORT NEW.value RETURN [OLD.value, NEW.value]";

    VPackBuilder builder;
    builder.openArray();
    for (size_t i = 1; i <= nDocs; i++) {
      builder.openArray();
      builder.add(VPackSlice::nullSlice());
      builder.add(VPackValue(i));
      builder.close();
    }
    builder.close();
    AssertQueryHasResult(vocbase, query, builder.slice());
  }

  // Overwrite
  {
    std::string query =
        std::string("FOR i IN 1.." + nDocsString +
                    " INSERT { _key: TO_STRING(i), value: -i } INTO ") +
        collectionName +
        " OPTIONS { overwrite: true } SORT NEW.value RETURN NEW.value";

    VPackBuilder builder;
    builder.openArray();
    for (int i = -nDocs; i <= -1; i++) {
      builder.add(VPackValue(i));
    }
    builder.close();
    AssertQueryHasResult(vocbase, query, builder.slice());
  }
}

TEST_P(InsertExecutorTestCount, insert_with_key_and_no_overwrite) {
  std::string query =
      std::string("FOR i IN 1.." + nDocsString +
                  " INSERT { _key: TO_STRING(i), value: i } INTO ") +
      collectionName + " SORT NEW.value RETURN NEW.value";

  VPackBuilder builder;
  builder.openArray();
  for (size_t i = 1; i <= nDocs; i++) {
    builder.add(VPackValue(i));
  }
  builder.close();

  // This is intentional: We write the entries once, then overwrite them again
  // The second query should fail with a uniqueness violation on _key
  AssertQueryHasResult(vocbase, query, builder.slice());
  AssertQueryFailsWith(vocbase, query, TRI_ERROR_ARANGO_UNIQUE_CONSTRAINT_VIOLATED);
}

TEST_P(InsertExecutorTestCount, insert_with_key_and_no_overwrite_ignore_errors) {
  std::string query =
      std::string("FOR i IN 1.." + nDocsString +
                  " INSERT { _key: TO_STRING(i), value: i } INTO ") +
      collectionName +
      " OPTIONS { ignoreErrors: true } SORT NEW.value RETURN NEW.value";

  VPackBuilder builder;
  builder.openArray();
  for (size_t i = 1; i <= nDocs; i++) {
    builder.add(VPackValue(i));
  }
  builder.close();

  // This is intentional: We write the entries once, then overwrite them again
  // The second query should fail with a uniqueness violation on _key
  AssertQueryHasResult(vocbase, query, builder.slice());
  AssertQueryHasResult(vocbase, query, VPackSlice::emptyArraySlice());
}

TEST_P(InsertExecutorTestCount, insert_inside_subquery) {
  std::string query = "FOR i IN 1.." + nDocsString +
                      " LET x = (INSERT { value: i } INTO " + collectionName +
                      " RETURN NEW)"
                      " SORT x[0].value"
                      " LIMIT 10, null "
                      " RETURN x[0].value";

  // TODO: is this correct?
  {
    VPackBuilder builder;
    builder.openArray();
    for (size_t i = 11; i <= nDocs; i++) {
      builder.add(VPackValue(i));
    }
    builder.close();
    AssertQueryHasResult(vocbase, query, builder.slice());
  }

  {
    VPackBuilder builder;
    builder.openArray();
    for (size_t i = 1; i <= nDocs; i++) {
      builder.add(VPackValue(i));
    }
    builder.close();
    AssertQueryHasResult(vocbase, checkQuery, builder.slice());
  }
}

TEST_P(InsertExecutorTestCount, aggregate_insert_inside_subquery) {
  std::string query = "FOR i IN 1.." + nDocsString +
                      " LET x = (INSERT { value: i } INTO " + collectionName +
                      " RETURN NEW)"
                      " COLLECT AGGREGATE sum = SUM(x[0].value)"
                      " RETURN sum";

  VPackBuilder builder;
  builder.openArray();
  builder.add(VPackValue(nDocs * (nDocs + 1) / 2));
  builder.close();
  AssertQueryHasResult(vocbase, query, builder.slice());
}

}  // namespace aql
}  // namespace tests
}  // namespace arangodb