This is an automated email from the ASF dual-hosted git repository. ppkarwasz pushed a commit to branch fix/build-project in repository https://gitbox.apache.org/repos/asf/logging-flume-mongodb.git
commit 5ba7e472b1d4d988812352bf0530c51c7938bef5 Author: Piotr P. Karwasz <[email protected]> AuthorDate: Wed Sep 2 17:49:34 2026 +0200 Remove dependency on old `mockito-all` --- .../apache/flume/sink/mongodb/TestMongoDbSink.java | 28 ++++++++++++++++++---- 1 file changed, 24 insertions(+), 4 deletions(-) diff --git a/flume-mongodb-sink/src/test/java/org/apache/flume/sink/mongodb/TestMongoDbSink.java b/flume-mongodb-sink/src/test/java/org/apache/flume/sink/mongodb/TestMongoDbSink.java index a2d2846..3dc4de0 100644 --- a/flume-mongodb-sink/src/test/java/org/apache/flume/sink/mongodb/TestMongoDbSink.java +++ b/flume-mongodb-sink/src/test/java/org/apache/flume/sink/mongodb/TestMongoDbSink.java @@ -21,6 +21,7 @@ import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; import static org.junit.Assert.fail; +import java.lang.reflect.Field; import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.HashMap; @@ -39,7 +40,6 @@ import org.apache.flume.event.EventBuilder; import org.apache.flume.instrumentation.SinkCounter; import org.bson.Document; import org.junit.Test; -import org.mockito.internal.util.reflection.Whitebox; public class TestMongoDbSink { @@ -70,6 +70,26 @@ public class TestMongoDbSink { return context; } + private static void setInternalState(Object target, String fieldName, Object value) { + try { + Field field = target.getClass().getDeclaredField(fieldName); + field.setAccessible(true); + field.set(target, value); + } catch (ReflectiveOperationException e) { + throw new IllegalStateException("Unable to set field " + fieldName, e); + } + } + + private static Object getInternalState(Object target, String fieldName) { + try { + Field field = target.getClass().getDeclaredField(fieldName); + field.setAccessible(true); + return field.get(target); + } catch (ReflectiveOperationException e) { + throw new IllegalStateException("Unable to read field " + fieldName, e); + } + } + private static MongoDbSink createSink(Context context, MongoDbWriter writer) { MongoDbSink sink = new MongoDbSink(); Channel channel = new MemoryChannel(); @@ -77,8 +97,8 @@ public class TestMongoDbSink { sink.setChannel(channel); channel.start(); Configurables.configure(sink, context); - Whitebox.setInternalState(sink, "writer", writer); - Whitebox.setInternalState(sink, "counter", new SinkCounter("test")); + setInternalState(sink, "writer", writer); + setInternalState(sink, "counter", new SinkCounter("test")); return sink; } @@ -254,7 +274,7 @@ public class TestMongoDbSink { sink.configure(context); @SuppressWarnings("unchecked") - Map<String, String> collectionMap = (Map<String, String>) Whitebox.getInternalState(sink, "collectionMap"); + Map<String, String> collectionMap = (Map<String, String>) getInternalState(sink, "collectionMap"); assertEquals("collectionA", collectionMap.get("typeA")); assertEquals("collectionB", collectionMap.get("typeB")); }
