io.stargate.it.http.graphql.cqlfirst.InsertTest 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.cqlfirst;

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

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.TestKeyspace;
import io.stargate.it.http.RestUtils;
import io.stargate.it.storage.StargateConnectionInfo;
import java.util.Arrays;
import java.util.Map;
import java.util.UUID;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

public class InsertTest extends BetterbotzTestBase {

  private static final String ID = UUID.randomUUID().toString();

  private static CqlFirstClient CLIENT;
  private static CqlIdentifier KEYSPACE_ID;

  @BeforeAll
  public static void setup(StargateConnectionInfo cluster, @TestKeyspace CqlIdentifier keyspaceId) {
    String host = cluster.seedAddress();
    CLIENT = new CqlFirstClient(host, RestUtils.getAuthToken(host));
    KEYSPACE_ID = keyspaceId;
  }

  @AfterEach
  public void cleanup(CqlSession session) {
    session.execute(String.format("DELETE FROM \"Products\" WHERE id = %s", ID));
  }

  @Test
  public void insertProduct() {
    Map response =
        insertProduct(
            ID, "Shiny Legs", "3199.99", "2011-02-02T20:05:00.000-08:00", "Normal legs but shiny.");
    assertThat(JsonPath.read(response, "$.insertProducts.applied")).isTrue();
    Map productInInsertResponse = JsonPath.read(response, "$.insertProducts.value");

    response = selectProduct(ID);
    Map productInSelectResponse = JsonPath.read(response, "$.Products.values[0]");

    for (Map product :
        Arrays.asList(productInInsertResponse, productInSelectResponse)) {
      assertIsProduct(
          product,
          ID,
          "Shiny Legs",
          "3199.99",
          "2011-02-02T20:05:00.000-08:00",
          "Normal legs but shiny.");
    }
  }

  @Test
  public void insertProductIfNotExistsApplied() {
    Map response =
        insertProductIfNotExists(
            ID, "Shiny Legs", "3199.99", "2011-02-02T20:05:00.000-08:00", "Normal legs but shiny.");
    assertThat(JsonPath.read(response, "$.insertProducts.applied")).isTrue();
    assertIsProduct(
        JsonPath.read(response, "$.insertProducts.value"),
        ID,
        "Shiny Legs",
        "3199.99",
        "2011-02-02T20:05:00.000-08:00",
        "Normal legs but shiny.");
  }

  @Test
  public void insertProductIfNotExistsNotApplied() {
    // Given a product that is already present
    insertProduct(
        ID, "Shiny Legs", "3199.99", "2011-02-02T20:05:00.000-08:00", "Normal legs but shiny.");

    // When trying to insert same id but different values
    Map response =
        insertProductIfNotExists(
            ID, "Shiny Legs", "3199.99", "2011-02-02T20:05:00.000-08:00", "New description.");

    // Then insert is not applied, and returns the original values
    assertThat(JsonPath.read(response, "$.insertProducts.applied")).isFalse();
    assertIsProduct(
        JsonPath.read(response, "$.insertProducts.value"),
        ID,
        "Shiny Legs",
        "3199.99",
        "2011-02-02T20:05:00.000-08:00",
        "Normal legs but shiny.");
  }

  private Map insertProduct(
      String id, String name, String price, String created, String description) {
    return insertProduct(id, name, price, created, description, false);
  }

  private Map insertProductIfNotExists(
      String id, String name, String price, String created, String description) {
    return insertProduct(id, name, price, created, description, true);
  }

  private Map insertProduct(
      String id,
      String name,
      String price,
      String created,
      String description,
      boolean ifNotExists) {
    return CLIENT.executeDmlQuery(
        KEYSPACE_ID,
        String.format(
            "mutation {\n"
                + "  insertProducts(\n"
                + "    value: {\n"
                + "      id: \"%s\"\n"
                + "      name: \"%s\"\n"
                + "      price: \"%s\"\n"
                + "      created: \"%s\"\n"
                + "      description: \"%s\"\n"
                + "    }\n,"
                + "    ifNotExists: %s"
                + "  ) {\n"
                + "    applied\n"
                + "    value { id, name, price, created, description }"
                + "  }\n"
                + "}",
            id, name, price, created, description, ifNotExists));
  }

  private Map selectProduct(String id) {
    return CLIENT.executeDmlQuery(
        KEYSPACE_ID,
        String.format(
            "{\n"
                + "  Products(\n"
                + "    value: { id: \"%s\" }"
                + "  ) {\n"
                + "    values { id, name, price, created, description }"
                + "  }\n"
                + "}",
            id));
  }
}