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
|
// 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());
}
|