VOL-2710: Add unit test cases for kafka-onos
Change-Id: Ieff0a32b3552eb8843fa933e57fa4148cb54853b
diff --git a/src/test/java/org/opencord/kafka/integrations/AaaKafkaIntegrationTest.java b/src/test/java/org/opencord/kafka/integrations/AaaKafkaIntegrationTest.java
new file mode 100644
index 0000000..f6fabde
--- /dev/null
+++ b/src/test/java/org/opencord/kafka/integrations/AaaKafkaIntegrationTest.java
@@ -0,0 +1,331 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+ *
+ * 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.
+ */
+
+package org.opencord.kafka.integrations;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.onlab.packet.MacAddress;
+import org.onlab.packet.RADIUS;
+import org.opencord.aaa.AaaMachineStatisticsDelegate;
+import org.opencord.aaa.AaaMachineStatisticsEventListener;
+import org.opencord.aaa.AaaMachineStatisticsService;
+import org.opencord.aaa.AaaStatistics;
+import org.opencord.aaa.AaaStatisticsSnapshot;
+import org.opencord.aaa.AaaSupplicantMachineStats;
+import org.opencord.aaa.AuthenticationEventListener;
+import org.opencord.aaa.AuthenticationRecord;
+import org.opencord.aaa.AuthenticationService;
+import org.opencord.aaa.AuthenticationStatisticsEventListener;
+import org.opencord.aaa.AuthenticationStatisticsService;
+import org.opencord.aaa.RadiusCommunicator;
+import org.opencord.aaa.RadiusOperationalStatusEventDelegate;
+import org.opencord.aaa.RadiusOperationalStatusEventListener;
+import org.opencord.aaa.RadiusOperationalStatusService;
+import org.opencord.kafka.EventBusService;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * set of unit test cases for AaaKafkaIntegration.
+ */
+class AaaKafkaIntegrationTest extends KafkaIntegrationTestBase {
+
+
+ private AaaKafkaIntegration aaaKafkaInt;
+ private AuthenticationEventListener authEventListener;
+ private AuthenticationStatisticsEventListener authStatisticsEventListener;
+ private RadiusOperationalStatusEventListener radiusOperStatusEventListener;
+ private AaaMachineStatisticsEventListener aaaMachineStatisticsEventListener;
+
+ @BeforeEach
+ void setUp() {
+ aaaKafkaInt = new AaaKafkaIntegration();
+
+ aaaKafkaInt.deviceService = new MockDeviceService();
+ aaaKafkaInt.eventBusService = new MockEventBusService();
+ aaaKafkaInt.ignore = new MockAuthenticationService();
+ aaaKafkaInt.ignore2 = new MockAuthenticationStatisticsService();
+ aaaKafkaInt.ignore3 = new MockRadiusOperationalStatusService();
+ aaaKafkaInt.ignore4 = new MockAaaMachineStatisticsService();
+ aaaKafkaInt.bindAuthenticationService(aaaKafkaInt.ignore);
+ aaaKafkaInt.bindAuthenticationStatService(aaaKafkaInt.ignore2);
+ aaaKafkaInt.bindRadiusOperationalStatusService(aaaKafkaInt.ignore3);
+ aaaKafkaInt.bindAaaMachineStatisticsService(aaaKafkaInt.ignore4);
+ aaaKafkaInt.activate();
+ }
+
+ @AfterEach
+ void tearDown() {
+ aaaKafkaInt.deactivate();
+ aaaKafkaInt.unbindRadiusOperationalStatusService(aaaKafkaInt.ignore3);
+ aaaKafkaInt.unbindAaaMachineStatisticsService(aaaKafkaInt.ignore4);
+ aaaKafkaInt = null;
+ }
+
+ /**
+ * testAuthenticationEvent to perform unit test for
+ * the AuthenticationEvent event.
+ */
+ @Test
+ void testAuthenticationEvent() {
+ authEventListener.event(getAuthenticationEvent());
+ assertEquals(MockEventBusService.authCounter, 1);
+ assertEquals(MockEventBusService.otherCounter, 0);
+ }
+
+ /**
+ * testAuthenticationStatisticsEvent to perform unit test for
+ * the AuthenticationStatisticsEvent event.
+ */
+ @Test
+ void testAuthenticationStatisticsEvent() {
+ authStatisticsEventListener.event(getAuthenticationStatisticsEvent());
+ assertEquals(MockEventBusService.authStatsCounter, 1);
+ assertEquals(MockEventBusService.otherCounter, 0);
+ }
+
+ /**
+ * testRadiusOperationalStatusEvent to perform unit test for
+ * the RadiusOperationalStatusEvent event.
+ */
+ @Test
+ void testRadiusOperationalStatusEvent() {
+ radiusOperStatusEventListener.event(getRadiusOperationalStatusEvent());
+ assertEquals(MockEventBusService.radiusOperstate, 1);
+ assertEquals(MockEventBusService.otherCounter, 0);
+ }
+
+ /**
+ * testAaaMachineStatisticsEvent to perform unit test for
+ * the AaaMachineStatisticsEvent event.
+ */
+ @Test
+ void testAaaMachineStatisticsEvent() {
+ aaaMachineStatisticsEventListener.event(getAaaMachineStatisticsEvent());
+ assertEquals(MockEventBusService.authStatsCounter, 1);
+ assertEquals(MockEventBusService.otherCounter, 0);
+ }
+
+ /**
+ * EventBusService mocker class.
+ */
+ private static class MockEventBusService implements EventBusService {
+ static int authCounter;
+ static int authStatsCounter;
+ static int radiusOperstate;
+ static int otherCounter;
+
+ MockEventBusService() {
+ authCounter = 0;
+ authStatsCounter = 0;
+ radiusOperstate = 0;
+ otherCounter = 0;
+ }
+
+ @Override
+ public void send(String topic, JsonNode data) {
+ switch (topic) {
+ case AaaKafkaIntegration.TOPIC:
+ authCounter++;
+ break;
+ case AaaKafkaIntegration.AUTHENTICATION_STATISTICS_TOPIC:
+ authStatsCounter++;
+ break;
+ case AaaKafkaIntegration.RADIUS_OPERATION_STATUS_TOPIC:
+ radiusOperstate++;
+ break;
+ default:
+ otherCounter++;
+ break;
+ }
+ }
+ }
+
+ /**
+ * AuthenticationService mocker class.
+ */
+ private class MockAuthenticationService implements AuthenticationService {
+ @Override
+ public Iterable<AuthenticationRecord> getAuthenticationRecords() {
+ return null;
+ }
+
+ @Override
+ public boolean removeAuthenticationStateByMac(MacAddress macAddress) {
+ return false;
+ }
+
+ @Override
+ public AaaSupplicantMachineStats getSupplicantMachineStats(String s) {
+ return null;
+ }
+
+ @Override
+ public void addListener(AuthenticationEventListener listener) {
+ authEventListener = listener;
+ }
+
+ @Override
+ public void removeListener(AuthenticationEventListener listener) {
+ authEventListener = null;
+ }
+ }
+
+ /**
+ * AuthenticationStatisticsService mocker class.
+ */
+ private class MockAuthenticationStatisticsService implements AuthenticationStatisticsService {
+ @Override
+ public AaaStatistics getAaaStats() {
+ return null;
+ }
+
+ @Override
+ public AaaStatisticsSnapshot getClusterStatistics() {
+ return null;
+ }
+
+ @Override
+ public void handleRoundtripTime(byte b) {
+
+ }
+
+ @Override
+ public void calculatePacketRoundtripTime() {
+
+ }
+
+ @Override
+ public void putOutgoingIdentifierToMap(byte b) {
+
+ }
+
+ @Override
+ public void resetAllCounters() {
+
+ }
+
+ @Override
+ public void addListener(AuthenticationStatisticsEventListener listener) {
+ authStatisticsEventListener = listener;
+ }
+
+ @Override
+ public void removeListener(AuthenticationStatisticsEventListener listener) {
+ authStatisticsEventListener = null;
+ }
+ }
+
+ /**
+ * RadiusOperationalStatusService mocker class.
+ */
+ private class MockRadiusOperationalStatusService implements RadiusOperationalStatusService {
+ @Override
+ public RadiusOperationalStatusEventDelegate getRadiusOprStDelegate() {
+ return null;
+ }
+
+ @Override
+ public String getRadiusServerOperationalStatus() {
+ return null;
+ }
+
+ @Override
+ public void setStatusServerReqSent(boolean b) {
+
+ }
+
+ @Override
+ public void setRadiusOperationalStatusEvaluationMode(
+ RadiusOperationalStatusService.RadiusOperationalStatusEvaluationMode
+ radiusOperationalStatusEvaluationMode) {
+
+ }
+
+ @Override
+ public void setOperationalStatusServerTimeoutInMillis(long l) {
+
+ }
+
+ @Override
+ public void checkServerOperationalStatus() {
+
+ }
+
+ @Override
+ public boolean isRadiusResponseForOperationalStatus(byte b) {
+ return false;
+ }
+
+ @Override
+ public void handleRadiusPacketForOperationalStatus(RADIUS radius) {
+
+ }
+
+ @Override
+ public void initialize(byte[] bytes, String s, RadiusCommunicator radiusCommunicator) {
+
+ }
+
+ @Override
+ public void setOutTimeInMillis(byte b) {
+
+ }
+
+ @Override
+ public void addListener(RadiusOperationalStatusEventListener listener) {
+ radiusOperStatusEventListener = listener;
+ }
+
+ @Override
+ public void removeListener(RadiusOperationalStatusEventListener listener) {
+ radiusOperStatusEventListener = null;
+ }
+ }
+
+ /**
+ * AaaMachineStatisticsService mocker class.
+ */
+ private class MockAaaMachineStatisticsService implements AaaMachineStatisticsService {
+ @Override
+ public AaaSupplicantMachineStats getSupplicantStats(Object o) {
+ return null;
+ }
+
+ @Override
+ public AaaMachineStatisticsDelegate getMachineStatsDelegate() {
+ return null;
+ }
+
+ @Override
+ public void logAaaSupplicantMachineStats(AaaSupplicantMachineStats aaaSupplicantMachineStats) {
+
+ }
+
+ @Override
+ public void addListener(AaaMachineStatisticsEventListener listener) {
+ aaaMachineStatisticsEventListener = listener;
+ }
+
+ @Override
+ public void removeListener(AaaMachineStatisticsEventListener listener) {
+ aaaMachineStatisticsEventListener = null;
+ }
+ }
+}
\ No newline at end of file