spock.genesis.generators.values.DateGenerator.groovy Maven / Gradle / Ivy

Go to download
package spock.genesis.generators.values

import groovy.transform.CompileStatic
import spock.genesis.generators.InfiniteGenerator
import spock.genesis.generators.InfiniteIterator

@CompileStatic
class DateGenerator extends InfiniteGenerator {

    final LongGenerator millisProvider

    DateGenerator() {
        this.millisProvider = new LongGenerator()
    }

    DateGenerator(Date minDate, Date maxDate) {
        this.millisProvider = new LongGenerator(minDate.time, maxDate.time)
    }

    InfiniteIterator iterator() {
        final Iterator TIME = millisProvider.iterator()
        new InfiniteIterator() {
            @Override
            Date next() {
                new Date(TIME.next())
            }
        }
    }

    @Override
    DateGenerator seed(Long seed) {
        super.seed(seed)
        millisProvider.seed(seed)
        this
    }
}