ICS sample producer and consumer

[nonrtric.git] / sample-services / ics-producer-consumer / consumer / src / test / java / com / demo / consumer / SimpleConsumerTest.java

diff --git a/sample-services/ics-producer-consumer/consumer/src/test/java/com/demo/consumer/SimpleConsumerTest.java b/sample-services/ics-producer-consumer/consumer/src/test/java/com/demo/consumer/SimpleConsumerTest.java
new file mode 100644 (file)
index 0000000..7a7c279
--- /dev/null
+++ b/sample-services/ics-producer-consumer/consumer/src/test/java/com/demo/consumer/SimpleConsumerTest.java
@@ -0,0 +1,86 @@
+/*-
+ * ========================LICENSE_START=================================
+ * O-RAN-SC
+ *
+ * Copyright (C) 2024: OpenInfra Foundation Europe
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ========================LICENSE_END===================================
+ */
+
+package com.demo.consumer;
+
+import com.demo.consumer.consumer.SimpleConsumer;
+import com.demo.consumer.messages.KafkaMessageHandler;
+import com.demo.consumer.messages.PropertiesHelper;
+
+import org.apache.kafka.clients.consumer.KafkaConsumer;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+
+import java.time.Duration;
+
+import java.util.Properties;
+
+import static org.mockito.Mockito.*;
+
+class SimpleConsumerTest {
+
+    private static final long TIME_OUT_MS = 1000;
+
+    @Mock
+    private KafkaConsumer<String, String> kafkaConsumer;
+
+    @InjectMocks
+    private SimpleConsumer simpleConsumer;
+
+    private AutoCloseable closable;
+
+    @BeforeEach
+    void setUp() throws Exception {
+        closable = MockitoAnnotations.openMocks(this);
+    }
+
+    @AfterEach
+    void tearDown() throws Exception {
+        closable.close();
+    }
+
+    @Test
+    void testRun() throws Exception {
+        // Mocking the properties object returned by PropertiesHelper.getProperties()
+        Properties properties = new Properties();
+        properties.setProperty("bootstrap.servers", "localhost:9092");
+        properties.setProperty("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
+        properties.setProperty("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
+
+        // Mocking PropertiesHelper.getProperties() to return the mocked Properties object
+        try (MockedStatic<PropertiesHelper> propertiesHelperMockedStatic = Mockito.mockStatic(PropertiesHelper.class)) {
+            propertiesHelperMockedStatic.when(PropertiesHelper::getProperties).thenReturn(properties);
+
+            String topicName = "testTopic";
+            int numberOfMessages = 10;
+            KafkaMessageHandler callback = mock(KafkaMessageHandler.class);
+
+            simpleConsumer.run(topicName, numberOfMessages, callback);
+            verify(kafkaConsumer, times(0)).poll(Duration.ofMillis(TIME_OUT_MS));
+        }
+    }
+
+}