DoubleArrays_assertContainsOnly_Test.java

/*
 * Created on Mar 29, 2013
 *
 * 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 @2013 the original author or authors.
 */
package org.fest.assertions.internal;

import static org.fest.assertions.error.ShouldContainOnly.shouldContainOnly;
import static org.fest.assertions.test.TestFailures.expectedAssertionErrorNotThrown;
import static org.fest.test.ExpectedException.none;
import static org.fest.util.Sets.newLinkedHashSet;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;

import java.util.Set;

import org.fest.assertions.description.Description;
import org.fest.assertions.error.ShouldNotBeNull;
import org.fest.test.ExpectedException;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

/**
 * @author Alex
 * @author Alex
 */
public class DoubleArrays_assertContainsOnly_Test {

  @Rule
  public ExpectedException thrown = none();

  private DoubleArrays arrays;
  private Description description;
  private Failures failures;

  @Before
  public void setUp() {
    arrays = new DoubleArrays();
    description = new TestDescription("testing");
    failures = spy(new Failures());
    Arrays parent = Arrays.instance();
    parent.failures = failures;
  }

  @Test
  public void should_fail_if_actual_is_null() {
    String message = ShouldNotBeNull.shouldNotBeNull().create(description);
    double[] other = new double[1];
    thrown.expect(AssertionError.class, message);
    arrays.assertContainsOnly(description, null, other);
  }

  @Test
  public void should_fail_if_other_is_null() {
    double[] actual = new double[1];
    thrown.expect(NullPointerException.class, "The array of values to look for should not be null");
    arrays.assertContainsOnly(description, actual, null);
  }

  @Test
  public void should_fail_if_other_is_empty() {
    double[] actual = new double[1];
    double[] other = new double[0];
    thrown.expect(IllegalArgumentException.class, "The array of values to look for should not be empty");
    arrays.assertContainsOnly(description, actual, other);
  }

  @Test
  public void should_fail_if_actual_does_not_contain_only_other() {
    double[] actual = { 1d, 2d, 3d, 4d };
    double[] other = { 1d, 2d, 3d };
    Set<Double> notExpected = asSet(actual);
    Set<Double> notFound = containsOnly(notExpected, other);
    try {
      arrays.assertContainsOnly(description, actual, other);
    } catch (AssertionError e) {
      verify(failures).failure(description, shouldContainOnly(actual, other, notFound, notExpected));
      return;
    }
    throw expectedAssertionErrorNotThrown();
  }

  @Test
  public void should_pass_if_actual_contains_only_other_in_any_order() {
    double[] actual = { 1d, 2d, 3d, 4d };
    double[] other = { 2d, 1d, 3d, 4d };
    arrays.assertContainsOnly(description, actual, other);
  }

  @Test
  public void should_pass_if_actual_contains_only_other_with_duplicates() {
    double[] actual = { 1d, 2d, 3d, 4d };
    double[] other = { 2d, 1d, 3d, 4d, 1d, 2d, 3d };
    arrays.assertContainsOnly(description, actual, other);
  }

  private Set<Double> containsOnly(Set<Double> actual, double[] other) {
    Set<Double> notExpected = newLinkedHashSet();
    Set<Double> notFound = newLinkedHashSet();
    notExpected = actual;
    for (double e : other) {
      if (notExpected.contains(e)) {
        notExpected.remove(e);
      } else {
        notFound.add(e);
      }
    }
    return notFound;
  }

  private Set<Double> asSet(double[] actual) {
    Set<Double> set = newLinkedHashSet();
    for (double e : actual) {
      set.add(e);
    }
    return set;
  }
}