All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.assertj.db.api.assertions.impl.AssertionsOnColumnContent Maven / Gradle / Ivy

Go to download

AssertJ-DB - Rich and fluent assertions for testing with database

There is a newer version: 2.0.2
Show newest version
/*
 * 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.
 *
 * Copyright 2015-2020 the original author or authors.
 */
package org.assertj.db.api.assertions.impl;

import org.assertj.core.api.WritableAssertionInfo;
import org.assertj.core.internal.Failures;
import org.assertj.db.api.AbstractAssert;
import org.assertj.db.type.*;
import org.assertj.db.util.Values;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import static org.assertj.db.error.ShouldContainsValue.shouldContainsValue;

/**
 * Implements the assertion methods on the content of a column.
 *
 * @author Régis Pouiller
 * @see org.assertj.db.api.assertions.AssertOnColumnContent
 * @since 1.1.0
 */
public class AssertionsOnColumnContent {

  /**
   * To notice failures in the assertion.
   */
  private static final Failures failures = Failures.instance();

  /**
   * Private constructor.
   */
  private AssertionsOnColumnContent() {
    // Empty
  }

  /**
   * Verifies that the column contains objects.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected object values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the objects in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, Object... expected) {
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (Object val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, Object.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains booleans.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected boolean values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the booleans in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, Boolean... expected) {
    AssertionsOnColumnType.isBoolean(assertion, info, valuesList, true);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (Boolean val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, Boolean.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains numbers.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected numbers values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the numbers in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, Number... expected) {
    AssertionsOnColumnType.isNumber(assertion, info, valuesList, true);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (Number val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, Number.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains bytes.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected bytes values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the bytes in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, byte[]... expected) {
    AssertionsOnColumnType.isBytes(assertion, info, valuesList, true);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (byte[] val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        throw failures.failure(info, shouldContainsValue(index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains texts.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected text values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the texts in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, String... expected) {
    AssertionsOnColumnType.isOfAnyTypeIn(assertion, info, valuesList, ValueType.TEXT, ValueType.NUMBER, ValueType.DATE,
                                         ValueType.TIME, ValueType.DATE_TIME, ValueType.UUID, ValueType.NOT_IDENTIFIED);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (String val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, String.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains characters.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected character values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the characters in parameter.
   * @since 1.2.0
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, Character... expected) {
    AssertionsOnColumnType.isText(assertion, info, valuesList, true);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (Character val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, Character.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains UUIDs.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected UUIDs values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the UUIDs in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, UUID... expected) {
    AssertionsOnColumnType.isOfAnyTypeIn(assertion, info, valuesList, ValueType.UUID, ValueType.NOT_IDENTIFIED);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (UUID val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, UUID.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains date values.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected date values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the date values in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, DateValue... expected) {
    AssertionsOnColumnType
            .isOfAnyTypeIn(assertion, info, valuesList, ValueType.DATE, ValueType.DATE_TIME, ValueType.NOT_IDENTIFIED);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (DateValue val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, DateValue.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains time values.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected time values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the time values in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, TimeValue... expected) {
    AssertionsOnColumnType.isOfAnyTypeIn(assertion, info, valuesList, ValueType.TIME, ValueType.NOT_IDENTIFIED);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (TimeValue val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, TimeValue.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }

  /**
   * Verifies that the column contains date/time values.
   *
   * @param         The type of the assertion which call this method.
   * @param assertion  The assertion which call this method.
   * @param info       Writable information about an assertion.
   * @param valuesList The list of values.
   * @param expected The expected date/time values.
   * @return {@code this} assertion object.
   * @throws AssertionError If the column are not containing the date/time values in parameter.
   */
  public static > A containsValues(A assertion, WritableAssertionInfo info,
                                                       List valuesList, DateTimeValue... expected) {
    AssertionsOnColumnType.isOfAnyTypeIn(assertion, info, valuesList, ValueType.DATE, ValueType.DATE_TIME,
                                         ValueType.NOT_IDENTIFIED);
    AssertionsOnNumberOfRows.hasNumberOfRows(assertion, info, valuesList.size(), expected.length);
    List list = new ArrayList<>(valuesList);
    int index = 0;
    for (DateTimeValue val : expected) {
      boolean found = false;
      List newList = new ArrayList<>();
      for (Value obj : list) {
        if (found || !Values.areEqual(obj, val)) {
          newList.add(obj);
        }
        else {
          found = true;
        }
      }
      if (!found) {
        List listForError = new ArrayList<>();
        for (Value obj : valuesList) {
          listForError.add(Values.getRepresentationFromValueInFrontOfExpected(obj, DateTimeValue.class));
        }
        throw failures.failure(info, shouldContainsValue(listForError, expected, val, index));
      }
      list = newList;
      index++;
    }
    return assertion;
  }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy