diff --git a/phileas-core/src/main/java/ai/philterd/phileas/services/FilterPolicyLoader.java b/phileas-core/src/main/java/ai/philterd/phileas/services/FilterPolicyLoader.java index 15c3249b4..ae1bc2b01 100644 --- a/phileas-core/src/main/java/ai/philterd/phileas/services/FilterPolicyLoader.java +++ b/phileas-core/src/main/java/ai/philterd/phileas/services/FilterPolicyLoader.java @@ -783,21 +783,21 @@ public List getFiltersForPolicy(final Policy policy, final Map getFiltersForPolicy(final Policy policy, final Map spans) { } + @Test + void buildPolicyServiceMemory() throws IOException { + final var properties = new Properties(); + properties.setProperty("filter.policies.service", "memory"); + final var config = new PhileasConfiguration(properties); + final var service = new PhileasFilterService(config); + + Assertions.assertInstanceOf(InMemoryPolicyService.class, service.getPolicyService()); + } + + @Test + void buildPolicyServiceLocal() throws IOException { + final var properties = new Properties(); + properties.setProperty("filter.policies.service", "local"); + final var config = new PhileasConfiguration(properties); + final var service = new PhileasFilterService(config); + + Assertions.assertInstanceOf(LocalPolicyService.class, service.getPolicyService()); + } } \ No newline at end of file