From 6980468356ec6544195318b3bc992b629de4b626 Mon Sep 17 00:00:00 2001
From: Stef Heyenrath <stef.heyenrath@gmail.com>
Date: Sat, 14 Feb 2026 13:06:26 +0100
Subject: [PATCH] T Unit

---
 src/WireMock.Net.ProtoBuf/Util/ProtoDefinitionDataHelper.cs | 4 ----
 test/WireMock.Net.TUnitTests/WireMock.Net.TUnitTests.csproj | 5 +++--
 test/WireMock.Net.TUnitTests/global.json                    | 5 +++++
 test/WireMock.Net.Tests/Grpc/ProtoDefinitionHelperTests.cs  | 5 +++--
 4 files changed, 11 insertions(+), 8 deletions(-)
 create mode 100644 test/WireMock.Net.TUnitTests/global.json

diff --git a/src/WireMock.Net.ProtoBuf/Util/ProtoDefinitionDataHelper.cs b/src/WireMock.Net.ProtoBuf/Util/ProtoDefinitionDataHelper.cs
index 666c260d..e147ba0b 100644
--- a/src/WireMock.Net.ProtoBuf/Util/ProtoDefinitionDataHelper.cs
+++ b/src/WireMock.Net.ProtoBuf/Util/ProtoDefinitionDataHelper.cs
@@ -1,9 +1,5 @@
 // Copyright © WireMock.Net
 
-using System.Collections.Generic;
-using System.IO;
-using System.Threading;
-using System.Threading.Tasks;
 using ProtoBufJsonConverter;
 using ProtoBufJsonConverter.Models;
 using Stef.Validation;
diff --git a/test/WireMock.Net.TUnitTests/WireMock.Net.TUnitTests.csproj b/test/WireMock.Net.TUnitTests/WireMock.Net.TUnitTests.csproj
index 2c6781d3..18c2582f 100644
--- a/test/WireMock.Net.TUnitTests/WireMock.Net.TUnitTests.csproj
+++ b/test/WireMock.Net.TUnitTests/WireMock.Net.TUnitTests.csproj
@@ -1,12 +1,13 @@
-<Project Sdk="Microsoft.NET.Sdk">
+<Project Sdk="Microsoft.NET.Sdk">
 
     <PropertyGroup>
         <TargetFramework>net8.0</TargetFramework>
+        <OutputType>Exe</OutputType>
         <ImplicitUsings>enable</ImplicitUsings>
         <Nullable>enable</Nullable>
         <IsTestProject>true</IsTestProject>
         <SonarQubeExclude>true</SonarQubeExclude>
-        <UseVSTest>false</UseVSTest>
+        <UseMicrosoftTestingPlatformRunner>true</UseMicrosoftTestingPlatformRunner>
     </PropertyGroup>
 
     <ItemGroup>
diff --git a/test/WireMock.Net.TUnitTests/global.json b/test/WireMock.Net.TUnitTests/global.json
new file mode 100644
index 00000000..8f73781c
--- /dev/null
+++ b/test/WireMock.Net.TUnitTests/global.json
@@ -0,0 +1,5 @@
+{
+    "test": {
+        "runner": "Microsoft.Testing.Platform"
+    }
+}
\ No newline at end of file
diff --git a/test/WireMock.Net.Tests/Grpc/ProtoDefinitionHelperTests.cs b/test/WireMock.Net.Tests/Grpc/ProtoDefinitionHelperTests.cs
index e77720cb..52e5e22e 100644
--- a/test/WireMock.Net.Tests/Grpc/ProtoDefinitionHelperTests.cs
+++ b/test/WireMock.Net.Tests/Grpc/ProtoDefinitionHelperTests.cs
@@ -13,12 +13,13 @@ public class ProtoDefinitionHelperTests
     public async Task FromDirectory_Greet_ShouldReturnModifiedProtoFiles()
     {
         // Arrange
+        var cancellationToken = TestContext.Current.CancellationToken;
         var directory = Path.Combine(Directory.GetCurrentDirectory(), "Grpc", "Test");
         var expectedFilename = "SubFolder/request.proto";
         var expectedComment = $"// {expectedFilename}";
 
         // Act
-        var protoDefinitionData = await ProtoDefinitionDataHelper.FromDirectory(directory);
+        var protoDefinitionData = await ProtoDefinitionDataHelper.FromDirectory(directory, cancellationToken);
         var protoDefinitions = protoDefinitionData.ToList("greet");
 
         // Assert
@@ -47,7 +48,7 @@ public class ProtoDefinitionHelperTests
         var directory = Path.Combine(Directory.GetCurrentDirectory(), "Grpc", "ot");
 
         // Act
-        var protoDefinitionData = await ProtoDefinitionDataHelper.FromDirectory(directory);
+        var protoDefinitionData = await ProtoDefinitionDataHelper.FromDirectory(directory, TestContext.Current.CancellationToken);
         var protoDefinitions = protoDefinitionData.ToList("trace_service");
 
         // Assert