blob: 9a5baf1155d4c795b4b6a892880537bd153f069b (
plain) (
blame)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
package mill.util
import ammonite.main.Router.Overrides
import ammonite.ops.pwd
import mill.define._
import mill.eval.Result
import mill.eval.Result.OuterStack
import utest.assert
import mill.util.Strict.Agg
import utest.framework.TestPath
import scala.collection.mutable
object TestUtil {
def getOutPath()(implicit fullName: sourcecode.FullName,
tp: TestPath) = {
pwd / 'target / 'workspace / (fullName.value.split('.') ++ tp.value)
}
def getOutPathStatic()(implicit fullName: sourcecode.FullName) = {
pwd / 'target / 'workspace / fullName.value.split('.')
}
def getSrcPathStatic()(implicit fullName: sourcecode.FullName) = {
pwd / 'target / 'worksources / fullName.value.split('.')
}
def getSrcPathBase() = {
pwd / 'target / 'worksources
}
class BaseModule(implicit millModuleEnclosing0: sourcecode.Enclosing,
millModuleLine0: sourcecode.Line,
millName0: sourcecode.Name,
overrides: Overrides)
extends mill.define.BaseModule(getSrcPathBase() / millModuleEnclosing0.value.split("\\.| |#")){
def millDiscover: Discover[this.type] = Discover[this.type]
}
object test{
def anon(inputs: Task[Int]*) = new Test(inputs)
def apply(inputs: Task[Int]*)
(implicit ctx: mill.define.Ctx)= {
new TestTarget(inputs, pure = inputs.nonEmpty)
}
}
class Test(val inputs: Seq[Task[Int]]) extends Task[Int]{
var counter = 0
var failure = Option.empty[String]
var exception = Option.empty[Throwable]
override def evaluate(args: Ctx) = {
failure.map(Result.Failure(_)) orElse
exception.map(Result.Exception(_, new OuterStack(Nil))) getOrElse
Result.Success(counter + args.args.map(_.asInstanceOf[Int]).sum)
}
override def sideHash = counter + failure.hashCode() + exception.hashCode()
}
/**
* A dummy target that takes any number of inputs, and whose output can be
* controlled externally, so you can construct arbitrary dataflow graphs and
* test how changes propagate.
*/
class TestTarget(inputs: Seq[Task[Int]],
val pure: Boolean)
(implicit ctx0: mill.define.Ctx)
extends Test(inputs) with Target[Int]{
val ctx = ctx0.copy(segments = ctx0.segments ++ Seq(ctx0.segment))
val readWrite = upickle.default.IntRW
}
def checkTopological(targets: Agg[Task[_]]) = {
val seen = mutable.Set.empty[Task[_]]
for(t <- targets.indexed.reverseIterator){
seen.add(t)
for(upstream <- t.inputs){
assert(!seen(upstream))
}
}
}
}
|