@@ -0,0 +1,3097 @@
+/*%0A * Copyright 2012-2018 the original author or authors.%0A *%0A * Licensed under the Apache License, Version 2.0 (the %22License%22);%0A * you may not use this file except in compliance with the License.%0A * You may obtain a copy of the License at%0A *%0A *      http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0A%0Apackage org.springframework.boot.actuate.cassandra;%0A%0Aimport com.datastax.driver.core.ResultSet;%0Aimport com.datastax.driver.core.Row;%0Aimport com.datastax.driver.core.querybuilder.Select;%0Aimport org.junit.Test;%0A%0Aimport org.springframework.boot.actuate.health.Health;%0Aimport org.springframework.boot.actuate.health.Status;%0Aimport org.springframework.data.cassandra.core.CassandraOperations;%0Aimport org.springframework.data.cassandra.core.cql.CqlOperations;%0A%0Aimport static org.assertj.core.api.Assertions.assertThat;%0Aimport static org.mockito.ArgumentMatchers.any;%0Aimport static org.mockito.BDDMockito.given;%0Aimport static org.mockito.Mockito.mock;%0A%0A/**%0A * Tests for %7B@link CassandraHealthIndicator%7D.%0A *%0A * @author Oleksii Bondar%0A */%0A%0Apublic class CassandraHealthIndicatorTests %7B%0A%0A%09@Test(expected = IllegalArgumentException.class)%0A%09public void throwsExceptionOnNullCassandraOperations() %7B%0A%09%09new CassandraHealthIndicator(null);%0A%09%7D%0A%0A%09@Test%0A%09public void verifyHealthStatusWhenExhausted() %7B%0A%09%09CassandraOperations cassandraOperations = mock(CassandraOperations.class);%0A%09%09CqlOperations cqlOperations = mock(CqlOperations.class);%0A%09%09ResultSet resultSet = mock(ResultSet.class);%0A%09%09CassandraHealthIndicator healthIndicator = new CassandraHealthIndicator(%0A%09%09%09%09cassandraOperations);%0A%09%09given(cassandraOperations.getCqlOperations()).willReturn(cqlOperations);%0A%09%09given(cqlOperations.queryForResultSet(any(Select.class))).willReturn(resultSet);%0A%09%09given(resultSet.isExhausted()).willReturn(true);%0A%0A%09%09Health health = healthIndicator.health();%0A%0A%09%09assertThat(health.getStatus()).isEqualTo(Status.UP);%0A%09%7D%0A%0A%09@Test%0A%09public void verifyHealthStatusWithVersion() %7B%0A%09%09CassandraOperations cassandraOperations = mock(CassandraOperations.class);%0A%09%09CqlOperations cqlOperations = mock(CqlOperations.class);%0A%09%09ResultSet resultSet = mock(ResultSet.class);%0A%09%09Row row = mock(Row.class);%0A%09%09CassandraHealthIndicator healthIndicator = new CassandraHealthIndicator(%0A%09%09%09%09cassandraOperations);%0A%09%09given(cassandraOperations.getCqlOperations()).willReturn(cqlOperations);%0A%09%09given(cqlOperations.queryForResultSet(any(Select.class))).willReturn(resultSet);%0A%09%09given(resultSet.isExhausted()).willReturn(false);%0A%09%09given(resultSet.one()).willReturn(row);%0A%09%09String expectedVersion = %221.0.0%22;%0A%09%09given(row.getString(0)).willReturn(expectedVersion);%0A%0A%09%09Health health = healthIndicator.health();%0A%0A%09%09assertThat(health.getStatus()).isEqualTo(Status.UP);%0A%09%09assertThat(health.getDetails().get(%22version%22)).isEqualTo(expectedVersion);%0A%09%7D%0A%0A%7D%0A

Add unit test for cassandra health checker