Tests for new factory methods.

open-telemetry · Jan 28, 2023 · 3132e3b · 3132e3b
1 parent e6e51d9
commit 3132e3b
Show file tree

Hide file tree

Showing 2 changed files with 66 additions and 0 deletions.
diff --git a/test/OpenTelemetry.Tests/Metrics/MeterProviderBuilderExtensionsTests.cs b/test/OpenTelemetry.Tests/Metrics/MeterProviderBuilderExtensionsTests.cs
@@ -306,6 +306,28 @@ public void MeterProviderNestedResolutionUsingConfigureTest()
             Assert.True(innerTestExecuted);
         }
 
+        [Fact]
+        public void MeterProviderAddReaderFactoryTest()
+        {
+            bool factoryInvoked = false;
+
+            using var meterProvider = Sdk.CreateMeterProviderBuilder()
+                .AddReader(sp =>
+                {
+                    factoryInvoked = true;
+
+                    Assert.NotNull(sp);
+
+                    return new MyReader();
+                })
+                .Build() as MeterProviderSdk;
+
+            Assert.True(factoryInvoked);
+
+            Assert.NotNull(meterProvider);
+            Assert.True(meterProvider.Reader is MyReader);
+        }
+
         private static void RunBuilderServiceLifecycleTest(
             MeterProviderBuilder builder,
             Func<MeterProviderSdk> buildFunc,

diff --git a/test/OpenTelemetry.Tests/Trace/TracerProviderBuilderExtensionsTest.cs b/test/OpenTelemetry.Tests/Trace/TracerProviderBuilderExtensionsTest.cs
@@ -394,6 +394,50 @@ public void TracerProviderNestedResolutionUsingConfigureTest()
             Assert.True(innerTestExecuted);
         }
 
+        [Fact]
+        public void TracerProviderSetSamplerFactoryTest()
+        {
+            bool factoryInvoked = false;
+
+            using var tracerProvider = Sdk.CreateTracerProviderBuilder()
+                .SetSampler(sp =>
+                {
+                    factoryInvoked = true;
+
+                    Assert.NotNull(sp);
+
+                    return new MySampler();
+                })
+                .Build() as TracerProviderSdk;
+
+            Assert.True(factoryInvoked);
+
+            Assert.NotNull(tracerProvider);
+            Assert.True(tracerProvider.Sampler is MySampler);
+        }
+
+        [Fact]
+        public void TracerProviderAddProcessorFactoryTest()
+        {
+            bool factoryInvoked = false;
+
+            using var tracerProvider = Sdk.CreateTracerProviderBuilder()
+                .AddProcessor(sp =>
+                {
+                    factoryInvoked = true;
+
+                    Assert.NotNull(sp);
+
+                    return new MyProcessor();
+                })
+                .Build() as TracerProviderSdk;
+
+            Assert.True(factoryInvoked);
+
+            Assert.NotNull(tracerProvider);
+            Assert.True(tracerProvider.Processor is MyProcessor);
+        }
+
         private static void RunBuilderServiceLifecycleTest(
             TracerProviderBuilder builder,
             Func<TracerProviderSdk> buildFunc,