kotlin-fork/kotlin-native/backend.native/tests/runtime/basic/worker_random.kt

/*
 * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
 * that can be found in the LICENSE file.
 */

package runtime.basic.worker_random

import kotlin.native.concurrent.*
import kotlin.collections.*
import kotlin.random.*
import kotlin.system.*
import kotlin.test.*

@Test
fun testRandomWorkers() {
    val seed = getTimeMillis()
    val workers = Array(5, { _ -> Worker.start() })

    val attempts = 3
    val results = Array(attempts, { ArrayList<Int>() } )
    for (attempt in 0 until attempts) {
        // Produce a list of random numbers in each worker
        val futures = Array(workers.size, { workerIndex ->
            workers[workerIndex].execute(TransferMode.SAFE, { workerIndex }) {
                input ->
                Array(10, { Random.nextInt() }).toList()
            }
        })
        // Now collect all results into current attempt's list
        val futureSet = futures.toSet()
        var finished = 0
        while (finished < futureSet.size) {
            val ready = waitForMultipleFutures(futureSet, 10000)
            ready.forEach { results[attempt].addAll(it.result) }
            finished += ready.size
        }
    }

    workers.forEach {
        it.requestTermination().result
    }
}