use of com.datastax.oss.driver.api.core.metadata.Metadata in project janusgraph by JanusGraph.
the class CQLStoreTest method testNewTableOpenDatabase.
@Test
public void testNewTableOpenDatabase() throws BackendException {
// arrange
String someTableName = "foo";
Metadata metadata = mock(Metadata.class);
KeyspaceMetadata keyspaceMetadata = mock(KeyspaceMetadata.class);
when(keyspaceMetadata.getTable(someTableName)).thenReturn(Optional.empty());
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getKeyspace(mockManager.getKeyspaceName())).thenReturn(Optional.of(keyspaceMetadata));
// act
mockManager.openDatabase(someTableName, null);
// assert
verify(session, times(1)).execute(any(Statement.class));
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project janusgraph by JanusGraph.
the class CQLStoreTest method testNewKeyspaceSession.
@Test
public void testNewKeyspaceSession() {
Metadata metadata = mock(Metadata.class);
Optional<KeyspaceMetadata> keyspaceMetadataOptional = Optional.empty();
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getKeyspace(TEST_KEYSPACE_NAME)).thenReturn(keyspaceMetadataOptional);
mockManager.initializeKeyspace();
verify(session, times(1)).execute(any(Statement.class));
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project spring-boot by spring-projects.
the class CassandraDriverHealthIndicatorTests method healthWithNodeVersionShouldAddVersionDetail.
@Test
void healthWithNodeVersionShouldAddVersionDetail() {
CqlSession session = mock(CqlSession.class);
Metadata metadata = mock(Metadata.class);
given(session.getMetadata()).willReturn(metadata);
Node node = mock(Node.class);
given(node.getState()).willReturn(NodeState.UP);
given(node.getCassandraVersion()).willReturn(Version.V4_0_0);
given(metadata.getNodes()).willReturn(createNodesWithRandomUUID(Collections.singletonList(node)));
CassandraDriverHealthIndicator healthIndicator = new CassandraDriverHealthIndicator(session);
Health health = healthIndicator.health();
assertThat(health.getStatus()).isEqualTo(Status.UP);
assertThat(health.getDetails().get("version")).isEqualTo(Version.V4_0_0);
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project spring-boot by spring-projects.
the class CassandraDriverReactiveHealthIndicatorTests method healthWithNodeVersionShouldAddVersionDetail.
@Test
void healthWithNodeVersionShouldAddVersionDetail() {
CqlSession session = mock(CqlSession.class);
Metadata metadata = mock(Metadata.class);
given(session.getMetadata()).willReturn(metadata);
Node node = mock(Node.class);
given(node.getState()).willReturn(NodeState.UP);
given(node.getCassandraVersion()).willReturn(Version.V4_0_0);
given(metadata.getNodes()).willReturn(createNodesWithRandomUUID(Collections.singletonList(node)));
CassandraDriverReactiveHealthIndicator healthIndicator = new CassandraDriverReactiveHealthIndicator(session);
Mono<Health> health = healthIndicator.health();
StepVerifier.create(health).consumeNextWith((h) -> {
assertThat(h.getStatus()).isEqualTo(Status.UP);
assertThat(h.getDetails()).containsOnlyKeys("version");
assertThat(h.getDetails().get("version")).isEqualTo(Version.V4_0_0);
}).verifyComplete();
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project janusgraph by JanusGraph.
the class CQLStoreTest method testExistTableOpenDatabase.
@Test
public void testExistTableOpenDatabase() throws BackendException {
// arrange
String someTableName = "foo";
Metadata metadata = mock(Metadata.class);
KeyspaceMetadata keyspaceMetadata = mock(KeyspaceMetadata.class);
TableMetadata tableMetadata = mock(TableMetadata.class);
when(keyspaceMetadata.getTable(someTableName)).thenReturn(Optional.of(tableMetadata));
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getKeyspace(mockManager.getKeyspaceName())).thenReturn(Optional.of(keyspaceMetadata));
// act
mockManager.openDatabase(someTableName, null);
// assert
verify(session, never()).execute(any(Statement.class));
}
Aggregations