Home | History | Annotate | Download | only in testing
      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
     34   public SampleElements<AnEnum> samples() {
     35     return new Enums();
     36   }
     37 
     38   @Override
     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 
     48   protected abstract Set<AnEnum> create(AnEnum[] elements);
     49 
     50   @Override
     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
     59   public List<AnEnum> order(List<AnEnum> insertionOrder) {
     60     Collections.sort(insertionOrder);
     61     return insertionOrder;
     62   }
     63 }
     64