io.github.databob.CollectionGenerator.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of databob.kotlin Show documentation

Kill all builders!

The newest version!

package io.github.databob

import java.lang.reflect.Type
import java.util.*
import java.util.stream.Stream

class CollectionGenerator : Generator {

    /**
     * Pre-packed generator instances
     */
    object instances {
        val random = CompositeGenerator(listOf(
            Generators.ofType { d ->
                if (d.mk().toss()) {
                    CollectionSizeRange(1, 5)
                } else CollectionSizeRange(0, 0)
            },
            CollectionGenerator())
        )
        val empty = CompositeGenerator(listOf(
            Generators.ofType { -> CollectionSizeRange(0, 0) },
            CollectionGenerator())
        )
        val nonEmpty = CompositeGenerator(listOf(
            Generators.ofType { -> CollectionSizeRange(1, 1) },
            CollectionGenerator())
        )
    }

    private fun ctr(databob: Databob, type: Type): Any = databob.mk(Class.forName(type.typeName).kotlin)

    private inline fun  construct(databob: Databob, ctrFn: () -> T): Array {
        return databob.mk().toRandomRange().map { ctrFn() }.toTypedArray()
    }

    private val generator = CompositeGenerator(
        Generators.ofType { t, d -> setOf(*construct(d) { ctr(d, t[0]) }) },
        Generators.ofType { t, d -> listOf(*construct(d) { ctr(d, t[0]) }) },
        Generators.ofType { t, d -> Vector(listOf(*construct(d) { ctr(d, t[0]) })) },
        Generators.ofType { t, d -> Stream.of(*construct(d) { ctr(d, t[0]) }) },
        Generators.ofType { t, d -> mapOf(*construct(d) { Pair(ctr(d, t[0]), ctr(d, t[1])) }) }
    )

    override fun mk(type: Type, databob: Databob): Any? = generator.mk(type, databob)
}