/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 **/

#include <cstddef>
#include <string>

#include "types/IntervalLit.hpp"
#include "types/NullType.hpp"
#include "types/Type.hpp"
#include "types/TypeFactory.hpp"
#include "types/TypeID.hpp"
#include "types/TypedValue.hpp"
#include "types/YearMonthIntervalType.hpp"
#include "types/tests/TypeTest_common.hpp"
#include "utility/MemStream.hpp"

#include "gtest/gtest.h"

namespace quickstep {

namespace {

static const YearMonthIntervalLit kSampleIntervals[] = {
    {0}, {1}, {2},
    {12}, {13}, {14},
    {24}, {25}, {26}};

static const char *kSampleStrings[] = {
    "0 mons", "1 mon", "2 mons",
    "1 year", "1 year 1 mon", "1 year 2 mons",
    "2 years", "2 years 1 mon", "2 years 2 mons"};

}  // namespace

TEST(YearMonthIntervalTypeTest, PrintValueTest) {
  const Type &yearmonth_interval_type = TypeFactory::GetType(kYearMonthInterval);
  for (std::size_t sample_idx = 0;
       sample_idx < (sizeof(kSampleIntervals) / sizeof(kSampleIntervals[0]));
       ++sample_idx) {
    EXPECT_EQ(std::string(kSampleStrings[sample_idx]),
              yearmonth_interval_type.printValueToString(TypedValue(kSampleIntervals[sample_idx])));

    // Also check the negative version of the same interval.
    if (kSampleIntervals[sample_idx].months != 0) {
      std::string negative_string("-");
      negative_string.append(kSampleStrings[sample_idx]);
      EXPECT_EQ(negative_string,
                yearmonth_interval_type.printValueToString(TypedValue(- kSampleIntervals[sample_idx])));
    }
  }
}

namespace {

static const std::size_t kPaddedLength = 200;

void CheckPrintValueToFile(const YearMonthIntervalLit literal_value,
                           const std::string &expected_string) {
  // Test writing to FILE* with a MemStream.
  MemStream memstream;

  const Type &yearmonth_interval_type = TypeFactory::GetType(kYearMonthInterval);
  const TypedValue yearmonth_interval_value
      = yearmonth_interval_type.makeValue(&literal_value);

  // First, check printing without any padding
  yearmonth_interval_type.printValueToFile(yearmonth_interval_value, memstream.file());
  EXPECT_STREQ(expected_string.c_str(), memstream.str());
  memstream.reset();

  // Specifying padding less than the length needed to print the value should
  // have no effect.
  yearmonth_interval_type.printValueToFile(yearmonth_interval_value, memstream.file(), 1);
  EXPECT_STREQ(expected_string.c_str(), memstream.str());
  memstream.reset();

  // Test padding up to a specified length.
  std::string padded(kPaddedLength - expected_string.size(), ' ');
  padded.append(expected_string);
  yearmonth_interval_type.printValueToFile(yearmonth_interval_value,
                                           memstream.file(),
                                           kPaddedLength);
  EXPECT_STREQ(padded.c_str(), memstream.str());
  memstream.reset();
}

}  // namespace

TEST(YearMonthIntervalTypeTest, PrintValueToFileTest) {
  for (std::size_t sample_idx = 0;
       sample_idx < (sizeof(kSampleIntervals) / sizeof(kSampleIntervals[0]));
       ++sample_idx) {
    CheckPrintValueToFile(kSampleIntervals[sample_idx],
                          kSampleStrings[sample_idx]);

    // Also check the negative version of the same interval.
    if (kSampleIntervals[sample_idx].months != 0) {
      std::string negative_string("-");
      negative_string.append(kSampleStrings[sample_idx]);
      CheckPrintValueToFile(- kSampleIntervals[sample_idx],
                            negative_string);
    }
  }
}

namespace {

static const YearMonthIntervalLit kAdditionalParseSampleIntervals[] = {
    {0},
    {0},
    {0},
    {12 * 12 + 6},
    {123},
    {123},
    {123},
    {123},
    {123 * 12},
    {123 * 12},
    {123 * 12},
    {123 * 10 * 12},
    {123 * 10 * 12},
    {123 * 100 * 12},
    {123 * 100 * 12},
    {123 * 1000 * 12},
    {123 * 1000 * 12}
};

static const char *kAdditionalParseSampleStrings[] = {
    "0 years 0 mons",
    "0 years",
    "0 mons",
    "12 y 6 month",
    "123 mon",
    "123 mons",
    "123 mONth",
    "123 months",
    "123 y",
    "123 year",
    "123 years",
    "123 deCaDe",
    "123 decades",
    "123 century",
    "123 centuries",
    "123 millennium",
    "123 millennia"
};

}  // namespace

TEST(YearMonthIntervalTypeTest, ParseValueFromStringTest) {
  // Test various valid strings.
  for (std::size_t sample_idx = 0;
       sample_idx < (sizeof(kSampleIntervals) / sizeof(kSampleIntervals[0]));
       ++sample_idx) {
      type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
          kSampleIntervals[sample_idx], kSampleStrings[sample_idx]);

      // Also check the negative version of the same interval.
      std::string negative_string("-");
      negative_string.append(kSampleStrings[sample_idx]);
      type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
          - kSampleIntervals[sample_idx], negative_string);
  }

  // Test additional strings that are still valid for parsing but aren't part
  // of the narrower canonical print format.
  for (std::size_t sample_idx = 0;
       sample_idx < (sizeof(kAdditionalParseSampleIntervals)
                     / sizeof(kAdditionalParseSampleIntervals[0]));
       ++sample_idx) {
      type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
          kAdditionalParseSampleIntervals[sample_idx],
          kAdditionalParseSampleStrings[sample_idx]);

      // Also check the negative version of the same interval.
      std::string negative_string("-");
      negative_string.append(kAdditionalParseSampleStrings[sample_idx]);
      type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
          - kAdditionalParseSampleIntervals[sample_idx], negative_string);
  }

  // Test some strings with quote marks as allowed by SQL.
  type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
      YearMonthIntervalLit{123},
      "'123 mons'");
  type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
      YearMonthIntervalLit{123},
      "'123' mons");
  type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
      YearMonthIntervalLit{-123},
      "'-123 mons'");
  type_test::CheckSuccessfulParseLiteralValueFromString<YearMonthIntervalType>(
      YearMonthIntervalLit{-123},
      "'-123' mons");

  // Test some parses that we expect to fail.
  const Type &yearmonth_interval_type = TypeFactory::GetType(kYearMonthInterval);
  TypedValue value;
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("foo", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("42", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("--1 month", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("1 year -1 month", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("-1 year -1 month", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("1 day", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("1 century 1 month", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("1 year 1 month ", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString(" 1 year 1 month ", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("2 millenniums", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("1 month ", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString(" 1 month ", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("'123' 'mons'", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("123 'mons'", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("-'123 mons'", &value));
  EXPECT_FALSE(yearmonth_interval_type.parseValueFromString("-'123' mons", &value));
}

TEST(YearMonthIntervalTypeTest, IsCoercibleFromTest) {
  type_test::CheckIsCoercibleFrom(YearMonthIntervalType::InstanceNonNullable(),
                                  {kYearMonthInterval});
}

TEST(YearMonthIntervalTypeTest, IsSafelyCoercibleFromTest) {
  type_test::CheckIsSafelyCoercibleFrom(YearMonthIntervalType::InstanceNonNullable(),
                                        {kYearMonthInterval});
}

TEST(YearMonthIntervalTypeTest, CoerceValueTest) {
  // No change.
  type_test::CheckLiteralCoerce<YearMonthIntervalType, YearMonthIntervalType>(
      YearMonthIntervalLit{0});

  // 123 months (10 years, 3 months) forward.
  type_test::CheckLiteralCoerce<YearMonthIntervalType, YearMonthIntervalType>(
      YearMonthIntervalLit{123});

  // 123 months (10 years, 3 months) backward.
  type_test::CheckLiteralCoerce<YearMonthIntervalType, YearMonthIntervalType>(
      YearMonthIntervalLit{-123});

  type_test::CheckNullLiteralCoerce<YearMonthIntervalType, YearMonthIntervalType>();
  type_test::CheckNullLiteralCoerce<YearMonthIntervalType, NullType>();
}

}  // namespace quickstep