add RequestDeserialized event

DataDog · Nov 18, 2019 · d7d7b83 · d7d7b83
1 parent 7d67160
commit d7d7b83
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/Datadog.Trace.TestHelpers/MockTracerAgent.cs b/test/Datadog.Trace.TestHelpers/MockTracerAgent.cs
@@ -56,6 +56,8 @@ public MockTracerAgent(int port = 8126, int retries = 5)
 
         public event EventHandler<EventArgs<HttpListenerContext>> RequestReceived;
 
+        public event EventHandler<EventArgs<IList<IList<Span>>>> RequestDeserialized;
+
         /// <summary>
         /// Gets or sets a value indicating whether to skip serialization of traces.
         /// </summary>
@@ -152,6 +154,11 @@ protected virtual void OnRequestReceived(HttpListenerContext context)
             RequestReceived?.Invoke(this, new EventArgs<HttpListenerContext>(context));
         }
 
+        protected virtual void OnRequestDeserialized(IList<IList<Span>> traces)
+        {
+            RequestDeserialized?.Invoke(this, new EventArgs<IList<IList<Span>>>(traces));
+        }
+
         private void AssertHeader(
             NameValueCollection headers,
             string headerKey,
@@ -182,6 +189,7 @@ private void HandleHttpRequests()
                     if (ShouldDeserializeTraces)
                     {
                         var spans = MessagePackSerializer.Deserialize<IList<IList<Span>>>(ctx.Request.InputStream);
+                        OnRequestDeserialized(spans);
 
                         lock (this)
                         {