// Copyright 2018 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 "config/client_config.h"

#include <memory>
#include <string>
#include <utility>

#include "./logging.h"
#include "config/cobalt_registry.pb.h"
#include "config/encoding_config.h"
#include "config/encodings.pb.h"
#include "config/metric_config.h"
#include "config/metrics.pb.h"
#include "third_party/googletest/googletest/include/gtest/gtest.h"
#include "util/crypto_util/base64.h"

namespace cobalt {
namespace config {

namespace {
void AddMetric(int customer_id, int project_id, int id,
               CobaltRegistry* cobalt_registry) {
  Metric* metric = cobalt_registry->add_metric_configs();
  metric->set_customer_id(customer_id);
  metric->set_project_id(project_id);
  metric->set_id(id);
}

void AddMetric(int id, CobaltRegistry* cobalt_registry) {
  AddMetric(id, id, id, cobalt_registry);
}

void AddEncodingConfig(int customer_id, int project_id, int id,
                       CobaltRegistry* cobalt_registry) {
  EncodingConfig* encoding_config = cobalt_registry->add_encoding_configs();
  encoding_config->set_customer_id(customer_id);
  encoding_config->set_project_id(project_id);
  encoding_config->set_id(id);
}

void AddEncodingConfig(int id, CobaltRegistry* cobalt_registry) {
  AddEncodingConfig(id, id, id, cobalt_registry);
}

std::unique_ptr<ClientConfig> CreateFromCopyOfRegistry(
    const CobaltRegistry registry) {
  std::string cobalt_registry_bytes;
  EXPECT_TRUE(registry.SerializeToString(&cobalt_registry_bytes));
  return ClientConfig::CreateFromCobaltRegistryBytes(cobalt_registry_bytes);
}

}  // namespace

TEST(ClientConfigTest, CreateFromCobaltProjectRegistryBytesValidConfig) {
  std::string cobalt_registry_bytes;
  CobaltRegistry cobalt_registry;
  AddMetric(1, 1, 42, &cobalt_registry);
  AddMetric(1, 1, 43, &cobalt_registry);
  AddEncodingConfig(1, 1, 42, &cobalt_registry);
  AddEncodingConfig(1, 1, 43, &cobalt_registry);
  ASSERT_TRUE(cobalt_registry.SerializeToString(&cobalt_registry_bytes));
  auto client_config_project_id_pair =
      ClientConfig::CreateFromCobaltProjectRegistryBytes(cobalt_registry_bytes);
  auto client_config = std::move(client_config_project_id_pair.first);
  ASSERT_NE(nullptr, client_config);
  EXPECT_EQ(1u, client_config_project_id_pair.second);
  EXPECT_EQ(nullptr, client_config->EncodingConfig(1, 1, 41));
  EXPECT_NE(nullptr, client_config->EncodingConfig(1, 1, 42));
  EXPECT_NE(nullptr, client_config->EncodingConfig(1, 1, 43));
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 41));
  EXPECT_NE(nullptr, client_config->Metric(1, 1, 42));
  EXPECT_NE(nullptr, client_config->Metric(1, 1, 43));
}

TEST(ClientConfigTest, CreateFromCobaltProjectRegistryBytesInvalidConfig) {
  std::string cobalt_registry_bytes;
  CobaltRegistry cobalt_registry;
  AddMetric(1, 1, 42, &cobalt_registry);
  AddMetric(1, 1, 43, &cobalt_registry);
  AddEncodingConfig(1, 2, 42, &cobalt_registry);
  AddEncodingConfig(1, 2, 43, &cobalt_registry);
  ASSERT_TRUE(cobalt_registry.SerializeToString(&cobalt_registry_bytes));
  auto client_config_project_id_pair =
      ClientConfig::CreateFromCobaltProjectRegistryBytes(cobalt_registry_bytes);
  auto client_config = std::move(client_config_project_id_pair.first);
  ASSERT_EQ(nullptr, client_config);
}

TEST(ClientConfigTest, CreateFromCobaltRegistryBytes) {
  std::string cobalt_registry_bytes;
  CobaltRegistry cobalt_registry;
  AddMetric(42, &cobalt_registry);
  AddMetric(43, &cobalt_registry);
  AddEncodingConfig(42, &cobalt_registry);
  AddEncodingConfig(43, &cobalt_registry);
  ASSERT_TRUE(cobalt_registry.SerializeToString(&cobalt_registry_bytes));
  auto client_config =
      ClientConfig::CreateFromCobaltRegistryBytes(cobalt_registry_bytes);
  ASSERT_NE(nullptr, client_config);
  EXPECT_EQ(nullptr, client_config->EncodingConfig(41, 41, 41));
  EXPECT_NE(nullptr, client_config->EncodingConfig(42, 42, 42));
  EXPECT_NE(nullptr, client_config->EncodingConfig(43, 43, 43));
  EXPECT_EQ(nullptr, client_config->Metric(41, 41, 41));
  EXPECT_NE(nullptr, client_config->Metric(42, 42, 42));
  EXPECT_NE(nullptr, client_config->Metric(43, 43, 43));
}

TEST(ClientConfigTest, CreateFromCobaltRegistryBase64) {
  std::string cobalt_registry_bytes;
  CobaltRegistry cobalt_registry;
  AddMetric(42, &cobalt_registry);
  AddMetric(43, &cobalt_registry);
  AddEncodingConfig(42, &cobalt_registry);
  AddEncodingConfig(43, &cobalt_registry);
  ASSERT_TRUE(cobalt_registry.SerializeToString(&cobalt_registry_bytes));
  std::string cobalt_registry_base64;
  crypto::Base64Encode(cobalt_registry_bytes, &cobalt_registry_base64);
  auto client_config =
      ClientConfig::CreateFromCobaltRegistryBase64(cobalt_registry_base64);
  ASSERT_NE(nullptr, client_config);
  EXPECT_EQ(nullptr, client_config->EncodingConfig(41, 41, 41));
  EXPECT_NE(nullptr, client_config->EncodingConfig(42, 42, 42));
  EXPECT_NE(nullptr, client_config->EncodingConfig(43, 43, 43));
  EXPECT_EQ(nullptr, client_config->Metric(41, 41, 41));
  EXPECT_NE(nullptr, client_config->Metric(42, 42, 42));
  EXPECT_NE(nullptr, client_config->Metric(43, 43, 43));
}

// Tests the method CreateFromCobaltRegistry along with the accessors
// is_single_project(), is_empty(), single_customer_id() and single_project_id()
// and IsLegacy(), in the case that the CobaltRegistry contains only Cobalt 0.1
// data and no Cobalt 1.0 data.
TEST(ClientConfigTest, CreateFromCobaltRegistry) {
  CobaltRegistry cobalt_registry;
  auto client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_TRUE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());
  // Check that even though we have not added any metrics, querying for
  // a metric does not cause a crash.
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 1));

  AddMetric(42, &cobalt_registry);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_TRUE(client_config->is_single_project());
  EXPECT_EQ(42u, client_config->single_customer_id());
  EXPECT_EQ(42u, client_config->single_project_id());
  EXPECT_TRUE(client_config->IsLegacy());

  AddEncodingConfig(42u, &cobalt_registry);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_TRUE(client_config->is_single_project());
  EXPECT_EQ(42u, client_config->single_customer_id());
  EXPECT_EQ(42u, client_config->single_project_id());
  EXPECT_TRUE(client_config->IsLegacy());

  AddMetric(43u, &cobalt_registry);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());
}

// Tests the method CreateFromCobaltRegistry along with the accessors
// is_single_project(), is_empty(), single_customer_id(), single_project_id()
// IsLegacy() and TakeCustomerConfig()
// in the case that the CobaltRegistry contains only Cobalt 1.0 data and no
// Cobalt 0.1 data.
//
// DEPRECATED: Remove this test once we stop supporting Cobalt 1.0 data
// in ClientConfig.
TEST(ClientConfigTest, CreateFromCobaltRegistry1Point0) {
  CobaltRegistry cobalt_registry;
  auto client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_TRUE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());
  // Check that even though we have not added any metrics, querying for
  // a metric does not cause a crash.
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 1));

  // Add one empty customer.
  auto* customer = cobalt_registry.add_customers();
  customer->set_customer_id(51);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());
  // Check that even though we have not added any metrics, querying for
  // a metric does not cause a crash.
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 1));

  // Add one project to that customer
  auto project = customer->add_projects();
  project->set_project_id(101);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_TRUE(client_config->is_single_project());
  EXPECT_EQ(51u, client_config->single_customer_id());
  EXPECT_EQ(101u, client_config->single_project_id());
  EXPECT_FALSE(client_config->IsLegacy());
  auto customer_config = client_config->TakeCustomerConfig();
  EXPECT_TRUE(customer_config);
  EXPECT_EQ(51u, customer_config->customer_id());
  // Check that even though we have not added any metrics, querying for
  // a metric does not cause a crash.
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 1));

  // Add a second project to that customer
  project = customer->add_projects();
  project->set_project_id(102);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  // Check that even though we have not added any metrics, querying for
  // a metric does not cause a crash.
  EXPECT_EQ(nullptr, client_config->Metric(1, 1, 1));
}

// Tests the method CreateFromCobaltRegistry along with the accessors
// is_single_project(), is_empty(), single_customer_id(), single_project_id()
// IsLegacy() and TakeCustomerConfig()
// in the case that the CobaltRegistry contains both Cobalt 1.0 data and
// Cobalt 0.1 data.
//
// DEPRECATED: Remove this test once we stop supporting Cobalt 1.0 data
// in ClientConfig.
TEST(ClientConfigTest, CreateFromCobaltRegistryMixed) {
  CobaltRegistry cobalt_registry;
  auto client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_TRUE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());

  // Add one empty customer.
  auto* customer = cobalt_registry.add_customers();
  customer->set_customer_id(51);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  EXPECT_TRUE(client_config->IsLegacy());

  // Add one project to that customer
  auto project = customer->add_projects();
  project->set_project_id(101);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_TRUE(client_config->is_single_project());
  EXPECT_EQ(51u, client_config->single_customer_id());
  EXPECT_EQ(101u, client_config->single_project_id());
  EXPECT_FALSE(client_config->IsLegacy());
  auto customer_config = client_config->TakeCustomerConfig();
  EXPECT_TRUE(customer_config);
  EXPECT_EQ(51u, customer_config->customer_id());

  // Add some Cobalt 0.1 data.
  AddMetric(42, &cobalt_registry);
  client_config = CreateFromCopyOfRegistry(cobalt_registry);
  EXPECT_FALSE(client_config->is_empty());
  EXPECT_FALSE(client_config->is_single_project());
  // Check that even though we have added some Cobalt 1.0 data we can
  // still query for Cobalt 0.1 metrics.
  EXPECT_NE(nullptr, client_config->Metric(42, 42, 42));
}

}  // namespace config
}  // namespace cobalt