1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
#include <gtest/gtest.h>
#include <stdlib.h>
#include "opentelemetry/nostd/shared_ptr.h"
#include "opentelemetry/nostd/string_view.h"
#include "opentelemetry/sdk/trace/provider.h"
#include "opentelemetry/trace/provider.h"
#include "opentelemetry/trace/tracer.h"
#include "opentelemetry/trace/tracer_provider.h"
#if defined(_MSC_VER)
# include "opentelemetry/sdk/common/env_variables.h"
using opentelemetry::sdk::common::setenv;
using opentelemetry::sdk::common::unsetenv;
#endif
#if OPENTELEMETRY_ABI_VERSION_NO >= 2
# include "opentelemetry/common/key_value_iterable.h"
#endif
using opentelemetry::trace::Tracer;
using opentelemetry::trace::TracerProvider;
namespace nostd = opentelemetry::nostd;
namespace trace_api = opentelemetry::trace;
namespace trace_sdk = opentelemetry::sdk::trace;
class TestProvider : public TracerProvider
{
#if OPENTELEMETRY_ABI_VERSION_NO >= 2
nostd::shared_ptr<Tracer> GetTracer(
nostd::string_view /* name */,
nostd::string_view /* version */,
nostd::string_view /* schema_url */,
const opentelemetry::common::KeyValueIterable * /* attributes */) noexcept override
{
return nostd::shared_ptr<Tracer>(nullptr);
}
#else
nostd::shared_ptr<Tracer> GetTracer(nostd::string_view /* name */,
nostd::string_view /* version */,
nostd::string_view /* schema_url */) noexcept override
{
return nostd::shared_ptr<Tracer>(nullptr);
}
#endif
};
TEST(Provider, SetTracerProviderDefault)
{
#ifndef NO_GETENV
unsetenv("OTEL_SDK_DISABLED");
#endif
auto tf = nostd::shared_ptr<TracerProvider>(new TestProvider());
trace_sdk::Provider::SetTracerProvider(tf);
ASSERT_EQ(tf, trace_api::Provider::GetTracerProvider());
}
#ifndef NO_GETENV
TEST(Provider, SetTracerProviderEnabled)
{
setenv("OTEL_SDK_DISABLED", "false", 1);
auto tf = nostd::shared_ptr<TracerProvider>(new TestProvider());
trace_sdk::Provider::SetTracerProvider(tf);
ASSERT_EQ(tf, trace_api::Provider::GetTracerProvider());
unsetenv("OTEL_SDK_DISABLED");
}
TEST(Provider, SetTracerProviderDisabled)
{
setenv("OTEL_SDK_DISABLED", "true", 1);
auto tf = nostd::shared_ptr<TracerProvider>(new TestProvider());
trace_sdk::Provider::SetTracerProvider(tf);
ASSERT_NE(tf, trace_api::Provider::GetTracerProvider());
unsetenv("OTEL_SDK_DISABLED");
}
#endif
|