UTBotJava/utbot-js/src/main/kotlin/fuzzer/JsFuzzing.kt at 0c5a4ce328b0cd08da074ff5a5b771d9f99a260d · UnitTestBot/UTBotJava · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package fuzzer

import framework.api.js.JsClassId
import fuzzer.providers.ArrayValueProvider
import fuzzer.providers.BoolValueProvider
import fuzzer.providers.MapValueProvider
import fuzzer.providers.MultipleValueProvider
import fuzzer.providers.NumberValueProvider
import fuzzer.providers.ObjectValueProvider
import fuzzer.providers.SetValueProvider
import fuzzer.providers.StringValueProvider
import org.utbot.framework.plugin.api.UtModel
import org.utbot.fuzzing.Fuzzing
import org.utbot.fuzzing.Seed
import org.utbot.fuzzing.fuzz

fun defaultValueProviders() = listOf(
    BoolValueProvider,
    NumberValueProvider,
    StringValueProvider,
    MapValueProvider,
    SetValueProvider,
    MultipleValueProvider,
    ObjectValueProvider(),
    ArrayValueProvider()
)

class JsFuzzing(
    val exec: suspend (JsMethodDescription, List<UtModel>) -> JsFeedback
) : Fuzzing<JsClassId, UtModel, JsMethodDescription, JsFeedback> {

    override fun generate(description: JsMethodDescription, type: JsClassId): Sequence<Seed<JsClassId, UtModel>> {
        return defaultValueProviders().asSequence().flatMap { provider ->
            if (provider.accept(type)) {
                provider.generate(description, type)
            } else {
                emptySequence()
            }
        }
    }

    override suspend fun handle(description: JsMethodDescription, values: List<UtModel>): JsFeedback {
        return exec(description, values)
    }
}

suspend fun runFuzzing(
    description: JsMethodDescription,
    exec: suspend (JsMethodDescription, List<UtModel>) -> JsFeedback
) = JsFuzzing(exec).fuzz(description)