• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2008 The Guava Authors
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.google.common.collect.testing;
18 
19 import com.google.common.annotations.GwtCompatible;
20 
21 import java.util.ArrayList;
22 import java.util.Arrays;
23 import java.util.Collection;
24 import java.util.Collections;
25 import java.util.List;
26 
27 /**
28  * Base class for testers of classes (including {@link Collection}
29  * and {@link java.util.Map Map}) that contain elements.
30  *
31  * @param <C> the type of the container
32  * @param <E> the type of the container's contents
33  *
34  * @author George van den Driessche
35  */
36 @GwtCompatible
37 public abstract class AbstractContainerTester<C, E>
38     extends AbstractTester<OneSizeTestContainerGenerator<C, E>> {
39   protected SampleElements<E> samples;
40   protected C container;
41 
setUp()42   @Override public void setUp() throws Exception {
43     super.setUp();
44     samples = this.getSubjectGenerator().samples();
45     resetContainer();
46   }
47 
48   /**
49    * @return the contents of the container under test, for use by
50    * {@link #expectContents(Object[]) expectContents(E...)} and its friends.
51    */
actualContents()52   protected abstract Collection<E> actualContents();
53 
54   /**
55    * Replaces the existing container under test with a new container created
56    * by the subject generator.
57    *
58    * @see #resetContainer(Object) resetContainer(C)
59    *
60    * @return the new container instance.
61    */
resetContainer()62   protected C resetContainer() {
63     return resetContainer(getSubjectGenerator().createTestSubject());
64   }
65 
66   /**
67    * Replaces the existing container under test with a new container.
68    * This is useful when a single test method needs to create multiple
69    * containers while retaining the ability to use
70    * {@link #expectContents(Object[]) expectContents(E...)} and other
71    * convenience methods. The creation of multiple containers in a single
72    * method is discouraged in most cases, but it is vital to the iterator tests.
73    *
74    * @return the new container instance
75    * @param newValue the new container instance
76    */
resetContainer(C newValue)77   protected C resetContainer(C newValue) {
78     container = newValue;
79     return container;
80   }
81 
82   /**
83    * @see #expectContents(java.util.Collection)
84    *
85    * @param elements expected contents of {@link #container}
86    */
expectContents(E... elements)87   protected final void expectContents(E... elements) {
88     expectContents(Arrays.asList(elements));
89   }
90 
91   /**
92    * Asserts that the collection under test contains exactly the given elements,
93    * respecting cardinality but not order. Subclasses may override this method
94    * to provide stronger assertions, e.g., to check ordering in lists, but
95    * realize that <strong>unless a test extends
96    * {@link com.google.common.collect.testing.testers.AbstractListTester
97    * AbstractListTester}, a call to {@code expectContents()} invokes this
98    * version</strong>.
99    *
100    * @param expected expected value of {@link #container}
101    */
102   /*
103    * TODO: improve this and other implementations and move out of this framework
104    * for wider use
105    *
106    * TODO: could we incorporate the overriding logic from AbstractListTester, by
107    * examining whether the features include KNOWN_ORDER?
108    */
expectContents(Collection<E> expected)109   protected void expectContents(Collection<E> expected) {
110     Helpers.assertEqualIgnoringOrder(expected, actualContents());
111   }
112 
expectUnchanged()113   protected void expectUnchanged() {
114     expectContents(getOrderedElements());
115   }
116 
117   /**
118    * Asserts that the collection under test contains exactly the elements it was
119    * initialized with plus the given elements, according to
120    * {@link #expectContents(java.util.Collection)}. In other words, for the
121    * default {@code expectContents()} implementation, the number of occurrences
122    * of each given element has increased by one since the test collection was
123    * created, and the number of occurrences of all other elements has not
124    * changed.
125    *
126    * <p>Note: This means that a test like the following will fail if
127    * {@code collection} is a {@code Set}:
128    *
129    * <pre>
130    * collection.add(existingElement);
131    * expectAdded(existingElement);</pre>
132    *
133    * <p>In this case, {@code collection} was not modified as a result of the
134    * {@code add()} call, and the test will fail because the number of
135    * occurrences of {@code existingElement} is unchanged.
136    *
137    * @param elements expected additional contents of {@link #container}
138    */
expectAdded(E... elements)139   protected final void expectAdded(E... elements) {
140     List<E> expected = Helpers.copyToList(getSampleElements());
141     expected.addAll(Arrays.asList(elements));
142     expectContents(expected);
143   }
144 
expectAdded(int index, E... elements)145   protected final void expectAdded(int index, E... elements) {
146     expectAdded(index, Arrays.asList(elements));
147   }
148 
expectAdded(int index, Collection<E> elements)149   protected final void expectAdded(int index, Collection<E> elements) {
150     List<E> expected = Helpers.copyToList(getSampleElements());
151     expected.addAll(index, elements);
152     expectContents(expected);
153   }
154 
155   /*
156    * TODO: if we're testing a list, we could check indexOf(). (Doing it in
157    * AbstractListTester isn't enough because many tests that run on lists don't
158    * extends AbstractListTester.) We could also iterate over all elements to
159    * verify absence
160    */
expectMissing(E... elements)161   protected void expectMissing(E... elements) {
162     for (E element : elements) {
163       assertFalse("Should not contain " + element,
164           actualContents().contains(element));
165     }
166   }
167 
createSamplesArray()168   protected E[] createSamplesArray() {
169     E[] array = getSubjectGenerator().createArray(getNumElements());
170     getSampleElements().toArray(array);
171     return array;
172   }
173 
createOrderedArray()174   protected E[] createOrderedArray() {
175     E[] array = getSubjectGenerator().createArray(getNumElements());
176     getOrderedElements().toArray(array);
177     return array;
178   }
179 
180   public static class ArrayWithDuplicate<E> {
181     public final E[] elements;
182     public final E duplicate;
183 
ArrayWithDuplicate(E[] elements, E duplicate)184     private ArrayWithDuplicate(E[] elements, E duplicate) {
185       this.elements = elements;
186       this.duplicate = duplicate;
187     }
188   }
189 
190   /**
191    * @return an array of the proper size with a duplicate element.
192    * The size must be at least three.
193    */
createArrayWithDuplicateElement()194   protected ArrayWithDuplicate<E> createArrayWithDuplicateElement() {
195     E[] elements = createSamplesArray();
196     E duplicate = elements[(elements.length / 2) - 1];
197     elements[(elements.length / 2) + 1] = duplicate;
198     return new ArrayWithDuplicate<E>(elements, duplicate);
199   }
200 
201   // Helper methods to improve readability of derived classes
202 
getNumElements()203   protected int getNumElements() {
204     return getSubjectGenerator().getCollectionSize().getNumElements();
205   }
206 
getSampleElements(int howMany)207   protected Collection<E> getSampleElements(int howMany) {
208     return getSubjectGenerator().getSampleElements(howMany);
209   }
210 
getSampleElements()211   protected Collection<E> getSampleElements() {
212     return getSampleElements(getNumElements());
213   }
214 
215   /**
216    * Returns the {@linkplain #getSampleElements() sample elements} as ordered by
217    * {@link TestContainerGenerator#order(List)}. Tests should used this method
218    * only if they declare requirement {@link
219    * com.google.common.collect.testing.features.CollectionFeature#KNOWN_ORDER}.
220    */
getOrderedElements()221   protected List<E> getOrderedElements() {
222     List<E> list = new ArrayList<E>();
223     for (E e : getSubjectGenerator().order(
224         new ArrayList<E>(getSampleElements()))) {
225       list.add(e);
226     }
227     return Collections.unmodifiableList(list);
228   }
229 
230   /**
231    * @return a suitable location for a null element, to use when initializing
232    * containers for tests that involve a null element being present.
233    */
getNullLocation()234   protected int getNullLocation() {
235     return getNumElements() / 2;
236   }
237 
238   @SuppressWarnings("unchecked")
createDisjointCollection()239   protected MinimalCollection<E> createDisjointCollection() {
240     return MinimalCollection.of(samples.e3, samples.e4);
241   }
242 
243   @SuppressWarnings("unchecked")
emptyCollection()244   protected MinimalCollection<E> emptyCollection() {
245     return MinimalCollection.<E>of();
246   }
247 }
248