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
|
// SPDX-License-Identifier: Apache-2.0
package firrtlTests.options
import firrtl.annotations.NoTargetAnnotation
import firrtl.options.Shell
import org.scalatest.flatspec.AnyFlatSpec
import org.scalatest.matchers.should.Matchers
class ShellSpec extends AnyFlatSpec with Matchers {
case object A extends NoTargetAnnotation
case object B extends NoTargetAnnotation
case object C extends NoTargetAnnotation
case object D extends NoTargetAnnotation
case object E extends NoTargetAnnotation
trait AlphabeticalCli { this: Shell =>
parser.opt[Unit]('c', "c-option").unbounded().action((x, c) => C +: c)
parser.opt[Unit]('d', "d-option").unbounded().action((x, c) => D +: c)
parser.opt[Unit]('e', "e-option").unbounded().action((x, c) => E +: c)
}
behavior.of("Shell")
it should "detect all registered libraries and transforms" in {
val shell = new Shell("foo")
info("Found FooTransform")
shell.registeredTransforms.map(_.getClass.getName) should contain("firrtlTests.options.FooTransform")
info("Found BarLibrary")
shell.registeredLibraries.map(_.getClass.getName) should contain("firrtlTests.options.BarLibrary")
}
it should "correctly order annotations and options" in {
val shell = new Shell("foo") with AlphabeticalCli
shell.parse(Array("-c", "-d", "-e"), Seq(A, B)).toSeq should be(Seq(A, B, C, D, E))
}
}
|
