1 /* 2 * Copyright (C) 2007 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 import com.google.common.collect.testing.SampleElements.Enums; 21 22 import java.util.Collections; 23 import java.util.List; 24 import java.util.Set; 25 26 /** 27 * An abstract TestSetGenerator for generating sets containing enum values. 28 * 29 * @author Kevin Bourrillion 30 */ 31 @GwtCompatible 32 public abstract class TestEnumSetGenerator implements TestSetGenerator<AnEnum> { 33 @Override samples()34 public SampleElements<AnEnum> samples() { 35 return new Enums(); 36 } 37 38 @Override create(Object... elements)39 public Set<AnEnum> create(Object... elements) { 40 AnEnum[] array = new AnEnum[elements.length]; 41 int i = 0; 42 for (Object e : elements) { 43 array[i++] = (AnEnum) e; 44 } 45 return create(array); 46 } 47 create(AnEnum[] elements)48 protected abstract Set<AnEnum> create(AnEnum[] elements); 49 50 @Override createArray(int length)51 public AnEnum[] createArray(int length) { 52 return new AnEnum[length]; 53 } 54 55 /** 56 * Sorts the enums according to their natural ordering. 57 */ 58 @Override order(List<AnEnum> insertionOrder)59 public List<AnEnum> order(List<AnEnum> insertionOrder) { 60 Collections.sort(insertionOrder); 61 return insertionOrder; 62 } 63 } 64