TestEnumSetGenerator.java revision dbd967a6e5c96cc1a97c5521f88dc1564ba2f81b 
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
17package com.google.common.collect.testing;
18
19import com.google.common.collect.testing.SampleElements.Enums;
20
21import java.util.Collections;
22import java.util.List;
23import java.util.Set;
24
25/**
26 * An abstract TestSetGenerator for generating sets containing enum values.
27 *
28 * <p>This class is GWT compatible.
29 *
30 * @author Kevin Bourrillion
31 */
32public 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