Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.stargate.it.grpc.ExecuteQueryTest Maven / Gradle / Ivy
/*
* Copyright The Stargate Authors
*
* 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.
*/
package io.stargate.it.grpc;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import com.datastax.oss.driver.api.core.CqlIdentifier;
import com.datastax.oss.driver.api.core.CqlSession;
import com.google.protobuf.BytesValue;
import com.google.protobuf.Int32Value;
import com.google.protobuf.InvalidProtocolBufferException;
import io.grpc.StatusRuntimeException;
import io.stargate.grpc.CqlDuration;
import io.stargate.grpc.Values;
import io.stargate.it.driver.CqlSessionExtension;
import io.stargate.it.driver.CqlSessionSpec;
import io.stargate.it.driver.TestKeyspace;
import io.stargate.proto.QueryOuterClass;
import io.stargate.proto.QueryOuterClass.Query;
import io.stargate.proto.QueryOuterClass.Response;
import io.stargate.proto.QueryOuterClass.ResultSet;
import io.stargate.proto.QueryOuterClass.SchemaChange;
import io.stargate.proto.QueryOuterClass.TypeSpec;
import io.stargate.proto.StargateGrpc.StargateBlockingStub;
import java.util.Arrays;
import java.util.HashSet;
import java.util.UUID;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
@ExtendWith(CqlSessionExtension.class)
@CqlSessionSpec(
initQueries = {
"CREATE TABLE IF NOT EXISTS test (k text, v int, PRIMARY KEY(k, v))",
"CREATE TABLE IF NOT EXISTS test_uuid (id uuid, value int, PRIMARY KEY(id, value))",
"CREATE TYPE IF NOT EXISTS address(street text, phone_numbers frozen>)",
"CREATE TABLE IF NOT EXISTS user(id int PRIMARY KEY, name text, address address)",
"CREATE TABLE IF NOT EXISTS test_duration(id int PRIMARY KEY, d duration)"
})
public class ExecuteQueryTest extends GrpcIntegrationTest {
@AfterEach
public void cleanup(CqlSession session) {
session.execute("TRUNCATE TABLE test");
session.execute("TRUNCATE TABLE test_uuid");
}
@Test
public void simpleQuery(@TestKeyspace CqlIdentifier keyspace)
throws InvalidProtocolBufferException {
StargateBlockingStub stub = stubWithCallCredentials();
Response response =
stub.executeQuery(
cqlQuery("INSERT INTO test (k, v) VALUES ('a', 1)", queryParameters(keyspace)));
assertThat(response).isNotNull();
response =
stub.executeQuery(
cqlQuery(
"INSERT INTO test (k, v) VALUES (?, ?)",
queryParameters(keyspace),
Values.of("b"),
Values.of(2)));
assertThat(response).isNotNull();
response = stub.executeQuery(cqlQuery("SELECT * FROM test", queryParameters(keyspace)));
assertThat(response.hasResultSet()).isTrue();
ResultSet rs = response.getResultSet();
assertThat(new HashSet<>(rs.getRowsList()))
.isEqualTo(
new HashSet<>(
Arrays.asList(
rowOf(Values.of("a"), Values.of(1)), rowOf(Values.of("b"), Values.of(2)))));
}
@Test
public void queryAfterSchemaChange() {
StargateBlockingStub stub = stubWithCallCredentials();
stub.executeQuery(cqlQuery("DROP KEYSPACE IF EXISTS ks1"));
// Create keyspace, table, and then insert some data
Response response =
stub.executeQuery(
cqlQuery(
"CREATE KEYSPACE IF NOT EXISTS ks1 WITH REPLICATION = {'class' : 'SimpleStrategy', 'replication_factor' : 1};"));
assertThat(response).isNotNull();
assertThat(response.hasSchemaChange()).isTrue();
assertThat(response.getSchemaChange())
.satisfies(
c -> {
assertThat(c.getChangeType()).isEqualTo(SchemaChange.Type.CREATED);
assertThat(c.getTarget()).isEqualTo(SchemaChange.Target.KEYSPACE);
assertThat(c.getKeyspace()).isEqualTo("ks1");
assertThat(c.hasName()).isFalse();
assertThat(c.getArgumentTypesCount()).isEqualTo(0);
});
response =
stub.executeQuery(
cqlQuery("CREATE TABLE IF NOT EXISTS ks1.tbl1 (k text, v int, PRIMARY KEY (k));"));
assertThat(response).isNotNull();
assertThat(response.hasSchemaChange()).isTrue();
assertThat(response.getSchemaChange())
.satisfies(
c -> {
assertThat(c.getChangeType()).isEqualTo(SchemaChange.Type.CREATED);
assertThat(c.getTarget()).isEqualTo(SchemaChange.Target.TABLE);
assertThat(c.getKeyspace()).isEqualTo("ks1");
assertThat(c.getName().getValue()).isEqualTo("tbl1");
assertThat(c.getArgumentTypesCount()).isEqualTo(0);
});
response = stub.executeQuery(cqlQuery("INSERT INTO ks1.tbl1 (k, v) VALUES ('a', 1)"));
assertThat(response).isNotNull();
assertThat(response.hasSchemaChange()).isFalse();
// Drop the keyspace to cause the existing prepared queries to be purged from the backend query
// cache
response = stub.executeQuery(cqlQuery("DROP KEYSPACE ks1;"));
assertThat(response).isNotNull();
assertThat(response.hasSchemaChange()).isTrue();
assertThat(response.getSchemaChange())
.satisfies(
c -> {
assertThat(c.getChangeType()).isEqualTo(SchemaChange.Type.DROPPED);
assertThat(c.getTarget()).isEqualTo(SchemaChange.Target.KEYSPACE);
assertThat(c.getKeyspace()).isEqualTo("ks1");
assertThat(c.hasName()).isFalse();
assertThat(c.getArgumentTypesCount()).isEqualTo(0);
});
response =
stub.executeQuery(
cqlQuery(
"CREATE KEYSPACE IF NOT EXISTS ks1 WITH REPLICATION = {'class' : 'SimpleStrategy', 'replication_factor' : 1};"));
assertThat(response).isNotNull();
response =
stub.executeQuery(
cqlQuery("CREATE TABLE IF NOT EXISTS ks1.tbl1 (k text, v int, PRIMARY KEY (k));"));
assertThat(response).isNotNull();
response = stub.executeQuery(cqlQuery("INSERT INTO ks1.tbl1 (k, v) VALUES ('a', 1)"));
assertThat(response).isNotNull();
}
@Test
public void simpleQueryWithPaging(@TestKeyspace CqlIdentifier keyspace)
throws InvalidProtocolBufferException {
StargateBlockingStub stub = stubWithCallCredentials();
for (int i = 0; i < 150; i++) {
stub.executeQuery(
cqlQuery(
String.format("INSERT INTO test (k, v) VALUES ('%d', %d)", i, i),
queryParameters(keyspace)));
}
// No explicit page size => should default to 100
Response response =
stub.executeQuery(cqlQuery("select k,v from test", queryParameters(keyspace)));
assertThat(response.hasResultSet()).isTrue();
ResultSet rs = response.getResultSet();
assertThat(rs.getRowsCount()).isEqualTo(100);
assertThat(rs.getPagingState()).isNotNull();
// Explicit page size
response =
stub.executeQuery(
cqlQuery(
"select k,v from test",
queryParameters(keyspace)
.setPageSize(Int32Value.newBuilder().setValue(2).build())));
assertThat(response.hasResultSet()).isTrue();
rs = response.getResultSet();
assertThat(rs.getRowsCount()).isEqualTo(2);
assertThat(rs.getPagingState()).isNotNull();
}
@Test
public void simpleQueryWithPagingUuidKey(@TestKeyspace CqlIdentifier keyspace) {
StargateBlockingStub stub = stubWithCallCredentials();
final UUID uuid = UUID.randomUUID();
final String uuidStr = uuid.toString();
for (int i = 0; i < 5; i++) {
stub.executeQuery(
cqlQuery(
String.format("INSERT INTO test_uuid (id, value) VALUES (%s, %d)", uuidStr, i),
queryParameters(keyspace)));
}
Response response =
stub.executeQuery(
cqlQuery(
"select id,value from test_uuid",
queryParameters(keyspace)
.setPageSize(Int32Value.newBuilder().setValue(2).build())));
assertThat(response.hasResultSet()).isTrue();
ResultSet rs = response.getResultSet();
assertThat(rs.getRowsCount()).isEqualTo(2);
BytesValue pagingState = rs.getPagingState();
assertThat(pagingState).isNotNull();
assertThat(rs.getRows(0)).isEqualTo(rowOf(Values.of(uuid), Values.of(Integer.valueOf(0))));
assertThat(rs.getRows(1)).isEqualTo(rowOf(Values.of(uuid), Values.of(Integer.valueOf(1))));
// And then actual paging:
response =
stub.executeQuery(
cqlQuery(
"select id,value from test_uuid",
queryParameters(keyspace)
.setPageSize(Int32Value.newBuilder().setValue(4).build())
.setPagingState(pagingState)));
assertThat(response.hasResultSet()).isTrue();
rs = response.getResultSet();
assertThat(rs.getRowsCount()).isEqualTo(3);
assertThat(rs.getRows(0)).isEqualTo(rowOf(Values.of(uuid), Values.of(Integer.valueOf(2))));
assertThat(rs.getRows(1)).isEqualTo(rowOf(Values.of(uuid), Values.of(Integer.valueOf(3))));
assertThat(rs.getRows(2)).isEqualTo(rowOf(Values.of(uuid), Values.of(Integer.valueOf(4))));
// but should now get empty paging state
pagingState = rs.getPagingState();
assertThat(pagingState).isNotNull();
assertThat(pagingState.getValue().size()).isEqualTo(0);
// which means we should NOT try to ask for more rows
// (would give "io.grpc.StatusRuntimeException: INTERNAL: Invalid value for the paging state")
}
@Test
public void useKeyspace(@TestKeyspace CqlIdentifier keyspace) {
StargateBlockingStub stub = stubWithCallCredentials();
assertThatThrownBy(
() -> {
Response response =
stub.executeQuery(Query.newBuilder().setCql("USE system").build());
assertThat(response).isNotNull();
})
.isInstanceOf(StatusRuntimeException.class)
.hasMessageContaining("USE not supported");
// Verify that system local doesn't work
assertThatThrownBy(
() -> {
Response response =
stub.executeQuery(Query.newBuilder().setCql("SELECT * FROM local").build());
assertThat(response).isNotNull();
})
.isInstanceOf(StatusRuntimeException.class)
.hasMessageContaining("No keyspace has been specified");
// Verify that setting the keyspace using parameters still works
Response response =
stub.executeQuery(
cqlQuery("INSERT INTO test (k, v) VALUES ('a', 1)", queryParameters(keyspace)));
assertThat(response).isNotNull();
}
@Test
public void dropTableAndChangeType(@TestKeyspace CqlIdentifier keyspace) {
StargateBlockingStub stub = stubWithCallCredentials();
Response response;
response =
stub.executeQuery(
cqlQuery(
"CREATE TABLE IF NOT EXISTS drop_table_test(pk BIGINT PRIMARY KEY)",
queryParameters(keyspace)));
assertThat(response).isNotNull();
response =
stub.executeQuery(
cqlQuery("INSERT INTO drop_table_test(pk) VALUES(1)", queryParameters(keyspace)));
assertThat(response).isNotNull();
response = stub.executeQuery(cqlQuery("DROP TABLE drop_table_test", queryParameters(keyspace)));
assertThat(response).isNotNull();
response =
stub.executeQuery(
cqlQuery(
"CREATE TABLE IF NOT EXISTS drop_table_test(pk TEXT PRIMARY KEY)",
queryParameters(keyspace)));
assertThat(response).isNotNull();
response =
stub.executeQuery(
cqlQuery("INSERT INTO drop_table_test(pk) VALUES('1')", queryParameters(keyspace)));
assertThat(response).isNotNull();
}
@Test
public void resultSetContainsUdt(@TestKeyspace CqlIdentifier keyspace) {
StargateBlockingStub stub = stubWithCallCredentials();
// Note: the table is empty but we're only interested in the metadata
Response response =
stub.executeQuery(
cqlQuery("SELECT address FROM user WHERE id=1", queryParameters(keyspace)));
TypeSpec addressType = response.getResultSet().getColumns(0).getType();
assertThat(addressType.getSpecCase()).isEqualTo(TypeSpec.SpecCase.UDT);
TypeSpec phoneNumbersType = addressType.getUdt().getFieldsMap().get("phone_numbers");
assertThat(phoneNumbersType.getSpecCase()).isEqualTo(TypeSpec.SpecCase.MAP);
assertThat(phoneNumbersType.getMap())
.satisfies(
map -> {
assertThat(map.getKey().getBasic()).isEqualTo(TypeSpec.Basic.VARCHAR);
assertThat(map.getValue().getBasic()).isEqualTo(TypeSpec.Basic.VARCHAR);
});
}
@Test
public void shouldHandleDurations(@TestKeyspace CqlIdentifier keyspace) {
StargateBlockingStub stub = stubWithCallCredentials();
CqlDuration duration = CqlDuration.newInstance(1, 5, 100);
stub.executeQuery(
cqlQuery(
"INSERT INTO test_duration (id, d) VALUES (1, ?)",
queryParameters(keyspace),
Values.of(duration)));
QueryOuterClass.Row row =
stub.executeQuery(
cqlQuery("SELECT d FROM test_duration WHERE id = 1", queryParameters(keyspace)))
.getResultSet()
.getRows(0);
QueryOuterClass.Value value = row.getValues(0);
assertThat(Values.duration(value)).isEqualTo(duration);
}
}
