11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2008 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.collect.testing.google; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 197dd252788645e940eada959bdde927426e2531c9Paul Duffinimport static com.google.common.base.Preconditions.checkArgument; 207dd252788645e940eada959bdde927426e2531c9Paul Duffin 217dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.Multiset; 227dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.Multiset.Entry; 237dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.Multisets; 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.AbstractCollectionTestSuiteBuilder; 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.AbstractTester; 267dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.FeatureSpecificTestSuiteBuilder; 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.Helpers; 287dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.OneSizeTestContainerGenerator; 297dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.SampleElements; 307dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.SetTestSuiteBuilder; 317dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.TestSetGenerator; 327dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.features.CollectionFeature; 337dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.features.Feature; 347dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.collect.testing.testers.CollectionSerializationEqualTester; 357dd252788645e940eada959bdde927426e2531c9Paul Duffinimport com.google.common.testing.SerializableTester; 367dd252788645e940eada959bdde927426e2531c9Paul Duffin 377dd252788645e940eada959bdde927426e2531c9Paul Duffinimport junit.framework.TestSuite; 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 397dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.ArrayList; 407dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.Collection; 417dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.Collections; 427dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.HashSet; 437dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.LinkedHashMap; 447dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.LinkedHashSet; 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.List; 467dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.Map; 477dd252788645e940eada959bdde927426e2531c9Paul Duffinimport java.util.Set; 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Creates, based on your criteria, a JUnit test suite that exhaustively tests 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * a {@code Multiset} implementation. 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Jared Levy 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Louis Wasserman 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class MultisetTestSuiteBuilder<E> extends 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert AbstractCollectionTestSuiteBuilder<MultisetTestSuiteBuilder<E>, E> { 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public static <E> MultisetTestSuiteBuilder<E> using( 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert TestMultisetGenerator<E> generator) { 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return new MultisetTestSuiteBuilder<E>().usingGenerator(generator); 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 637dd252788645e940eada959bdde927426e2531c9Paul Duffin public enum NoRecurse implements Feature<Void> { 647dd252788645e940eada959bdde927426e2531c9Paul Duffin NO_ENTRY_SET; 657dd252788645e940eada959bdde927426e2531c9Paul Duffin 667dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 677dd252788645e940eada959bdde927426e2531c9Paul Duffin public Set<Feature<? super Void>> getImpliedFeatures() { 687dd252788645e940eada959bdde927426e2531c9Paul Duffin return Collections.emptySet(); 697dd252788645e940eada959bdde927426e2531c9Paul Duffin } 707dd252788645e940eada959bdde927426e2531c9Paul Duffin } 717dd252788645e940eada959bdde927426e2531c9Paul Duffin 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override protected List<Class<? extends AbstractTester>> getTesters() { 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert List<Class<? extends AbstractTester>> testers 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert = Helpers.copyToList(super.getTesters()); 757dd252788645e940eada959bdde927426e2531c9Paul Duffin testers.add(CollectionSerializationEqualTester.class); 760888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetAddTester.class); 770888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetContainsTester.class); 780888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetCountTester.class); 790888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetElementSetTester.class); 800888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetEqualsTester.class); 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert testers.add(MultisetReadsTester.class); 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert testers.add(MultisetSetCountConditionallyTester.class); 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert testers.add(MultisetSetCountUnconditionallyTester.class); 840888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetRemoveTester.class); 850888a09821a98ac0680fad765217302858e70fa4Paul Duffin testers.add(MultisetEntrySetTester.class); 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert testers.add(MultisetIteratorTester.class); 877dd252788645e940eada959bdde927426e2531c9Paul Duffin testers.add(MultisetSerializationTester.class); 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return testers; 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 907dd252788645e940eada959bdde927426e2531c9Paul Duffin 917dd252788645e940eada959bdde927426e2531c9Paul Duffin private static Set<Feature<?>> computeEntrySetFeatures( 927dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> features) { 937dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> derivedFeatures = new HashSet<Feature<?>>(); 947dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.addAll(features); 957dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.GENERAL_PURPOSE); 967dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SUPPORTS_ADD); 977dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.ALLOWS_NULL_VALUES); 987dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.add(CollectionFeature.REJECTS_DUPLICATES_AT_CREATION); 997dd252788645e940eada959bdde927426e2531c9Paul Duffin if (!derivedFeatures.remove(CollectionFeature.SERIALIZABLE_INCLUDING_VIEWS)) { 1007dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SERIALIZABLE); 1017dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1027dd252788645e940eada959bdde927426e2531c9Paul Duffin return derivedFeatures; 1037dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1047dd252788645e940eada959bdde927426e2531c9Paul Duffin 1057dd252788645e940eada959bdde927426e2531c9Paul Duffin static Set<Feature<?>> computeElementSetFeatures( 1067dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> features) { 1077dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> derivedFeatures = new HashSet<Feature<?>>(); 1087dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.addAll(features); 1097dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.GENERAL_PURPOSE); 1107dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SUPPORTS_ADD); 1117dd252788645e940eada959bdde927426e2531c9Paul Duffin if (!derivedFeatures.remove(CollectionFeature.SERIALIZABLE_INCLUDING_VIEWS)) { 1127dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SERIALIZABLE); 1137dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1147dd252788645e940eada959bdde927426e2531c9Paul Duffin return derivedFeatures; 1157dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1167dd252788645e940eada959bdde927426e2531c9Paul Duffin 1177dd252788645e940eada959bdde927426e2531c9Paul Duffin private static Set<Feature<?>> computeReserializedMultisetFeatures( 1187dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> features) { 1197dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<Feature<?>> derivedFeatures = new HashSet<Feature<?>>(); 1207dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.addAll(features); 1217dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SERIALIZABLE); 1227dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedFeatures.remove(CollectionFeature.SERIALIZABLE_INCLUDING_VIEWS); 1237dd252788645e940eada959bdde927426e2531c9Paul Duffin return derivedFeatures; 1247dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1257dd252788645e940eada959bdde927426e2531c9Paul Duffin 1267dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 1277dd252788645e940eada959bdde927426e2531c9Paul Duffin protected List<TestSuite> createDerivedSuites(FeatureSpecificTestSuiteBuilder< 1287dd252788645e940eada959bdde927426e2531c9Paul Duffin ?, ? extends OneSizeTestContainerGenerator<Collection<E>, E>> parentBuilder) { 1297dd252788645e940eada959bdde927426e2531c9Paul Duffin List<TestSuite> derivedSuites = new ArrayList<TestSuite>( 1307dd252788645e940eada959bdde927426e2531c9Paul Duffin super.createDerivedSuites(parentBuilder)); 1317dd252788645e940eada959bdde927426e2531c9Paul Duffin 1327dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedSuites.add(createElementSetTestSuite(parentBuilder)); 1337dd252788645e940eada959bdde927426e2531c9Paul Duffin 1347dd252788645e940eada959bdde927426e2531c9Paul Duffin if (!parentBuilder.getFeatures().contains(NoRecurse.NO_ENTRY_SET)) { 1357dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedSuites.add( 1367dd252788645e940eada959bdde927426e2531c9Paul Duffin SetTestSuiteBuilder.using(new EntrySetGenerator<E>(parentBuilder.getSubjectGenerator())) 1377dd252788645e940eada959bdde927426e2531c9Paul Duffin .named(getName() + ".entrySet") 1387dd252788645e940eada959bdde927426e2531c9Paul Duffin .withFeatures(computeEntrySetFeatures(parentBuilder.getFeatures())) 1397dd252788645e940eada959bdde927426e2531c9Paul Duffin .suppressing(parentBuilder.getSuppressedTests()) 1407dd252788645e940eada959bdde927426e2531c9Paul Duffin .createTestSuite()); 1417dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1427dd252788645e940eada959bdde927426e2531c9Paul Duffin 1437dd252788645e940eada959bdde927426e2531c9Paul Duffin if (parentBuilder.getFeatures().contains(CollectionFeature.SERIALIZABLE)) { 1447dd252788645e940eada959bdde927426e2531c9Paul Duffin derivedSuites.add(MultisetTestSuiteBuilder 1457dd252788645e940eada959bdde927426e2531c9Paul Duffin .using(new ReserializedMultisetGenerator<E>(parentBuilder.getSubjectGenerator())) 1467dd252788645e940eada959bdde927426e2531c9Paul Duffin .named(getName() + " reserialized") 1477dd252788645e940eada959bdde927426e2531c9Paul Duffin .withFeatures(computeReserializedMultisetFeatures(parentBuilder.getFeatures())) 1487dd252788645e940eada959bdde927426e2531c9Paul Duffin .suppressing(parentBuilder.getSuppressedTests()) 1497dd252788645e940eada959bdde927426e2531c9Paul Duffin .createTestSuite()); 1507dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1517dd252788645e940eada959bdde927426e2531c9Paul Duffin return derivedSuites; 1527dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1537dd252788645e940eada959bdde927426e2531c9Paul Duffin 1547dd252788645e940eada959bdde927426e2531c9Paul Duffin TestSuite createElementSetTestSuite(FeatureSpecificTestSuiteBuilder< 1557dd252788645e940eada959bdde927426e2531c9Paul Duffin ?, ? extends OneSizeTestContainerGenerator<Collection<E>, E>> parentBuilder) { 1567dd252788645e940eada959bdde927426e2531c9Paul Duffin return SetTestSuiteBuilder 1577dd252788645e940eada959bdde927426e2531c9Paul Duffin .using(new ElementSetGenerator<E>(parentBuilder.getSubjectGenerator())) 1587dd252788645e940eada959bdde927426e2531c9Paul Duffin .named(getName() + ".elementSet") 1597dd252788645e940eada959bdde927426e2531c9Paul Duffin .withFeatures(computeElementSetFeatures(parentBuilder.getFeatures())) 1607dd252788645e940eada959bdde927426e2531c9Paul Duffin .suppressing(parentBuilder.getSuppressedTests()) 1617dd252788645e940eada959bdde927426e2531c9Paul Duffin .createTestSuite(); 1627dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1637dd252788645e940eada959bdde927426e2531c9Paul Duffin 1647dd252788645e940eada959bdde927426e2531c9Paul Duffin static class ElementSetGenerator<E> implements TestSetGenerator<E> { 1657dd252788645e940eada959bdde927426e2531c9Paul Duffin final OneSizeTestContainerGenerator<Collection<E>, E> gen; 1667dd252788645e940eada959bdde927426e2531c9Paul Duffin 1677dd252788645e940eada959bdde927426e2531c9Paul Duffin ElementSetGenerator(OneSizeTestContainerGenerator<Collection<E>, E> gen) { 1687dd252788645e940eada959bdde927426e2531c9Paul Duffin this.gen = gen; 1697dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1707dd252788645e940eada959bdde927426e2531c9Paul Duffin 1717dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 1727dd252788645e940eada959bdde927426e2531c9Paul Duffin public SampleElements<E> samples() { 1737dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.samples(); 1747dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1757dd252788645e940eada959bdde927426e2531c9Paul Duffin 1767dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 1777dd252788645e940eada959bdde927426e2531c9Paul Duffin public Set<E> create(Object... elements) { 1787dd252788645e940eada959bdde927426e2531c9Paul Duffin Object[] duplicated = new Object[elements.length * 2]; 1797dd252788645e940eada959bdde927426e2531c9Paul Duffin for (int i = 0; i < elements.length; i++) { 1807dd252788645e940eada959bdde927426e2531c9Paul Duffin duplicated[i] = elements[i]; 1817dd252788645e940eada959bdde927426e2531c9Paul Duffin duplicated[i + elements.length] = elements[i]; 1827dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1837dd252788645e940eada959bdde927426e2531c9Paul Duffin return ((Multiset<E>) gen.create(duplicated)).elementSet(); 1847dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1857dd252788645e940eada959bdde927426e2531c9Paul Duffin 1867dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 1877dd252788645e940eada959bdde927426e2531c9Paul Duffin public E[] createArray(int length) { 1887dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.createArray(length); 1897dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1907dd252788645e940eada959bdde927426e2531c9Paul Duffin 1917dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 1927dd252788645e940eada959bdde927426e2531c9Paul Duffin public Iterable<E> order(List<E> insertionOrder) { 1937dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.order(new ArrayList<E>(new LinkedHashSet<E>(insertionOrder))); 1947dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1957dd252788645e940eada959bdde927426e2531c9Paul Duffin } 1967dd252788645e940eada959bdde927426e2531c9Paul Duffin 1977dd252788645e940eada959bdde927426e2531c9Paul Duffin static class EntrySetGenerator<E> implements TestSetGenerator<Multiset.Entry<E>> { 1987dd252788645e940eada959bdde927426e2531c9Paul Duffin final OneSizeTestContainerGenerator<Collection<E>, E> gen; 1997dd252788645e940eada959bdde927426e2531c9Paul Duffin 2007dd252788645e940eada959bdde927426e2531c9Paul Duffin private EntrySetGenerator(OneSizeTestContainerGenerator<Collection<E>, E> gen) { 2017dd252788645e940eada959bdde927426e2531c9Paul Duffin this.gen = gen; 2027dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2037dd252788645e940eada959bdde927426e2531c9Paul Duffin 2047dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2057dd252788645e940eada959bdde927426e2531c9Paul Duffin public SampleElements<Multiset.Entry<E>> samples() { 2067dd252788645e940eada959bdde927426e2531c9Paul Duffin SampleElements<E> samples = gen.samples(); 2077dd252788645e940eada959bdde927426e2531c9Paul Duffin return new SampleElements<Multiset.Entry<E>>( 2087dd252788645e940eada959bdde927426e2531c9Paul Duffin Multisets.immutableEntry(samples.e0, 3), 2097dd252788645e940eada959bdde927426e2531c9Paul Duffin Multisets.immutableEntry(samples.e1, 4), 2107dd252788645e940eada959bdde927426e2531c9Paul Duffin Multisets.immutableEntry(samples.e2, 1), 2117dd252788645e940eada959bdde927426e2531c9Paul Duffin Multisets.immutableEntry(samples.e3, 5), 2127dd252788645e940eada959bdde927426e2531c9Paul Duffin Multisets.immutableEntry(samples.e4, 2)); 2137dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2147dd252788645e940eada959bdde927426e2531c9Paul Duffin 2157dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2167dd252788645e940eada959bdde927426e2531c9Paul Duffin public Set<Multiset.Entry<E>> create(Object... entries) { 2177dd252788645e940eada959bdde927426e2531c9Paul Duffin List<Object> contents = new ArrayList<Object>(); 2187dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<E> elements = new HashSet<E>(); 2197dd252788645e940eada959bdde927426e2531c9Paul Duffin for (Object o : entries) { 2207dd252788645e940eada959bdde927426e2531c9Paul Duffin @SuppressWarnings("unchecked") 2217dd252788645e940eada959bdde927426e2531c9Paul Duffin Multiset.Entry<E> entry = (Entry<E>) o; 2227dd252788645e940eada959bdde927426e2531c9Paul Duffin checkArgument(elements.add(entry.getElement()), 2237dd252788645e940eada959bdde927426e2531c9Paul Duffin "Duplicate keys not allowed in EntrySetGenerator"); 2247dd252788645e940eada959bdde927426e2531c9Paul Duffin for (int i = 0; i < entry.getCount(); i++) { 2257dd252788645e940eada959bdde927426e2531c9Paul Duffin contents.add(entry.getElement()); 2267dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2277dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2287dd252788645e940eada959bdde927426e2531c9Paul Duffin return ((Multiset<E>) gen.create(contents.toArray())).entrySet(); 2297dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2307dd252788645e940eada959bdde927426e2531c9Paul Duffin 2317dd252788645e940eada959bdde927426e2531c9Paul Duffin @SuppressWarnings("unchecked") 2327dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2337dd252788645e940eada959bdde927426e2531c9Paul Duffin public Multiset.Entry<E>[] createArray(int length) { 2347dd252788645e940eada959bdde927426e2531c9Paul Duffin return new Multiset.Entry[length]; 2357dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2367dd252788645e940eada959bdde927426e2531c9Paul Duffin 2377dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2387dd252788645e940eada959bdde927426e2531c9Paul Duffin public Iterable<Entry<E>> order(List<Entry<E>> insertionOrder) { 2397dd252788645e940eada959bdde927426e2531c9Paul Duffin // We mimic the order from gen. 2407dd252788645e940eada959bdde927426e2531c9Paul Duffin Map<E, Entry<E>> map = new LinkedHashMap<E, Entry<E>>(); 2417dd252788645e940eada959bdde927426e2531c9Paul Duffin for (Entry<E> entry : insertionOrder) { 2427dd252788645e940eada959bdde927426e2531c9Paul Duffin map.put(entry.getElement(), entry); 2437dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2447dd252788645e940eada959bdde927426e2531c9Paul Duffin 2457dd252788645e940eada959bdde927426e2531c9Paul Duffin Set<E> seen = new HashSet<E>(); 2467dd252788645e940eada959bdde927426e2531c9Paul Duffin List<Entry<E>> order = new ArrayList<Entry<E>>(); 2477dd252788645e940eada959bdde927426e2531c9Paul Duffin for (E e : gen.order(new ArrayList<E>(map.keySet()))) { 2487dd252788645e940eada959bdde927426e2531c9Paul Duffin if (seen.add(e)) { 2497dd252788645e940eada959bdde927426e2531c9Paul Duffin order.add(map.get(e)); 2507dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2517dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2527dd252788645e940eada959bdde927426e2531c9Paul Duffin return order; 2537dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2547dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2557dd252788645e940eada959bdde927426e2531c9Paul Duffin 2567dd252788645e940eada959bdde927426e2531c9Paul Duffin static class ReserializedMultisetGenerator<E> implements TestMultisetGenerator<E>{ 2577dd252788645e940eada959bdde927426e2531c9Paul Duffin final OneSizeTestContainerGenerator<Collection<E>, E> gen; 2587dd252788645e940eada959bdde927426e2531c9Paul Duffin 2597dd252788645e940eada959bdde927426e2531c9Paul Duffin private ReserializedMultisetGenerator(OneSizeTestContainerGenerator<Collection<E>, E> gen) { 2607dd252788645e940eada959bdde927426e2531c9Paul Duffin this.gen = gen; 2617dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2627dd252788645e940eada959bdde927426e2531c9Paul Duffin 2637dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2647dd252788645e940eada959bdde927426e2531c9Paul Duffin public SampleElements<E> samples() { 2657dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.samples(); 2667dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2677dd252788645e940eada959bdde927426e2531c9Paul Duffin 2687dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2697dd252788645e940eada959bdde927426e2531c9Paul Duffin public Multiset<E> create(Object... elements) { 2707dd252788645e940eada959bdde927426e2531c9Paul Duffin return (Multiset<E>) SerializableTester.reserialize(gen.create(elements)); 2717dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2727dd252788645e940eada959bdde927426e2531c9Paul Duffin 2737dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2747dd252788645e940eada959bdde927426e2531c9Paul Duffin public E[] createArray(int length) { 2757dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.createArray(length); 2767dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2777dd252788645e940eada959bdde927426e2531c9Paul Duffin 2787dd252788645e940eada959bdde927426e2531c9Paul Duffin @Override 2797dd252788645e940eada959bdde927426e2531c9Paul Duffin public Iterable<E> order(List<E> insertionOrder) { 2807dd252788645e940eada959bdde927426e2531c9Paul Duffin return gen.order(insertionOrder); 2817dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2827dd252788645e940eada959bdde927426e2531c9Paul Duffin } 2831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 2841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 285