diff options
Diffstat (limited to 'test/cpp/ext/otel/otel_test_library.cc')
| -rw-r--r-- | test/cpp/ext/otel/otel_test_library.cc | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/test/cpp/ext/otel/otel_test_library.cc b/test/cpp/ext/otel/otel_test_library.cc index 6f8e55a96d..19777d4d59 100644 --- a/test/cpp/ext/otel/otel_test_library.cc +++ b/test/cpp/ext/otel/otel_test_library.cc @@ -90,6 +90,9 @@ void OpenTelemetryPluginEnd2EndTest::Init( const std::map<std::string, std::string>& labels_to_inject, absl::AnyInvocable<bool(absl::string_view /*target*/) const> target_selector, + absl::AnyInvocable<bool(const grpc_core::ChannelArgs& /*channel_args*/) + const> + server_selector, absl::AnyInvocable<bool(absl::string_view /*target*/) const> target_attribute_filter, absl::AnyInvocable<bool(absl::string_view /*generic_method*/) const> @@ -121,6 +124,7 @@ void OpenTelemetryPluginEnd2EndTest::Init( } ot_builder.SetLabelsInjector(std::move(labels_injector)); ot_builder.SetTargetSelector(std::move(target_selector)); + ot_builder.SetServerSelector(std::move(server_selector)); ot_builder.SetTargetAttributeFilter(std::move(target_attribute_filter)); ot_builder.SetGenericMethodAttributeFilter( std::move(generic_method_attribute_filter)); @@ -160,8 +164,7 @@ void OpenTelemetryPluginEnd2EndTest::Init( void OpenTelemetryPluginEnd2EndTest::TearDown() { server_->Shutdown(); grpc_shutdown_blocking(); - delete grpc_core::ServerCallTracerFactory::Get(grpc_core::ChannelArgs()); - grpc_core::ServerCallTracerFactory::RegisterGlobal(nullptr); + grpc_core::ServerCallTracerFactory::TestOnlyReset(); } void OpenTelemetryPluginEnd2EndTest::ResetStub( |