guava

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
import java.util.Collections;
22
import java.util.List;
23
import java.util.Set;
24

25
/**
26
 * An abstract TestSetGenerator for generating sets containing enum values.
27
 *
28
 * @author Kevin Bourrillion
29
 */
30
@GwtCompatible
31
@ElementTypesAreNonnullByDefault
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
  /** Sorts the enums according to their natural ordering. */
56
  @Override
57
  public List<AnEnum> order(List<AnEnum> insertionOrder) {
58
    Collections.sort(insertionOrder);
59
    return insertionOrder;
60
  }
61
}
62