001/* 002 * Copyright (C) 2008 The Guava Authors 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package com.google.common.collect.testing; 018 019import com.google.common.annotations.GwtCompatible; 020import com.google.common.collect.testing.features.CollectionSize; 021import java.util.ArrayList; 022import java.util.Arrays; 023import java.util.Collection; 024import java.util.List; 025import org.checkerframework.checker.nullness.qual.Nullable; 026 027/** 028 * Generator for collection of a particular size. 029 * 030 * @author George van den Driessche 031 */ 032@GwtCompatible 033@ElementTypesAreNonnullByDefault 034public final class OneSizeGenerator<T, E extends @Nullable Object> 035 implements OneSizeTestContainerGenerator<T, E> { 036 private final TestContainerGenerator<T, E> generator; 037 private final CollectionSize collectionSize; 038 039 public OneSizeGenerator(TestContainerGenerator<T, E> generator, CollectionSize collectionSize) { 040 this.generator = generator; 041 this.collectionSize = collectionSize; 042 } 043 044 @Override 045 public TestContainerGenerator<T, E> getInnerGenerator() { 046 return generator; 047 } 048 049 @Override 050 public SampleElements<E> samples() { 051 return generator.samples(); 052 } 053 054 @Override 055 public T create(Object... elements) { 056 return generator.create(elements); 057 } 058 059 @Override 060 public E[] createArray(int length) { 061 return generator.createArray(length); 062 } 063 064 @Override 065 public T createTestSubject() { 066 Collection<E> elements = getSampleElements(getCollectionSize().getNumElements()); 067 return generator.create(elements.toArray()); 068 } 069 070 @Override 071 public Collection<E> getSampleElements(int howMany) { 072 SampleElements<E> samples = samples(); 073 List<E> allSampleElements = 074 Arrays.asList(samples.e0(), samples.e1(), samples.e2(), samples.e3(), samples.e4()); 075 return new ArrayList<>(allSampleElements.subList(0, howMany)); 076 } 077 078 @Override 079 public CollectionSize getCollectionSize() { 080 return collectionSize; 081 } 082 083 @Override 084 public Iterable<E> order(List<E> insertionOrder) { 085 return generator.order(insertionOrder); 086 } 087}