io.stargate.it.http.graphql.graphqlfirst.BulkInsertTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of testing Show documentation
There is a newer version: 2.1.0-BETA-19
/*
 * 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.graphqlfirst;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatCode;

import com.datastax.oss.driver.api.core.CqlIdentifier;
import com.datastax.oss.driver.api.core.CqlSession;
import com.jayway.jsonpath.JsonPath;
import io.stargate.it.driver.CqlSessionExtension;
import io.stargate.it.driver.TestKeyspace;
import io.stargate.it.http.RestUtils;
import io.stargate.it.storage.StargateConnectionInfo;
import java.util.UUID;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

@ExtendWith(CqlSessionExtension.class)
public class BulkInsertTest extends GraphqlFirstTestBase {

  private static GraphqlFirstClient CLIENT;
  private static String KEYSPACE;
  private static CqlSession SESSION;

  @BeforeAll
  public static void setup(
      StargateConnectionInfo cluster, @TestKeyspace CqlIdentifier keyspaceId, CqlSession session) {
    CLIENT =
        new GraphqlFirstClient(
            cluster.seedAddress(), RestUtils.getAuthToken(cluster.seedAddress()));
    KEYSPACE = keyspaceId.asInternal();
    SESSION = session;
    CLIENT.deploySchema(
        KEYSPACE,
        "type User @cql_input {\n"
            + "  id: ID!\n"
            + "  name: String\n"
            + "  username: String\n"
            + "}\n"
            + "type Query { user(id: ID!): User }\n"
            + "type InsertUserResponse @cql_payload {\n"
            + "  applied: Boolean!\n"
            + "  user: User!\n"
            + "}\n"
            + "type Mutation {\n"
            + "  bulkInsertUsers(users: [UserInput!]): [User]\n"
            + "  bulkInsertUsersCustom(users: [UserInput!]): [InsertUserResponse]\n"
            + "  bulkInsertUsersBoolean(users: [UserInput!]): [Boolean]\n"
            + "}");
  }

  @BeforeEach
  public void cleanupData() {
    SESSION.execute("truncate table \"User\"");
  }

  @Test
  @DisplayName("Should map two inserts in one graphQL statement, returning list of entities")
  public void testBulkInsertReturnListOfEntities() {
    // given, when
    Object response =
        CLIENT.executeKeyspaceQuery(
            KEYSPACE,
            "mutation {\n"
                + "bulkInsertUsers (users: [\n"
                + " { name: \"Ada Lovelace\", username: \"@ada\"},\n"
                + " { name: \"Alan Turing\", username: \"@alan\"}\n"
                + "]) \n"
                + "{ \n"
                + "id, name, username }\n"
                + "}");

    // then
    assertThatCode(() -> UUID.fromString(JsonPath.read(response, "$.bulkInsertUsers[0].id")))
        .doesNotThrowAnyException();
    assertThatCode(() -> UUID.fromString(JsonPath.read(response, "$.bulkInsertUsers[1].id")))
        .doesNotThrowAnyException();

    String name = JsonPath.read(response, "$.bulkInsertUsers[0].name");
    assertThat(name).isEqualTo("Ada Lovelace");
    String username = JsonPath.read(response, "$.bulkInsertUsers[0].username");
    assertThat(username).isEqualTo("@ada");
    name = JsonPath.read(response, "$.bulkInsertUsers[1].name");
    assertThat(name).isEqualTo("Alan Turing");
    username = JsonPath.read(response, "$.bulkInsertUsers[1].username");
    assertThat(username).isEqualTo("@alan");
  }

  @Test
  @DisplayName("Should map two inserts in one graphQL statement, returning list of custom payloads")
  public void testBulkInsertReturnListOfCustomPayloads() {
    // given, when
    Object response =
        CLIENT.executeKeyspaceQuery(
            KEYSPACE,
            "mutation {\n"
                + "bulkInsertUsersCustom (users: [\n"
                + " { name: \"Ada Lovelace\", username: \"@ada\"},\n"
                + " { name: \"Alan Turing\", username: \"@alan\"}\n"
                + "]) { \n"
                + "    applied"
                + "    user { id, name, username }\n"
                + "   }\n"
                + "}");

    // then
    assertThatCode(
            () -> UUID.fromString(JsonPath.read(response, "$.bulkInsertUsersCustom[0].user.id")))
        .doesNotThrowAnyException();
    assertThatCode(
            () -> UUID.fromString(JsonPath.read(response, "$.bulkInsertUsersCustom[1].user.id")))
        .doesNotThrowAnyException();

    String name = JsonPath.read(response, "$.bulkInsertUsersCustom[0].user.name");
    assertThat(name).isEqualTo("Ada Lovelace");
    String username = JsonPath.read(response, "$.bulkInsertUsersCustom[0].user.username");
    assertThat(username).isEqualTo("@ada");
    name = JsonPath.read(response, "$.bulkInsertUsersCustom[1].user.name");
    assertThat(name).isEqualTo("Alan Turing");
    username = JsonPath.read(response, "$.bulkInsertUsersCustom[1].user.username");
    assertThat(username).isEqualTo("@alan");
  }

  @Test
  @DisplayName("Should map two inserts in one graphQL statement, returning list of booleans")
  public void testBulkInsertReturnListOfBooleans() {
    // given, when
    Object response =
        CLIENT.executeKeyspaceQuery(
            KEYSPACE,
            "mutation {\n"
                + "bulkInsertUsersBoolean (users: [\n"
                + " { name: \"Ada Lovelace\", username: \"@ada\" },\n"
                + " { name: \"Alan Turing\", username: \"@alan\"}\n"
                + "]) \n"
                + "}");

    // then
    assertThat(JsonPath.read(response, "$.bulkInsertUsersBoolean[0]")).isTrue();
    assertThat(JsonPath.read(response, "$.bulkInsertUsersBoolean[1]")).isTrue();
  }
}