001/* 002 * Copyright (C) 2007 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.SampleElements.Enums; 021import java.util.Collections; 022import java.util.List; 023import java.util.Set; 024 025/** 026 * An abstract TestSetGenerator for generating sets containing enum values. 027 * 028 * @author Kevin Bourrillion 029 */ 030@GwtCompatible 031public abstract class TestEnumSetGenerator implements TestSetGenerator<AnEnum> { 032 @Override 033 public SampleElements<AnEnum> samples() { 034 return new Enums(); 035 } 036 037 @Override 038 public Set<AnEnum> create(Object... elements) { 039 AnEnum[] array = new AnEnum[elements.length]; 040 int i = 0; 041 for (Object e : elements) { 042 array[i++] = (AnEnum) e; 043 } 044 return create(array); 045 } 046 047 protected abstract Set<AnEnum> create(AnEnum[] elements); 048 049 @Override 050 public AnEnum[] createArray(int length) { 051 return new AnEnum[length]; 052 } 053 054 /** Sorts the enums according to their natural ordering. */ 055 @Override 056 public List<AnEnum> order(List<AnEnum> insertionOrder) { 057 Collections.sort(insertionOrder); 058 return insertionOrder; 059 } 060}