use of com.datastax.oss.driver.api.core.metadata.Metadata in project janusgraph by JanusGraph.
the class CQLStoreTest method testExistKeyspaceSession.
@Test
public void testExistKeyspaceSession() {
Metadata metadata = mock(Metadata.class);
KeyspaceMetadata keyspaceMetadata = mock(KeyspaceMetadata.class);
Optional<KeyspaceMetadata> keyspaceMetadataOptional = Optional.of(keyspaceMetadata);
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getKeyspace(TEST_KEYSPACE_NAME)).thenReturn(keyspaceMetadataOptional);
mockManager.initializeKeyspace();
verify(session, never()).execute(any(Statement.class));
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project zipkin by openzipkin.
the class SchemaTest method ensureKeyspaceMetadata_failsWhenOneVersionLessThan3_11_3.
@Test
public void ensureKeyspaceMetadata_failsWhenOneVersionLessThan3_11_3() {
CqlSession session = mock(CqlSession.class);
Metadata metadata = mock(Metadata.class);
Node node1 = mock(Node.class);
Node node2 = mock(Node.class);
Map<UUID, Node> nodes = new LinkedHashMap<>();
nodes.put(UUID.fromString("11111111-1111-1111-1111-111111111111"), node1);
nodes.put(UUID.fromString("22222222-2222-2222-2222-222222222222"), node2);
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getNodes()).thenReturn(nodes);
when(node1.getCassandraVersion()).thenReturn(Version.parse("3.11.3"));
when(node2.getCassandraVersion()).thenReturn(Version.parse("3.11.2"));
assertThatThrownBy(() -> Schema.ensureKeyspaceMetadata(session, "zipkin2")).isInstanceOf(RuntimeException.class).hasMessage("Node 22222222-2222-2222-2222-222222222222 is running Cassandra 3.11.2, but minimum version is 3.11.3");
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project zipkin by openzipkin.
the class SchemaTest method ensureKeyspaceMetadata_passesWhenVersion3_11_3AndKeyspaceMetadataIsNotNull.
@Test
public void ensureKeyspaceMetadata_passesWhenVersion3_11_3AndKeyspaceMetadataIsNotNull() {
CqlSession session = mock(CqlSession.class);
Metadata metadata = mock(Metadata.class);
Node node = mock(Node.class);
KeyspaceMetadata keyspaceMetadata = mock(KeyspaceMetadata.class);
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getNodes()).thenReturn(Collections.singletonMap(UUID.fromString("11111111-1111-1111-1111-111111111111"), node));
when(node.getCassandraVersion()).thenReturn(Version.parse("3.11.3"));
when(metadata.getKeyspace("zipkin2")).thenReturn(Optional.of(keyspaceMetadata));
assertThat(Schema.ensureKeyspaceMetadata(session, "zipkin2")).isSameAs(keyspaceMetadata);
}
use of com.datastax.oss.driver.api.core.metadata.Metadata in project zipkin by openzipkin.
the class SchemaTest method ensureKeyspaceMetadata_passesWhenVersion3_11_4AndKeyspaceMetadataIsNotNull.
@Test
public void ensureKeyspaceMetadata_passesWhenVersion3_11_4AndKeyspaceMetadataIsNotNull() {
CqlSession session = mock(CqlSession.class);
Metadata metadata = mock(Metadata.class);
Node node = mock(Node.class);
KeyspaceMetadata keyspaceMetadata = mock(KeyspaceMetadata.class);
when(session.getMetadata()).thenReturn(metadata);
when(metadata.getNodes()).thenReturn(Collections.singletonMap(UUID.fromString("11111111-1111-1111-1111-111111111111"), node));
when(node.getCassandraVersion()).thenReturn(Version.parse("3.11.4"));
when(metadata.getKeyspace("zipkin2")).thenReturn(Optional.of(keyspaceMetadata));
assertThat(Schema.ensureKeyspaceMetadata(session, "zipkin2")).isSameAs(keyspaceMetadata);
}
Aggregations