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