io.stargate.it.http.graphql.cqlfirst.CollectionsTest 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.http.graphql.cqlfirst;
import static org.assertj.core.api.Assertions.assertThat;
import com.datastax.oss.driver.api.core.CqlIdentifier;
import com.jayway.jsonpath.JsonPath;
import io.stargate.it.BaseIntegrationTest;
import io.stargate.it.driver.CqlSessionExtension;
import io.stargate.it.driver.CqlSessionSpec;
import io.stargate.it.driver.TestKeyspace;
import io.stargate.it.http.RestUtils;
import io.stargate.it.storage.StargateConnectionInfo;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
@ExtendWith(CqlSessionExtension.class)
@CqlSessionSpec(
initQueries = {
"CREATE TABLE IF NOT EXISTS \"CollectionsSimple\" (\n"
+ " id uuid PRIMARY KEY,\n"
+ " \"listValue1\" frozen>,\n"
+ " \"setValue1\" frozen>,\n"
+ " \"mapValue1\" frozen>,\n"
+ ")",
"CREATE TABLE IF NOT EXISTS \"CollectionsNested\" (\n"
+ " id uuid PRIMARY KEY,\n"
+ " \"listValue1\" frozen>>>,\n"
+ " \"setValue1\" frozen>>>,\n"
+ " \"mapValue1\" frozen>>>\n"
+ ")"
})
public class CollectionsTest extends BaseIntegrationTest {
private static CqlFirstClient CLIENT;
@BeforeAll
public static void setup(StargateConnectionInfo cluster) {
String host = cluster.seedAddress();
CLIENT = new CqlFirstClient(host, RestUtils.getAuthToken(host));
}
@Test
public void shouldInsertAndUpdateSimpleListSetsAndMaps(@TestKeyspace CqlIdentifier keyspaceId) {
// When inserting a new row:
Map response =
CLIENT.executeDmlQuery(
keyspaceId,
"mutation {\n"
+ " insertCollectionsSimple(value: {\n"
+ " id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"\n"
+ " listValue1: [1, 2, 3]\n"
+ " setValue1: [\"a\", \"b\", \"c\"]\n"
+ " mapValue1: [\n"
+ " {key: 1, value: \"one\"}\n"
+ " {key: 2, value: \"two\"}\n"
+ " ]\n"
+ " }) { applied }}");
assertThat(JsonPath.read(response, "$.insertCollectionsSimple.applied")).isTrue();
// Then the data can be read back:
String getQuery =
"{\n"
+ " CollectionsSimple(value: {id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"}) {\n"
+ " values { listValue1, setValue1, mapValue1 { key, value } }\n"
+ " }}";
response = CLIENT.executeDmlQuery(keyspaceId, getQuery);
List listValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].listValue1");
assertThat(listValue1).containsExactly(1, 2, 3);
List setValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].setValue1");
assertThat(setValue1).containsExactly("a", "b", "c");
List> mapValue1 =
JsonPath.read(response, "$.CollectionsSimple.values[0].mapValue1");
assertThat(mapValue1).hasSize(2);
assertThat(mapValue1.get(0)).hasSize(2).containsEntry("key", 1).containsEntry("value", "one");
assertThat(mapValue1.get(1)).hasSize(2).containsEntry("key", 2).containsEntry("value", "two");
// When updating the row:
response =
CLIENT.executeDmlQuery(
keyspaceId,
"mutation {\n"
+ " updateCollectionsSimple(\n"
+ " value: {\n"
+ " id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\""
+ " listValue1: [4, 5]\n"
+ " setValue1: [\"d\"]\n"
+ " mapValue1: [ {key: 3, value: \"three\"} ]\n"
+ " }) { applied } }");
assertThat(JsonPath.read(response, "$.updateCollectionsSimple.applied")).isTrue();
// Then the changes are reflected:
response = CLIENT.executeDmlQuery(keyspaceId, getQuery);
listValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].listValue1");
assertThat(listValue1).containsExactly(4, 5);
setValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].setValue1");
assertThat(setValue1).containsExactly("d");
mapValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].mapValue1");
assertThat(mapValue1).hasSize(1);
assertThat(mapValue1.get(0)).hasSize(2).containsEntry("key", 3).containsEntry("value", "three");
}
@Test
public void shouldInsertAndUpdateNestedListSetsAndMaps(@TestKeyspace CqlIdentifier keyspaceId) {
// When inserting a new row:
Map response =
CLIENT.executeDmlQuery(
keyspaceId,
"mutation {\n"
+ " insertCollectionsNested(value: {\n"
+ " id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"\n"
+ " listValue1: [ [ {key: 3, value: \"three\"} ] ]\n"
+ " setValue1: [ [ \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\" ] ]\n"
+ " mapValue1: [\n"
+ " {\n"
+ " key: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"\n"
+ " value: [ { key: 123, value: \"one-two-three\" } ]\n"
+ " }\n"
+ " ]\n"
+ " }) { applied }}");
assertThat(JsonPath.read(response, "$.insertCollectionsNested.applied")).isTrue();
// Then the data can be read back:
String getQuery =
"{\n"
+ " CollectionsNested(value: {id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"}) {\n"
+ " values {\n"
+ " listValue1 { key, value}\n"
+ " setValue1\n"
+ " mapValue1 { key, value { key, value } } }\n"
+ " }}";
response = CLIENT.executeDmlQuery(keyspaceId, getQuery);
List>> listValue1 =
JsonPath.read(response, "$.CollectionsNested.values[0].listValue1");
assertThat(listValue1).hasSize(1);
List> nestedMapInListValue1 = listValue1.get(0);
assertThat(nestedMapInListValue1).hasSize(1);
assertThat(nestedMapInListValue1.get(0))
.hasSize(2)
.containsEntry("key", 3)
.containsEntry("value", "three");
List> setValue1 =
JsonPath.read(response, "$.CollectionsNested.values[0].setValue1");
assertThat(setValue1).hasSize(1);
List nestedSetInSetValue1 = setValue1.get(0);
assertThat(nestedSetInSetValue1).containsExactly("792d0a56-bb46-4bc2-bc41-5f4a94a83da9");
List> mapValue1 =
JsonPath.read(response, "$.CollectionsNested.values[0].mapValue1");
assertThat(mapValue1).hasSize(1);
Object keyInMapValue1 = mapValue1.get(0).get("key");
assertThat(keyInMapValue1).isEqualTo("792d0a56-bb46-4bc2-bc41-5f4a94a83da9");
@SuppressWarnings("unchecked")
List> nestedMapInMapValue1 =
(List>) mapValue1.get(0).get("value");
assertThat(nestedMapInMapValue1).hasSize(1);
assertThat(nestedMapInMapValue1.get(0))
.hasSize(2)
.containsEntry("key", "123")
.containsEntry("value", "one-two-three");
}
@Test
public void shouldInsertNullCollections(@TestKeyspace CqlIdentifier keyspaceId) {
// When inserting a new row:
Map response =
CLIENT.executeDmlQuery(
keyspaceId,
"mutation {\n"
+ " insertCollectionsSimple(value: {\n"
+ " id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"\n"
+ " listValue1: null\n"
+ " setValue1: null\n"
+ " mapValue1: null\n"
+ " }) { applied }}");
assertThat(JsonPath.read(response, "$.insertCollectionsSimple.applied")).isTrue();
// Then the data can be read back:
String getQuery =
"{\n"
+ " CollectionsSimple(value: {id: \"792d0a56-bb46-4bc2-bc41-5f4a94a83da9\"}) {\n"
+ " values { listValue1, setValue1, mapValue1 { key, value } }\n"
+ " }}";
response = CLIENT.executeDmlQuery(keyspaceId, getQuery);
List listValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].listValue1");
assertThat(listValue1).isNull();
List setValue1 = JsonPath.read(response, "$.CollectionsSimple.values[0].setValue1");
assertThat(setValue1).isNull();
List> mapValue1 =
JsonPath.read(response, "$.CollectionsSimple.values[0].mapValue1");
assertThat(mapValue1).isNull();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy