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; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.features.CollectionSize; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.ArrayList; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Arrays; 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Collection; 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.List; 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Generator for collection of a particular size. 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * <p>This class is GWT compatible. 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author George van den Driessche 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic final class OneSizeGenerator<T, E> 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert implements OneSizeTestContainerGenerator<T, E> { 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private final TestContainerGenerator<T, E> generator; 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private final CollectionSize collectionSize; 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public OneSizeGenerator(TestContainerGenerator<T, E> generator, 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert CollectionSize collectionSize) { 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert this.generator = generator; 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert this.collectionSize = collectionSize; 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public TestContainerGenerator<T, E> getInnerGenerator() { 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator; 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public SampleElements<E> samples() { 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator.samples(); 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public T create(Object... elements) { 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator.create(elements); 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public E[] createArray(int length) { 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator.createArray(length); 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public T createTestSubject() { 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Collection<E> elements = getSampleElements( 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert getCollectionSize().getNumElements()); 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator.create(elements.toArray()); 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public Collection<E> getSampleElements(int howMany) { 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert SampleElements<E> samples = samples(); 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @SuppressWarnings("unchecked") 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert List<E> allSampleElements = Arrays.asList( 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert samples.e0, samples.e1, samples.e2, samples.e3, samples.e4); 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return new ArrayList<E>(allSampleElements.subList(0, howMany)); 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public CollectionSize getCollectionSize() { 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return collectionSize; 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public Iterable<E> order(List<E> insertionOrder) { 871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return generator.order(insertionOrder); 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 90