001/*
002 * Licensed to the author under one or more
003 * contributor license agreements. See the NOTICE file distributed with
004 * this work for additional information regarding copyright ownership.
005 * The ASF licenses this file to You under the Apache License, Version 2.0
006 * (the "License"); you may not use this file except in compliance with
007 * the License. You may obtain a copy of the License at
008 *
009 * http://www.apache.org/licenses/LICENSE-2.0
010 *
011 * Unless required by applicable law or agreed to in writing, software
012 * distributed under the License is distributed on an "AS IS" BASIS,
013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014 * See the License for the specific language governing permissions and
015 * limitations under the License.
016 */
017package de.cuioss.test.generator.internal.net.java.quickcheck.generator.support;
018
019import java.util.HashSet;
020import java.util.Set;
021
022import de.cuioss.test.generator.internal.net.java.quickcheck.Generator;
023import de.cuioss.test.generator.internal.net.java.quickcheck.StatefulGenerator;
024
025public class SetGenerator<T> implements Generator<Set<T>> {
026
027    public static final int MAX_SIZE = ListGenerator.MAX_SIZE;
028    public static final int MAX_TRIES = VetoableGenerator.DEFAULT_MAX_TRIES;
029    private final ListGenerator<T> lists;
030    private final StatefulGenerator<T> content;
031
032    public SetGenerator(Generator<? extends T> content) {
033        this(content, MAX_TRIES);
034    }
035
036    public SetGenerator(Generator<? extends T> content, int tries) {
037        this(content, new IntegerGenerator(0, MAX_SIZE), tries);
038    }
039
040    public SetGenerator(Generator<? extends T> content, Generator<Integer> size, int tries) {
041        this.content = new UniqueValuesGenerator<>(content, tries);
042        this.lists = new ListGenerator<>(this.content, size);
043    }
044
045    @Override
046    public Set<T> next() {
047        content.reset();
048        return new HashSet<>(lists.next());
049    }
050}