[Exporter.Geneva] Add test case for multithreading scenarios (#549)

open-telemetry · Jul 30, 2022 · c062c5a · c062c5a
1 parent e0fc583
commit c062c5a
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/test/OpenTelemetry.Exporter.Geneva.Tests/GenevaLogExporterTests.cs b/test/OpenTelemetry.Exporter.Geneva.Tests/GenevaLogExporterTests.cs
@@ -24,6 +24,7 @@
 using System.Runtime.CompilerServices;
 using System.Runtime.InteropServices;
 using System.Threading;
+using System.Threading.Tasks;
 using Microsoft.Extensions.Logging;
 using OpenTelemetry.Logs;
 using Xunit;
@@ -871,6 +872,23 @@ public void SuccessfulExportOnLinux()
 
                     // Validation
                     Assert.Equal(messagePackDataSize, receivedDataSize);
+
+                    logRecordList.Clear();
+
+                    // Emit log on a different thread to test for multithreading scenarios
+                    var thread = new Thread(() =>
+                    {
+                        logger.LogInformation("Hello from another thread {food} {price}.", "artichoke", 3.99);
+                    });
+                    thread.Start();
+                    thread.Join();
+
+                    // logRecordList should have a singleLogRecord entry after the logger.LogInformation call
+                    Assert.Single(logRecordList);
+
+                    messagePackDataSize = exporter.SerializeLogRecord(logRecordList[0]);
+                    receivedDataSize = serverSocket.Receive(receivedData);
+                    Assert.Equal(messagePackDataSize, receivedDataSize);
                 }
                 finally
                 {

diff --git a/test/OpenTelemetry.Exporter.Geneva.Tests/GenevaTraceExporterTests.cs b/test/OpenTelemetry.Exporter.Geneva.Tests/GenevaTraceExporterTests.cs
@@ -24,6 +24,7 @@
 using System.Runtime.CompilerServices;
 using System.Runtime.InteropServices;
 using System.Threading;
+using System.Threading.Tasks;
 using OpenTelemetry.Trace;
 using Xunit;
 
@@ -414,6 +415,7 @@ public void GenevaTraceExporter_Success_Linux()
                         })
                         .Build();
                     using Socket serverSocket = server.Accept();
+                    serverSocket.ReceiveTimeout = 10000;
 
                     // Create a test exporter to get MessagePack byte data for validation of the data received via Socket.
                     var exporter = new GenevaTraceExporter(new GenevaExporterOptions
@@ -429,7 +431,8 @@ public void GenevaTraceExporter_Success_Linux()
 
                     // Emit trace and grab a copy of internal buffer for validation.
                     var source = new ActivitySource(sourceName);
-                    int messagePackDataSize;
+                    int messagePackDataSize = 0;
+
                     using (var activity = source.StartActivity("Foo", ActivityKind.Internal))
                     {
                         messagePackDataSize = exporter.SerializeActivity(activity);
@@ -441,6 +444,20 @@ public void GenevaTraceExporter_Success_Linux()
 
                     // Validation
                     Assert.Equal(messagePackDataSize, receivedDataSize);
+
+                    // Create activity on a different thread to test for multithreading scenarios
+                    var thread = new Thread(() =>
+                    {
+                        using (var activity = source.StartActivity("ActivityFromAnotherThread", ActivityKind.Internal))
+                        {
+                            messagePackDataSize = exporter.SerializeActivity(activity);
+                        }
+                    });
+                    thread.Start();
+                    thread.Join();
+
+                    receivedDataSize = serverSocket.Receive(receivedData);
+                    Assert.Equal(messagePackDataSize, receivedDataSize);
                 }
                 catch (Exception)
                 {