1 files changed, 7 insertions, 11 deletions
diff --git a/RhSolutions.Api.Tests/ProductQueryModifierTests.cs b/RhSolutions.Api.Tests/ProductQueryModifierTests.cs
index 16c6054..a642f9c 100644
--- a/RhSolutions.Api.Tests/ProductQueryModifierTests.cs
+++ b/RhSolutions.Api.Tests/ProductQueryModifierTests.cs
@@ -1,18 +1,14 @@
+using Microsoft.Extensions.DependencyInjection;
+
 namespace RhSolutions.Api.Tests;
 
 public abstract class ProductQueryModifierTests
 {
-	protected ProductQueryModifierFactory _factory;
-
-	[SetUp]
-	public void Setup()
-	{
-		_factory = new ProductQueryModifierFactory();
-	}
-	public void Execute(string productType, string query, string expected)
+	public void Invoke(string productType, string query, string expected)
 	{
-		var modifier = _factory.GetModifier(productType);		
-		Assert.That(modifier.TryQueryModify(query, out var actual), Is.True);
+		var modifier = TestServiceCollection.ServiceProvider?.GetRequiredKeyedService<IProductMLModifier>(productType);
+		string actual = string.Empty;
+		Assert.That(modifier?.TryQueryModify(query, out actual), Is.True);
 		Assert.That(actual, Is.EqualTo(expected));
 	}
-}
-\ No newline at end of file
+}