diff options
Diffstat (limited to 'src/jaegertracing/opentelemetry-cpp/sdk/test/trace/always_on_sampler_test.cc')
-rw-r--r-- | src/jaegertracing/opentelemetry-cpp/sdk/test/trace/always_on_sampler_test.cc | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/src/jaegertracing/opentelemetry-cpp/sdk/test/trace/always_on_sampler_test.cc b/src/jaegertracing/opentelemetry-cpp/sdk/test/trace/always_on_sampler_test.cc new file mode 100644 index 000000000..c483f9b8b --- /dev/null +++ b/src/jaegertracing/opentelemetry-cpp/sdk/test/trace/always_on_sampler_test.cc @@ -0,0 +1,60 @@ +// Copyright The OpenTelemetry Authors +// SPDX-License-Identifier: Apache-2.0 + +#include "opentelemetry/nostd/span.h" +#include "opentelemetry/sdk/trace/samplers/always_on.h" +#include "opentelemetry/trace/span_context_kv_iterable_view.h" + +#include <gtest/gtest.h> +#include <map> + +using namespace opentelemetry::sdk::trace; +using namespace opentelemetry::nostd; +using opentelemetry::trace::SpanContext; +namespace trace_api = opentelemetry::trace; + +TEST(AlwaysOnSampler, ShouldSample) +{ + AlwaysOnSampler sampler; + + // A buffer of trace_id with no specific meaning + constexpr uint8_t buf[] = {0, 1, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4, 5, 6, 7}; + + trace_api::TraceId trace_id_invalid; + trace_api::TraceId trace_id_valid(buf); + std::map<std::string, int> key_value_container = {{"key", 0}}; + + using L = std::vector<std::pair<trace_api::SpanContext, std::map<std::string, std::string>>>; + L l1 = {{trace_api::SpanContext(false, false), {}}, {trace_api::SpanContext(false, false), {}}}; + + opentelemetry::trace::SpanContextKeyValueIterableView<L> links{l1}; + + // Test with invalid (empty) trace id and empty parent context + auto sampling_result = sampler.ShouldSample( + SpanContext::GetInvalid(), trace_id_invalid, "invalid trace id test", + trace_api::SpanKind::kServer, + opentelemetry::common::KeyValueIterableView<std::map<std::string, int>>(key_value_container), + links); + + ASSERT_EQ(Decision::RECORD_AND_SAMPLE, sampling_result.decision); + ASSERT_EQ(nullptr, sampling_result.attributes); + ASSERT_EQ("", sampling_result.trace_state->ToHeader()); + + // Test with a valid trace id and empty parent context + sampling_result = sampler.ShouldSample( + SpanContext::GetInvalid(), trace_id_valid, "valid trace id test", + trace_api::SpanKind::kServer, + opentelemetry::common::KeyValueIterableView<std::map<std::string, int>>(key_value_container), + links); + + ASSERT_EQ(Decision::RECORD_AND_SAMPLE, sampling_result.decision); + ASSERT_EQ(nullptr, sampling_result.attributes); + ASSERT_EQ("", sampling_result.trace_state->ToHeader()); +} + +TEST(AlwaysOnSampler, GetDescription) +{ + AlwaysOnSampler sampler; + + ASSERT_EQ("AlwaysOnSampler", sampler.GetDescription()); +} |