add RequestDeserialized event

DataDog · Sep 3, 2019 · a52ea35 · a52ea35
1 parent 09b6539
commit a52ea35
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>
@@ -148,6 +150,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,
@@ -178,6 +185,7 @@ private void HandleHttpRequests()
                     if (ShouldDeserializeTraces)
                     {
                         var spans = MessagePackSerializer.Deserialize<IList<IList<Span>>>(ctx.Request.InputStream);
+                        OnRequestDeserialized(spans);
 
                         lock (this)
                         {