#include "envoy/config/trace/v3/http_tracer.pb.h"
#include "envoy/config/trace/v3/xray.pb.h"
#include "envoy/config/trace/v3/xray.pb.validate.h"
#include "envoy/registry/registry.h"

#include "extensions/tracers/xray/config.h"

#include "test/mocks/server/instance.h"
#include "test/mocks/server/tracer_factory.h"
#include "test/mocks/server/tracer_factory_context.h"
#include "test/test_common/environment.h"

#include "gmock/gmock.h"
#include "gtest/gtest.h"

using ::testing::Throw;

namespace Envoy {
namespace Extensions {
namespace Tracers {
namespace XRay {
namespace {

TEST(XRayTracerConfigTest, XRayHttpTracerWithTypedConfig) {
  NiceMock<Server::Configuration::MockTracerFactoryContext> context;

  const std::string yaml_string = R"EOF(
  http:
    name: xray
    typed_config:
      "@type": type.googleapis.com/envoy.config.trace.v3.XRayConfig
      daemon_endpoint:
        protocol: UDP
        address: 127.0.0.1
        port_value: 2000
      segment_name: AwsAppMesh
      sampling_rule_manifest:
        filename: "rules.json")EOF";

  envoy::config::trace::v3::Tracing configuration;
  TestUtility::loadFromYaml(yaml_string, configuration);

  XRayTracerFactory factory;
  auto message = Config::Utility::translateToFactoryConfig(
      configuration.http(), ProtobufMessage::getStrictValidationVisitor(), factory);
  Tracing::HttpTracerSharedPtr xray_tracer = factory.createHttpTracer(*message, context);
  ASSERT_NE(nullptr, xray_tracer);
}

TEST(XRayTracerConfigTest, XRayHttpTracerWithInvalidFileName) {
  NiceMock<Server::Configuration::MockTracerFactoryContext> context;
  NiceMock<Api::MockApi> api;
  NiceMock<Filesystem::MockInstance> file_system;

  // fake invalid file
  EXPECT_CALL(file_system, fileReadToEnd("rules.json"))
      .WillRepeatedly(Throw(EnvoyException("failed to open file.")));
  EXPECT_CALL(api, fileSystem()).WillRepeatedly(ReturnRef(file_system));
  EXPECT_CALL(context.server_factory_context_, api()).WillRepeatedly(ReturnRef(api));

  const std::string yaml_string = R"EOF(
  http:
    name: xray
    typed_config:
      "@type": type.googleapis.com/envoy.config.trace.v3.XRayConfig
      daemon_endpoint:
        protocol: UDP
        address: 127.0.0.1
        port_value: 2000
      segment_name: AwsAppMesh
      sampling_rule_manifest:
        filename: "rules.json")EOF";

  envoy::config::trace::v3::Tracing configuration;
  TestUtility::loadFromYaml(yaml_string, configuration);

  XRayTracerFactory factory;
  auto message = Config::Utility::translateToFactoryConfig(
      configuration.http(), ProtobufMessage::getStrictValidationVisitor(), factory);

  Tracing::HttpTracerSharedPtr xray_tracer = factory.createHttpTracer(*message, context);
  ASSERT_NE(nullptr, xray_tracer);
}

TEST(XRayTracerConfigTest, ProtocolNotUDPThrows) {
  NiceMock<Server::Configuration::MockTracerFactoryContext> context;
  const std::string yaml_string = R"EOF(
  http:
    name: xray
    typed_config:
      "@type": type.googleapis.com/envoy.config.trace.v3.XRayConfig
      daemon_endpoint:
        protocol: TCP
        address: 127.0.0.1
        port_value: 2000
      segment_name: AwsAppMesh
      sampling_rule_manifest:
        filename: "rules.json")EOF";

  envoy::config::trace::v3::Tracing configuration;
  TestUtility::loadFromYaml(yaml_string, configuration);

  XRayTracerFactory factory;
  auto message = Config::Utility::translateToFactoryConfig(
      configuration.http(), ProtobufMessage::getStrictValidationVisitor(), factory);

  ASSERT_THROW(factory.createHttpTracer(*message, context), EnvoyException);
}

TEST(XRayTracerConfigTest, UsingNamedPortThrows) {
  NiceMock<Server::Configuration::MockTracerFactoryContext> context;
  const std::string yaml_string = R"EOF(
  http:
    name: xray
    typed_config:
      "@type": type.googleapis.com/envoy.config.trace.v3.XRayConfig
      daemon_endpoint:
        protocol: UDP
        address: 127.0.0.1
        named_port: SMTP
      segment_name: AwsAppMesh
      sampling_rule_manifest:
        filename: "rules.json")EOF";

  envoy::config::trace::v3::Tracing configuration;
  TestUtility::loadFromYaml(yaml_string, configuration);

  XRayTracerFactory factory;
  auto message = Config::Utility::translateToFactoryConfig(
      configuration.http(), ProtobufMessage::getStrictValidationVisitor(), factory);

  ASSERT_THROW(factory.createHttpTracer(*message, context), EnvoyException);
}

TEST(XRayTracerConfigTest, XRayHttpTracerWithSegmentFieldsTypedConfig) {
  NiceMock<Server::Configuration::MockTracerFactoryContext> context;

  const std::string yaml_string = R"EOF(
    http:
      name: xray
      typed_config:
        "@type": type.googleapis.com/envoy.config.trace.v3.XRayConfig
        daemon_endpoint:
          protocol: UDP
          address: 127.0.0.1
          port_value: 2000
        segment_name: AwsAppMesh
        sampling_rule_manifest:
          filename: "rules.json"
        segment_fields:
          origin: AWS::Origin::Name
          aws:
            key: value
            list:
              - test
              - test
    )EOF";

  envoy::config::trace::v3::Tracing configuration;
  TestUtility::loadFromYaml(yaml_string, configuration);

  XRayTracerFactory factory;
  auto message = Config::Utility::translateToFactoryConfig(
      configuration.http(), ProtobufMessage::getStrictValidationVisitor(), factory);
  Tracing::HttpTracerSharedPtr xray_tracer = factory.createHttpTracer(*message, context);
  ASSERT_NE(nullptr, xray_tracer);
}

} // namespace
} // namespace XRay
} // namespace Tracers
} // namespace Extensions
} // namespace Envoy