@@ -103,7 +103,8 @@ TEST(OpenCensusTracerTest, Span) {
103103 registerSpanCatcher ();
104104 OpenCensusConfig oc_config;
105105 NiceMock<LocalInfo::MockLocalInfo> local_info;
106- std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info));
106+ Api::ApiPtr api = Api::createApiForTest ();
107+ std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info, *api));
107108
108109 NiceMock<Tracing::MockConfig> config;
109110 Http::TestHeaderMapImpl request_headers{
@@ -182,6 +183,7 @@ void testIncomingHeaders(
182183 registerSpanCatcher ();
183184 OpenCensusConfig oc_config;
184185 NiceMock<LocalInfo::MockLocalInfo> local_info;
186+ Api::ApiPtr api = Api::createApiForTest ();
185187 oc_config.add_incoming_trace_context (OpenCensusConfig::NONE);
186188 oc_config.add_incoming_trace_context (OpenCensusConfig::B3);
187189 oc_config.add_incoming_trace_context (OpenCensusConfig::TRACE_CONTEXT);
@@ -192,7 +194,7 @@ void testIncomingHeaders(
192194 oc_config.add_outgoing_trace_context (OpenCensusConfig::TRACE_CONTEXT);
193195 oc_config.add_outgoing_trace_context (OpenCensusConfig::GRPC_TRACE_BIN);
194196 oc_config.add_outgoing_trace_context (OpenCensusConfig::CLOUD_TRACE_CONTEXT);
195- std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info));
197+ std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info, *api ));
196198 NiceMock<Tracing::MockConfig> config;
197199 Http::TestHeaderMapImpl request_headers{
198200 {" :path" , " /" },
@@ -279,7 +281,8 @@ namespace {
279281int SamplerTestHelper (const OpenCensusConfig& oc_config) {
280282 registerSpanCatcher ();
281283 NiceMock<LocalInfo::MockLocalInfo> local_info;
282- std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info));
284+ Api::ApiPtr api = Api::createApiForTest ();
285+ std::unique_ptr<Tracing::Driver> driver (new OpenCensus::Driver (oc_config, local_info, *api));
283286 auto span = ::opencensus::trace::Span::StartSpan (" test_span" );
284287 span.End ();
285288 // Retrieve SpanData from the OpenCensus trace exporter.
0 commit comments