// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <gtest/gtest.h>

#include "garnet/lib/perfmon/config.h"

namespace perfmon {
namespace {

// We use fake events here as these values are just passed to the driver,
// and this let's us be architecture-independent.
constexpr EventId kEventOne = MakeEventId(kGroupMisc, 1);
constexpr EventId kEventTwo = MakeEventId(kGroupMisc, 2);

TEST(Config, Events) {
  Config config;

  EXPECT_EQ(config.GetEventCount(), 0u);

  EXPECT_EQ(config.AddEvent(kEventOne, 0, 0),
            Config::Status::OK);
  EXPECT_EQ(config.GetEventCount(), 1u);

  EXPECT_EQ(config.AddEvent(kEventTwo, 1000, Config::kFlagOs),
            Config::Status::OK);
  EXPECT_EQ(config.GetEventCount(), 2u);

  std::vector<Config::EventConfig> events;
  config.IterateOverEvents([&events](const Config::EventConfig& event) {
    events.push_back(event);
  });

  EXPECT_EQ(events.size(), 2u);
  for (const auto& event : events) {
    switch (event.event) {
    case kEventOne:
      EXPECT_EQ(event.rate, 0u);
      EXPECT_EQ(event.flags, 0u);
      break;
    case kEventTwo:
      EXPECT_EQ(event.rate, 1000u);
      EXPECT_EQ(event.flags, Config::kFlagOs);
      break;
    }
  }
}

TEST(Config, SampleMode) {
  Config config;

  EXPECT_EQ(config.AddEvent(kEventOne, 1000, Config::kFlagOs),
            Config::Status::OK);

  EXPECT_EQ(config.GetMode(), CollectionMode::kSample);
}

TEST(Config, TallyMode) {
  Config config;

  EXPECT_EQ(config.AddEvent(kEventTwo, 0, Config::kFlagOs),
            Config::Status::OK);

  EXPECT_EQ(config.GetMode(), CollectionMode::kTally);
}

TEST(Config, Reset) {
  Config config;

  EXPECT_EQ(config.AddEvent(kEventOne, 10, Config::kFlagOs),
            Config::Status::OK);
  EXPECT_EQ(config.AddEvent(kEventTwo, 1000, Config::kFlagUser),
            Config::Status::OK);
  EXPECT_EQ(config.GetEventCount(), 2u);

  config.Reset();
  EXPECT_EQ(config.GetEventCount(), 0u);
}

TEST(Config, ToString) {
  Config config;

  EXPECT_EQ(config.AddEvent(kEventOne, 10, Config::kFlagOs),
            Config::Status::OK);

  EXPECT_EQ(config.ToString(), "event 0x2001, rate 10, flags 0x1");
}

}  // namespace
}  // namespace perfmon