// Copyright 2017 Google Inc. All Rights Reserved.
//
// 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.

#include <string>

#include "base/time.h"
#include "cobalt/base/c_val_time_interval_timer_stats.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using ::testing::InSequence;
using ::testing::Return;
using ::testing::SaveArg;
using ::testing::_;

namespace base {

TEST(CValTimeIntervalTimerStatsTest, DefaultValues) {
  const std::string name = "TimeIntervalTimerStats";
  const size_t time_interval_in_ms = 50;

  base::CValTimeIntervalTimerStats<> cval(name, time_interval_in_ms);

  base::CValManager* cvm = base::CValManager::GetInstance();
  base::optional<std::string> count =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Cnt", name.c_str()));
  base::optional<std::string> avg =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Avg", name.c_str()));
  base::optional<std::string> min =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Min", name.c_str()));
  base::optional<std::string> max =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Max", name.c_str()));
  base::optional<std::string> std =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Std", name.c_str()));

  EXPECT_TRUE(count);
  EXPECT_EQ(*count, "0");
  EXPECT_TRUE(avg);
  EXPECT_EQ(*avg, "0us");
  EXPECT_TRUE(min);
  EXPECT_EQ(*min, "0us");
  EXPECT_TRUE(max);
  EXPECT_EQ(*max, "0us");
  EXPECT_TRUE(std);
  EXPECT_EQ(*std, "0us");
}

TEST(CValTimeIntervalTimerStatsTest, NoTimeout) {
  const std::string name = "TimeIntervalTimerStats";
  const size_t time_interval_in_ms = 50;

  base::CValTimeIntervalTimerStats<> cval(name, time_interval_in_ms);
  cval.Start(base::TimeTicks::FromInternalValue(1000));
  cval.Stop(base::TimeTicks::FromInternalValue(4000));
  cval.Start(base::TimeTicks::FromInternalValue(4000));
  cval.Stop(base::TimeTicks::FromInternalValue(13000));
  cval.Start(base::TimeTicks::FromInternalValue(13000));
  cval.Stop(base::TimeTicks::FromInternalValue(14000));
  cval.Start(base::TimeTicks::FromInternalValue(14000));
  cval.Stop(base::TimeTicks::FromInternalValue(21000));

  base::CValManager* cvm = base::CValManager::GetInstance();
  base::optional<std::string> count =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Cnt", name.c_str()));
  base::optional<std::string> avg =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Avg", name.c_str()));
  base::optional<std::string> min =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Min", name.c_str()));
  base::optional<std::string> max =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Max", name.c_str()));
  base::optional<std::string> std =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Std", name.c_str()));

  EXPECT_TRUE(count);
  EXPECT_EQ(*count, "0");
  EXPECT_TRUE(avg);
  EXPECT_EQ(*avg, "0us");
  EXPECT_TRUE(min);
  EXPECT_EQ(*min, "0us");
  EXPECT_TRUE(max);
  EXPECT_EQ(*max, "0us");
  EXPECT_TRUE(std);
  EXPECT_EQ(*std, "0us");
}

TEST(CValTimeIntervalTimerStatsTest, OneTimeout) {
  const std::string name = "TimeIntervalTimerStats";
  const size_t time_interval_in_ms = 50;

  base::CValTimeIntervalTimerStats<CValPublic> cval(name, time_interval_in_ms);
  cval.Start(base::TimeTicks::FromInternalValue(1000));
  cval.Stop(base::TimeTicks::FromInternalValue(4000));
  cval.Start(base::TimeTicks::FromInternalValue(4000));
  cval.Stop(base::TimeTicks::FromInternalValue(13000));
  cval.Start(base::TimeTicks::FromInternalValue(13000));
  cval.Stop(base::TimeTicks::FromInternalValue(14000));
  cval.Start(base::TimeTicks::FromInternalValue(14000));
  cval.Stop(base::TimeTicks::FromInternalValue(21000));
  cval.Start(base::TimeTicks::FromInternalValue(21000));
  cval.Stop(base::TimeTicks::FromInternalValue(26000));
  cval.Start(base::TimeTicks::FromInternalValue(26000));
  cval.Stop(base::TimeTicks::FromInternalValue(55000));

  base::CValManager* cvm = base::CValManager::GetInstance();
  base::optional<std::string> count =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Cnt", name.c_str()));
  base::optional<std::string> avg =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Avg", name.c_str()));
  base::optional<std::string> min =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Min", name.c_str()));
  base::optional<std::string> max =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Max", name.c_str()));
  base::optional<std::string> std =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Std", name.c_str()));

  EXPECT_TRUE(count);
  EXPECT_EQ(*count, "5");
  EXPECT_TRUE(avg);
  EXPECT_EQ(*avg, "5.0ms");
  EXPECT_TRUE(min);
  EXPECT_EQ(*min, "1000us");
  EXPECT_TRUE(max);
  EXPECT_EQ(*max, "9.0ms");
  EXPECT_TRUE(std);
  EXPECT_EQ(*std, "2.8ms");
}

TEST(CValTimeIntervalTimerStatsTest, TwoTimeouts) {
  const std::string name = "TimeIntervalTimerStats";
  const size_t time_interval_in_ms = 50;

  base::CValTimeIntervalTimerStats<CValDebug> cval(name, time_interval_in_ms);
  cval.Start(base::TimeTicks::FromInternalValue(1000));
  cval.Stop(base::TimeTicks::FromInternalValue(4000));
  cval.Start(base::TimeTicks::FromInternalValue(4000));
  cval.Stop(base::TimeTicks::FromInternalValue(13000));
  cval.Start(base::TimeTicks::FromInternalValue(13000));
  cval.Stop(base::TimeTicks::FromInternalValue(14000));
  cval.Start(base::TimeTicks::FromInternalValue(14000));
  cval.Stop(base::TimeTicks::FromInternalValue(21000));
  cval.Start(base::TimeTicks::FromInternalValue(21000));
  cval.Stop(base::TimeTicks::FromInternalValue(26000));
  cval.Start(base::TimeTicks::FromInternalValue(26000));
  cval.Stop(base::TimeTicks::FromInternalValue(55000));
  cval.Start(base::TimeTicks::FromInternalValue(55000));
  cval.Stop(base::TimeTicks::FromInternalValue(110000));

  base::CValManager* cvm = base::CValManager::GetInstance();
  base::optional<std::string> count =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Cnt", name.c_str()));
  base::optional<std::string> avg =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Avg", name.c_str()));
  base::optional<std::string> min =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Min", name.c_str()));
  base::optional<std::string> max =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Max", name.c_str()));
  base::optional<std::string> std =
      cvm->GetValueAsPrettyString(StringPrintf("%s.Std", name.c_str()));

  EXPECT_TRUE(count);
  EXPECT_EQ(*count, "1");
  EXPECT_TRUE(avg);
  EXPECT_EQ(*avg, "29.0ms");
  EXPECT_TRUE(min);
  EXPECT_EQ(*min, "29.0ms");
  EXPECT_TRUE(max);
  EXPECT_EQ(*max, "29.0ms");
  EXPECT_TRUE(std);
  EXPECT_EQ(*std, "0us");
}

}  // namespace base