Fix assertions in LoggerMessageFactoryCustomizationTest

apache · Dec 9, 2024 · 8ee4237 · 8ee4237
1 parent 18a1deb
commit 8ee4237
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...st/src/test/java/org/apache/logging/log4j/core/LoggerMessageFactoryCustomizationTest.java b/...st/src/test/java/org/apache/logging/log4j/core/LoggerMessageFactoryCustomizationTest.java
@@ -60,8 +60,8 @@ private static void assertTestMessageFactories(
             Logger logger,
             Class<? extends MessageFactory> messageFactoryClass,
             Class<? extends FlowMessageFactory> flowMessageFactoryClass) {
-        assertThat((MessageFactory) logger.getMessageFactory()).isInstanceOf(messageFactoryClass);
-        assertThat(logger.getFlowMessageFactory()).isInstanceOf(flowMessageFactoryClass);
+        assertThat(logger.getMessageFactory().getClass()).isEqualTo(messageFactoryClass);
+        assertThat(logger.getFlowMessageFactory().getClass()).isEqualTo(flowMessageFactoryClass);
     }
 
     public static class TestMessageFactory extends AbstractMessageFactory {