/*- * ============LICENSE_START======================================================= * SDC * ================================================================================ * Copyright (C) 2019 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.openecomp.sdc.be.dao.cassandra; import com.datastax.driver.core.Session; import com.datastax.driver.mapping.MappingManager; import com.datastax.driver.mapping.Result; import fj.data.Either; import org.apache.commons.lang3.tuple.ImmutablePair; import org.junit.Before; import org.junit.Test; import org.mockito.InjectMocks; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.MockitoAnnotations; import org.openecomp.sdc.be.datatypes.enums.EnvironmentStatusEnum; import org.openecomp.sdc.be.resources.data.OperationalEnvironmentEntry; import java.util.List; import static org.junit.Assert.assertTrue; public class OperationalEnvironmentDaoTest { @InjectMocks OperationalEnvironmentDao testSubject; @Mock CassandraClient clientMock; @Before public void setUp() throws Exception { MockitoAnnotations.openMocks(this); } @Test public void testInit() throws Exception { // default test testSubject.init(); Mockito.when(clientMock.isConnected()).thenReturn(true); Session sessMock = Mockito.mock(Session.class); MappingManager mappMock = Mockito.mock(MappingManager.class); ImmutablePair ipMock = ImmutablePair.of(sessMock, mappMock); Either, CassandraOperationStatus> value = Either.left(ipMock); Mockito.when(clientMock.connect(Mockito.any())).thenReturn(value); testSubject.init(); } @Test public void testInitException() throws Exception { // default test testSubject.init(); Mockito.when(clientMock.isConnected()).thenReturn(true); Either, CassandraOperationStatus> value = Either .right(CassandraOperationStatus.CLUSTER_NOT_CONNECTED); Mockito.when(clientMock.connect(Mockito.any())).thenReturn(value); try { testSubject.init(); } catch (Exception e) { assertTrue(e.getClass() == RuntimeException.class); } } @Test public void testSave() throws Exception { OperationalEnvironmentEntry operationalEnvironmentEntry = null; CassandraOperationStatus result; // default test result = testSubject.save(operationalEnvironmentEntry); } @Test public void testGet() throws Exception { String envId = ""; Either result; // default test result = testSubject.get(envId); } @Test public void testDelete() throws Exception { String envId = ""; CassandraOperationStatus result; // default test result = testSubject.delete(envId); } @Test public void testDeleteAll() throws Exception { CassandraOperationStatus result; // default test result = testSubject.deleteAll(); Mockito.when(clientMock.isConnected()).thenReturn(true); Session sessMock = Mockito.mock(Session.class); MappingManager mappMock = Mockito.mock(MappingManager.class); ImmutablePair ipMock = ImmutablePair.of(sessMock, mappMock); Either, CassandraOperationStatus> value = Either.left(ipMock); Mockito.when(clientMock.connect(Mockito.any())).thenReturn(value); testSubject.init(); result = testSubject.deleteAll(); } @Test public void testIsTableEmpty() throws Exception { String tableName = ""; Either result; // default test result = testSubject.isTableEmpty(tableName); } @Test public void testGetByEnvironmentsStatus() throws Exception { Either, CassandraOperationStatus> result; Mockito.when(clientMock.isConnected()).thenReturn(true); Session sessMock = Mockito.mock(Session.class); MappingManager mappMock = Mockito.mock(MappingManager.class); ImmutablePair ipMock = ImmutablePair.of(sessMock, mappMock); Either, CassandraOperationStatus> value = Either.left(ipMock); Mockito.when(clientMock.connect(Mockito.any())).thenReturn(value); OperationalEnvironmentsAccessor value2 = Mockito.mock(OperationalEnvironmentsAccessor.class); Mockito.when(mappMock.createAccessor(OperationalEnvironmentsAccessor.class)).thenReturn(value2); Result value3 = Mockito.mock(Result.class); Mockito.when(value2.getByEnvironmentsStatus(Mockito.any(String.class))).thenReturn(value3); testSubject.init(); // default test result = testSubject.getByEnvironmentsStatus(EnvironmentStatusEnum.COMPLETED); } @Test public void testGetByEnvironmentsStatusNull() throws Exception { Either, CassandraOperationStatus> result; Mockito.when(clientMock.isConnected()).thenReturn(true); Session sessMock = Mockito.mock(Session.class); MappingManager mappMock = Mockito.mock(MappingManager.class); ImmutablePair ipMock = ImmutablePair.of(sessMock, mappMock); Either, CassandraOperationStatus> value = Either.left(ipMock); Mockito.when(clientMock.connect(Mockito.any())).thenReturn(value); OperationalEnvironmentsAccessor value2 = Mockito.mock(OperationalEnvironmentsAccessor.class); Mockito.when(mappMock.createAccessor(OperationalEnvironmentsAccessor.class)).thenReturn(value2); Result value3 = null; Mockito.when(value2.getByEnvironmentsStatus(Mockito.any(String.class))).thenReturn(value3); testSubject.init(); // default test result = testSubject.getByEnvironmentsStatus(EnvironmentStatusEnum.COMPLETED); } }