io.stargate.it.http.graphql.cqlfirst.AsyncDirectiveTest 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.Map;
import java.util.UUID;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;

public class AsyncDirectiveTest extends BetterbotzTestBase {

  private static final String FIRST_PRODUCT_ID = UUID.randomUUID().toString();
  private static final String SECOND_PRODUCT_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", FIRST_PRODUCT_ID));
    session.execute(String.format("DELETE FROM \"Products\" WHERE id = %s", SECOND_PRODUCT_ID));
  }

  @Test
  @DisplayName("Should insert using async directive")
  public void shouldInsertWhenUsingAsyncDirective() {
    Map response = insertProductWithAsyncDirective(FIRST_PRODUCT_ID);
    assertThat(JsonPath.read(response, "$.insertProducts.accepted")).isTrue();
  }

  @Test
  @DisplayName("Should insert using both async and atomic directives")
  public void shouldInsertWhenUsingBothAsyncAndAtomicDirectives() {
    Map response = insertProductWithAsyncAndAtomicDirectives(FIRST_PRODUCT_ID);
    assertThat(JsonPath.read(response, "$.insertProducts.accepted")).isTrue();
  }

  @Test
  @DisplayName("Should bulk insert using async directive")
  public void shouldBulkInsertWhenUsingAsyncDirective() {
    Map response =
        bulkInsertProductsWithAsyncDirective(FIRST_PRODUCT_ID, SECOND_PRODUCT_ID);
    assertThat(JsonPath.read(response, "$.bulkInsertProducts[0].accepted")).isTrue();
    assertThat(JsonPath.read(response, "$.bulkInsertProducts[1].accepted")).isTrue();
  }

  @Test
  @DisplayName("Should bulk insert using both async and atomic directives")
  public void shouldBulkInsertWhenUsingBothAsyncAndAtomicDirective() {
    Map response =
        bulkInsertProductsWithAsyncAndAtomicDirectives(FIRST_PRODUCT_ID, SECOND_PRODUCT_ID);
    assertThat(JsonPath.read(response, "$.bulkInsertProducts[0].accepted")).isTrue();
    assertThat(JsonPath.read(response, "$.bulkInsertProducts[1].accepted")).isTrue();
  }

  private Map bulkInsertProductsWithAsyncAndAtomicDirectives(
      String firstProductId, String secondProductId) {
    return bulkInsertProductWithDirectives(firstProductId, secondProductId, "@async @atomic");
  }

  private Map bulkInsertProductsWithAsyncDirective(
      String firstProductId, String secondProductId) {
    return bulkInsertProductWithDirectives(firstProductId, secondProductId, "@async");
  }

  private Map insertProductWithAsyncDirective(String productId) {
    return insertProductWithDirectives(productId, "@async");
  }

  private Map insertProductWithAsyncAndAtomicDirectives(String productId) {
    return insertProductWithDirectives(productId, "@async @atomic");
  }

  private Map insertProductWithDirectives(String id, String directives) {
    return CLIENT.executeDmlQuery(
        KEYSPACE_ID,
        String.format(
            "mutation %s {\n"
                + "  insertProducts(\n"
                + "    value: {\n"
                + "      id: \"%s\"\n"
                + "      name: \"%s\"\n"
                + "      price: \"%s\"\n"
                + "      created: \"%s\"\n"
                + "      description: \"%s\"\n"
                + "    }\n,"
                + "    ifNotExists: true"
                + "  ) {\n"
                + "    accepted\n"
                + "  }\n"
                + "}",
            directives,
            id,
            "Shiny Legs",
            "3199.99",
            "2011-02-02T20:05:00.000-08:00",
            "Normal legs but shiny."));
  }

  private Map bulkInsertProductWithDirectives(
      String firstProductId, String secondProductId, String directives) {
    return CLIENT.executeDmlQuery(
        KEYSPACE_ID,
        String.format(
            "mutation %s {\n"
                + "  bulkInsertProducts(\n"
                + "    values: [{\n"
                + "      id: \"%s\"\n"
                + "      name: \"%s\"\n"
                + "      price: \"%s\"\n"
                + "      created: \"%s\"\n"
                + "      description: \"%s\"\n"
                + "    }, \n"
                + "    {\n"
                + "      id: \"%s\"\n"
                + "      name: \"%s\"\n"
                + "      price: \"%s\"\n"
                + "      created: \"%s\"\n"
                + "      description: \"%s\"\n"
                + "    }\n,"
                + "]\n,"
                + "    ifNotExists: true"
                + "  ) {\n"
                + "    accepted\n"
                + "  }\n"
                + "}",
            directives,
            firstProductId,
            "Shiny Legs",
            "3199.99",
            "2011-02-02T20:05:00.000-08:00",
            "Normal legs but shiny.",
            secondProductId,
            "Other product",
            "3000.99",
            "2012-02-02T20:05:00.000-08:00",
            "Other legs."));
  }
}