mirror of https://github.com/sbt/sbt.git
45 lines
1.4 KiB
Scala
45 lines
1.4 KiB
Scala
/*
|
|
* sbt
|
|
* Copyright 2011 - 2018, Lightbend, Inc.
|
|
* Copyright 2008 - 2010, Mark Harrah
|
|
* Licensed under Apache License 2.0 (see LICENSE)
|
|
*/
|
|
|
|
import sbt._
|
|
|
|
import org.scalacheck._
|
|
import Prop._
|
|
import TaskGen._
|
|
import math.abs
|
|
|
|
object TaskRunnerForkTest extends Properties("TaskRunner Fork") {
|
|
property("fork m tasks and wait for all to complete") = forAll(MaxTasksGen, MaxWorkersGen) {
|
|
(m: Int, workers: Int) =>
|
|
val values = (0 until m).toList
|
|
checkResult(tryRun(values.fork(f => ()).join.map(_.toList), false, workers), values)
|
|
true
|
|
}
|
|
property("Fork and reduce 2") = forAll(MaxTasksGen, MaxWorkersGen) { (m: Int, workers: Int) =>
|
|
(m > 1) ==> {
|
|
val task = (0 to m) fork { _ * 10 } reduced { _ + _ }
|
|
checkResult(tryRun(task, false, workers), 5 * (m + 1) * m)
|
|
}
|
|
}
|
|
property("Double join") = forAll(MaxJoinGen, MaxJoinGen, MaxWorkersGen) {
|
|
(a: Int, b: Int, workers: Int) =>
|
|
runDoubleJoin(abs(a), abs(b), workers)
|
|
true
|
|
}
|
|
def runDoubleJoin(a: Int, b: Int, workers: Int): Unit = {
|
|
def inner = List.range(0, b).map(j => task(j).named(j.toString)).join
|
|
tryRun(List.range(0, a).map(_ => inner).join, false, workers)
|
|
()
|
|
}
|
|
property("fork and reduce") = forAll(TaskListGen, MaxWorkersGen) { (m: List[Int], workers: Int) =>
|
|
m.nonEmpty ==> {
|
|
val expected = m.sum
|
|
checkResult(tryRun(m.tasks.reduced(_ + _), false, workers), expected)
|
|
}
|
|
}
|
|
}
|