catalog-dao/src/test/java/org/openecomp/sdc/be/dao/cassandra/CassandraClientTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.dao.cassandra;
  22
  23 import com.datastax.driver.core.Cluster;
  24 import com.datastax.driver.core.Cluster.Builder;
  25 import com.datastax.driver.core.Session;
  26 import com.datastax.driver.mapping.MappingManager;
  27 import fj.data.Either;
  28 import mockit.Deencapsulation;
  29 import org.apache.commons.lang3.tuple.ImmutablePair;
  30 import org.apache.tinkerpop.gremlin.structure.T;
  31 import org.junit.Test;
  32 import org.mockito.Mockito;
  33 import org.openecomp.sdc.be.config.ConfigurationManager;
  34 import org.openecomp.sdc.be.utils.DAOConfDependentTest;
  35
  36 public class CassandraClientTest extends DAOConfDependentTest {
  37
  38         private CassandraClient createTestSubject() {
  39                 return new CassandraClient();
  40         }
  41
  42         @Test
  43         public void testSetLocalDc() throws Exception {
  44                 CassandraClient testSubject;
  45                 Cluster.Builder clusterBuilder = null;
  46
  47                 Builder mock = Mockito.mock(Cluster.Builder.class);
  48                 Mockito.when(mock.withLoadBalancingPolicy(Mockito.any())).thenReturn(new Builder());
  49                 // default test
  50                 testSubject = createTestSubject();
  51                 Deencapsulation.invoke(testSubject, "setLocalDc", new Object[] { Cluster.Builder.class });
  52
  53                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig()
  54                                 .setLocalDataCenter("mock");
  55
  56                 testSubject = createTestSubject();
  57                 Deencapsulation.invoke(testSubject, "setLocalDc", mock);
  58         }
  59
  60         @Test
  61         public void testEnableSsl() throws Exception {
  62                 CassandraClient testSubject;
  63                 Cluster.Builder clusterBuilder = null;
  64
  65                 Builder mock = Mockito.mock(Cluster.Builder.class);
  66                 Mockito.when(mock.withSSL()).thenReturn(new Builder());
  67
  68                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setSsl(false);
  69                 // default test
  70                 testSubject = createTestSubject();
  71                 Deencapsulation.invoke(testSubject, "enableSsl", mock);
  72
  73                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setSsl(true);
  74                 // default test
  75                 testSubject = createTestSubject();
  76                 Deencapsulation.invoke(testSubject, "enableSsl", mock);
  77                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setTruststorePath(null);
  78                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig()
  79                                 .setTruststorePassword(null);
  80                 Deencapsulation.invoke(testSubject, "enableSsl", mock);
  81         }
  82
  83         @Test
  84         public void testEnableAuthentication() throws Exception {
  85                 CassandraClient testSubject;
  86                 Builder mock = Mockito.mock(Cluster.Builder.class);
  87                 Mockito.when(mock.withCredentials(Mockito.any(), Mockito.any())).thenReturn(new Builder());
  88
  89                 // default test
  90                 testSubject = createTestSubject();
  91                 Deencapsulation.invoke(testSubject, "enableAuthentication", Cluster.Builder.class);
  92
  93                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setAuthenticate(true);
  94                 Deencapsulation.invoke(testSubject, "enableAuthentication", mock);
  95                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setUsername(null);
  96                 ConfigurationManager.getConfigurationManager().getConfiguration().getCassandraConfig().setPassword(null);
  97                 Deencapsulation.invoke(testSubject, "enableAuthentication", mock);
  98         }
  99
 100         @Test
 101         public void testConnect() throws Exception {
 102                 CassandraClient testSubject;
 103                 String keyspace = "";
 104                 Either<ImmutablePair<Session, MappingManager>, CassandraOperationStatus> result;
 105
 106                 // default test
 107                 testSubject = createTestSubject();
 108                 result = testSubject.connect(keyspace);
 109         }
 110
 111         @Test
 112         public void testSave() throws Exception {
 113                 CassandraClient testSubject;
 114                 T entity = null;
 115                 Class<T> clazz = null;
 116                 MappingManager manager = null;
 117                 CassandraOperationStatus result;
 118
 119                 // default test
 120                 testSubject = createTestSubject();
 121                 result = testSubject.save(entity, clazz, manager);
 122         }
 123
 124         @Test
 125         public void testGetById() throws Exception {
 126                 CassandraClient testSubject;
 127                 String id = "";
 128                 Class<T> clazz = null;
 129                 MappingManager manager = null;
 130                 Either<T, CassandraOperationStatus> result;
 131
 132                 // default test
 133                 testSubject = createTestSubject();
 134                 result = testSubject.getById(id, clazz, manager);
 135         }
 136
 137         @Test
 138         public void testDelete() throws Exception {
 139                 CassandraClient testSubject;
 140                 String id = "";
 141                 Class<T> clazz = null;
 142                 MappingManager manager = null;
 143                 CassandraOperationStatus result;
 144
 145                 // default test
 146                 testSubject = createTestSubject();
 147                 result = testSubject.delete(id, clazz, manager);
 148         }
 149
 150         @Test
 151         public void testIsConnected() throws Exception {
 152                 CassandraClient testSubject;
 153                 boolean result;
 154
 155                 // default test
 156                 testSubject = createTestSubject();
 157                 result = testSubject.isConnected();
 158         }
 159
 160         @Test
 161         public void testCloseClient() throws Exception {
 162                 CassandraClient testSubject;
 163
 164                 // default test
 165                 testSubject = createTestSubject();
 166                 testSubject.closeClient();
 167         }
 168 }