[CCSDK-1241] Increase GRToolkit Unit Test Coverage

Add unit tests for GRToolkit. Slight changes to make
code testable.

Change-Id: Ib435da58b62e7b8edda4876f0f3a262cbc41a8ca
Issue-ID: CCSDK-1241
Signed-off-by: Haddox, Anthony <ah0647@att.com>

2 files changed, 385 insertions, 0 deletions

diff --git a/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/GrToolkitProviderTest.java b/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/GrToolkitProviderTest.java
new file mode 100644
index 00000000..dabdf206
--- /dev/null
+++ b/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/GrToolkitProviderTest.java
@@ -0,0 +1,344 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * openECOMP : SDN-C
+ * ================================================================================
+ * Copyright (C) 2018 AT&T Intellectual Property. All rights
+ * 			reserved.
+ * ================================================================================
+ * 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 org.onap.ccsdk.sli.plugins.grtoolkit;
+import com.google.common.util.concurrent.ListenableFuture;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.contrib.java.lang.system.EnvironmentVariables;
+import org.onap.ccsdk.sli.core.dblib.DBLibConnection;
+import org.onap.ccsdk.sli.core.dblib.DbLibService;
+import org.onap.ccsdk.sli.plugins.grtoolkit.data.ClusterActor;
+import org.opendaylight.controller.cluster.access.concepts.MemberName;
+import org.opendaylight.controller.cluster.datastore.DistributedDataStoreInterface;
+import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
+import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.AdminHealthOutput;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.ClusterHealthOutput;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.DatabaseHealthOutput;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.FailoverOutput;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.FailoverOutputBuilder;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.SiteHealthOutput;
+import org.opendaylight.yang.gen.v1.org.onap.ccsdk.sli.plugins.gr.toolkit.rev180926.SiteIdentifierOutput;
+import org.opendaylight.yangtools.yang.common.RpcResult;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.ExecutionException;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.when;
+
+public class GrToolkitProviderTest {
+    GrToolkitProvider provider;
+    GrToolkitProvider providerSpy;
+    DataBroker dataBroker;
+    NotificationPublishService notificationProviderService;
+    RpcProviderRegistry rpcProviderRegistry;
+    DistributedDataStoreInterface configDatastore;
+    DbLibService dbLibService;
+    DBLibConnection connection;
+
+    @Rule
+    public final EnvironmentVariables environmentVariables = new EnvironmentVariables();
+
+    @Before
+    public void setup() {
+        environmentVariables.set("SDNC_CONFIG_DIR","src/test/resources/");
+        dataBroker = mock(DataBroker.class);
+        notificationProviderService = mock(NotificationPublishService.class);
+        rpcProviderRegistry = mock(RpcProviderRegistry.class);
+        configDatastore = mock(DistributedDataStoreInterface.class);
+        dbLibService = mock(DbLibService.class);
+        connection = mock(DBLibConnection.class);
+
+        ActorContext actorContext = mock(ActorContext.class);
+        MemberName memberName = MemberName.forName("Test");
+
+        when(actorContext.getCurrentMemberName()).thenReturn(memberName);
+        when(configDatastore.getActorContext()).thenReturn(actorContext);
+
+        try {
+            when(connection.isReadOnly()).thenReturn(false);
+            when(connection.isClosed()).thenReturn(false);
+            when(dbLibService.isActive()).thenReturn(true);
+            when(dbLibService.getConnection()).thenReturn(connection);
+        } catch(SQLException e) {
+            fail();
+        }
+
+        provider = new GrToolkitProvider(dataBroker, notificationProviderService,
+                rpcProviderRegistry, configDatastore, dbLibService);
+        providerSpy = spy(provider);
+    }
+
+    @Test
+    public void closeTest() {
+        try {
+            provider.close();
+        }
+        catch(Exception e) {
+            // Exception expected
+        }
+    }
+
+    @Test
+    public void onDataTreeChangedTest() {
+        provider.onDataTreeChanged(new ArrayList());
+        // onDataTreeChanged is an empty stub
+    }
+
+    @Test
+    public void clusterHealthTest() {
+        ListenableFuture<RpcResult<ClusterHealthOutput>> result = provider.clusterHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void siteHealthTest() {
+        ListenableFuture<RpcResult<SiteHealthOutput>> result = provider.siteHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void siteHealth6NodeTest() {
+        Map<String, ClusterActor> memberMap = new HashMap<>();
+        ClusterActor actor;
+        for(int ndx = 0; ndx < 6; ndx++) {
+            actor = new ClusterActor();
+            actor.setNode("member-" + (ndx + 1));
+            actor.setUp(true);
+            actor.setUnreachable(false);
+
+            memberMap.put(actor.getNode(),  actor);
+        }
+
+        try {
+            Field field = provider.getClass().getDeclaredField("siteConfiguration");
+            field.setAccessible(true);
+            field.set(provider, GrToolkitProvider.SiteConfiguration.GEO);
+
+            field = provider.getClass().getDeclaredField("memberMap");
+            field.setAccessible(true);
+            field.set(provider, memberMap);
+
+
+            actor = new ClusterActor();
+            actor.setNode("member-1");
+            field = provider.getClass().getDeclaredField("self");
+            field.setAccessible(true);
+            field.set(provider, actor);
+
+            field = provider.getClass().getDeclaredField("member");
+            field.setAccessible(true);
+            field.set(provider, actor.getNode());
+        }
+        catch(IllegalAccessException | NoSuchFieldException e) {
+            fail();
+        }
+
+        ListenableFuture<RpcResult<SiteHealthOutput>> result = provider.siteHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void databaseHealthTest() {
+        ListenableFuture<RpcResult<DatabaseHealthOutput>> result = provider.databaseHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void databaseHealthWhenROTest() {
+        try {
+            when(connection.isReadOnly()).thenReturn(true);
+        } catch(SQLException e) {
+            fail();
+        }
+        ListenableFuture<RpcResult<DatabaseHealthOutput>> result = provider.databaseHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void databaseHealthWhenExceptionTest() {
+        try {
+            when(connection.isReadOnly()).thenThrow(new SQLException());
+        } catch(SQLException e) {
+            //expected
+        }
+        ListenableFuture<RpcResult<DatabaseHealthOutput>> result = provider.databaseHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void adminHealthTest() {
+        ListenableFuture<RpcResult<AdminHealthOutput>> result = provider.adminHealth(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void siteIdentifierTest() {
+        ListenableFuture<RpcResult<SiteIdentifierOutput>> result = provider.siteIdentifier(null);
+        try {
+            assertEquals("200", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void failoverTest() {
+        ListenableFuture<RpcResult<FailoverOutput>> result = provider.failover(null);
+        try {
+            assertEquals("400", result.get().getResult().getStatus());
+        } catch(InterruptedException | ExecutionException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void executeCommandTest() {
+        try {
+            Method method = provider.getClass().getDeclaredMethod("executeCommand", String.class);
+            method.setAccessible(true);
+            method.invoke(provider, "ls");
+        }
+        catch(NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void isolateSiteFromClusterTest() {
+        try {
+            ClusterActor actor = new ClusterActor();
+            actor.setNode("some-node");
+            actor.setAkkaPort("2550");
+            ArrayList<ClusterActor> activeList = new ArrayList<>();
+            activeList.add(actor);
+            ArrayList<ClusterActor> standbyList = new ArrayList<>();
+            standbyList.add(actor);
+            Method method = provider.getClass().getDeclaredMethod("isolateSiteFromCluster", ArrayList.class, ArrayList.class, String.class);
+            method.setAccessible(true);
+            method.invoke(provider, activeList, standbyList, "80");
+        }
+        catch(NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void downUnreachableNodesTest() {
+        try {
+            ClusterActor actor = new ClusterActor();
+            actor.setNode("some-node");
+            actor.setAkkaPort("2550");
+            ArrayList<ClusterActor> activeList = new ArrayList<>();
+            activeList.add(actor);
+            ArrayList<ClusterActor> standbyList = new ArrayList<>();
+            standbyList.add(actor);
+            Method method = provider.getClass().getDeclaredMethod("downUnreachableNodes", ArrayList.class, ArrayList.class, String.class);
+            method.setAccessible(true);
+            method.invoke(provider, activeList, standbyList, "80");
+        }
+        catch(NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void changeClusterVotingTest() {
+        try {
+            ClusterActor actor = new ClusterActor();
+            actor.setMember("some-member");
+            actor.setNode("some-Node");
+            ArrayList<ClusterActor> activeList = new ArrayList<>();
+            activeList.add(actor);
+            ArrayList<ClusterActor> standbyList = new ArrayList<>();
+            standbyList.add(actor);
+            Field field = provider.getClass().getDeclaredField("self");
+            field.setAccessible(true);
+            field.set(provider, actor);
+            Method method = provider.getClass().getDeclaredMethod("changeClusterVoting", FailoverOutputBuilder.class, ArrayList.class, ArrayList.class, String.class);
+            method.setAccessible(true);
+            method.invoke(provider, new FailoverOutputBuilder(), activeList, standbyList, "80");
+        }
+        catch(NoSuchMethodException | IllegalAccessException | InvocationTargetException | NoSuchFieldException e) {
+            fail();
+        }
+    }
+
+    @Test
+    public void backupMdSalTest() {
+        try {
+            ClusterActor actor = new ClusterActor();
+            actor.setNode("some-Node");
+            actor.setAkkaPort("2550");
+            ArrayList<ClusterActor> activeList = new ArrayList<>();
+            activeList.add(actor);
+            Method method = provider.getClass().getDeclaredMethod("backupMdSal", ArrayList.class, String.class);
+            method.setAccessible(true);
+            method.invoke(provider, activeList, "80");
+        }
+        catch(NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
+            fail();
+        }
+    }
+
+}
diff --git a/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/data/MemberBuilderTest.java b/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/data/MemberBuilderTest.java
new file mode 100644
index 00000000..4b657cf0
--- /dev/null
+++ b/grToolkit/provider/src/test/java/org/onap/ccsdk/sli/plugins/grtoolkit/data/MemberBuilderTest.java
@@ -0,0 +1,41 @@
+package org.onap.ccsdk.sli.plugins.grtoolkit.data;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+
+import static org.junit.Assert.*;
+
+public class MemberBuilderTest {
+    MemberBuilder builder;
+    ClusterActor actor;
+
+    @Before
+    public void setUp() {
+        actor = new ClusterActor();
+        actor.setUp(true);
+        actor.setVoting(true);
+    }
+
+    @Test
+    public void constructorTest() {
+        ArrayList<String> actorList = new ArrayList<>();
+        ArrayList<String> shardList = new ArrayList<>();
+        HashMap<String, Integer> commitMap = new HashMap<>();
+        actorList.add("Some-Actor");
+        shardList.add("Some-shard");
+        commitMap.put("Some-shard", 4);
+        commitMap.put("Some-other-shard", -4);
+        actor.setShardLeader(actorList);
+        actor.setReplicaShards(shardList);
+        actor.setNonReplicaShards(shardList);
+        actor.setCommits(commitMap);
+        assertNotNull(actor.toString());
+        assertEquals("", actor.getSite());
+        assertEquals(1, actor.getNonReplicaShards().size());
+        builder = new MemberBuilder(actor);
+        assertNotNull(builder.build());
+    }
+}
\ No newline at end of file