//===- llvm/unittest/Support/Chrono.cpp - Time utilities tests ------------===//
//
//           The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/Chrono.h"
#include "llvm/ADT/SmallVector.h"
#include "gtest/gtest.h"

using namespace llvm;
using namespace llvm::sys;
using namespace std::chrono;

namespace {

TEST(Chrono, TimeTConversion) {
  EXPECT_EQ(time_t(0), toTimeT(toTimePoint(time_t(0))));
  EXPECT_EQ(time_t(1), toTimeT(toTimePoint(time_t(1))));
  EXPECT_EQ(time_t(47), toTimeT(toTimePoint(time_t(47))));

  TimePoint<> TP;
  EXPECT_EQ(TP, toTimePoint(toTimeT(TP)));
  TP += seconds(1);
  EXPECT_EQ(TP, toTimePoint(toTimeT(TP)));
  TP += hours(47);
  EXPECT_EQ(TP, toTimePoint(toTimeT(TP)));
}

TEST(Chrono, StringConversion) {
  std::string S;
  raw_string_ostream OS(S);
  OS << system_clock::now();

  // Do a basic sanity check on the output.
  // The format we expect is YYYY-MM-DD HH:MM:SS.MMMUUUNNN
  StringRef Date, Time;
  std::tie(Date, Time) = StringRef(OS.str()).split(' ');

  SmallVector<StringRef, 3> Components;
  Date.split(Components, '-');
  ASSERT_EQ(3u, Components.size());
  EXPECT_EQ(4u, Components[0].size());
  EXPECT_EQ(2u, Components[1].size());
  EXPECT_EQ(2u, Components[2].size());

  StringRef Sec, Nano;
  std::tie(Sec, Nano) = Time.split('.');

  Components.clear();
  Sec.split(Components, ':');
  ASSERT_EQ(3u, Components.size());
  EXPECT_EQ(2u, Components[0].size());
  EXPECT_EQ(2u, Components[1].size());
  EXPECT_EQ(2u, Components[2].size());
  EXPECT_EQ(9u, Nano.size());
}

// Test that toTimePoint and toTimeT can be called with a arguments with varying
// precisions.
TEST(Chrono, ImplicitConversions) {
  std::time_t TimeT = 47;
  TimePoint<seconds> Sec = toTimePoint(TimeT);
  TimePoint<milliseconds> Milli = toTimePoint(TimeT);
  TimePoint<microseconds> Micro = toTimePoint(TimeT);
  TimePoint<nanoseconds> Nano = toTimePoint(TimeT);
  EXPECT_EQ(Sec, Milli);
  EXPECT_EQ(Sec, Micro);
  EXPECT_EQ(Sec, Nano);
  EXPECT_EQ(TimeT, toTimeT(Sec));
  EXPECT_EQ(TimeT, toTimeT(Milli));
  EXPECT_EQ(TimeT, toTimeT(Micro));
  EXPECT_EQ(TimeT, toTimeT(Nano));
}

} // anonymous namespace