From 8e88dea41756c2005008c9f482c3429a3a0e0f4c Mon Sep 17 00:00:00 2001
From: Jakub Chocholowicz <jakub.chocholowicz@gmail.com>
Date: Tue, 19 May 2020 17:24:17 +0200
Subject: [PATCH] Merging v1

---
 .../Interfaces/IVsTestConsoleWrapper2.cs                       | 3 ++-
 .../VsTestConsoleWrapper.cs                                    | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
index 11d1980b53..10e4a7379a 100644
--- a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
+++ b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
@@ -18,6 +18,7 @@ public interface IVsTestConsoleWrapper2 : IVsTestConsoleWrapper
         /// </summary>
         /// <param name="attachments">List of attachements</param>
         /// <param name="testSessionEventsHandler">EventHandler to receive session complete event</param>
-        void OnTestSessionEnd(IEnumerable<AttachmentSet> attachments, ITestSessionEventsHandler testSessionEventsHandler);
+        void FinalizeMultiTestRuns(IEnumerable<AttachmentSet> attachments, ITestSessionEventsHandler testSessionEventsHandler);
+       // void FinalizeTests(IEnumerable<AttachmentSet> attachments, ITestSessionEventsHandler testSessionEventsHandler);
     }
 }
diff --git a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
index 378850eda0..48d44765d5 100644
--- a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
+++ b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
@@ -279,6 +279,7 @@ public void OnTestSessionEnd(IEnumerable<AttachmentSet> attachments, ITestSessio
             this.EnsureInitialized();
 
             this.requestSender.OnTestSessionEnd(attachments, testSessionEventsHandler);
+            this.requestSender.FinalizeSession(attachments, testSessionEventsHandler);
         }
 
         #endregion