Skip to content

Commit 6f7c9a5

Browse files
authored
fix: Fix subcommands options in fuzz-testing (#2684)
1 parent 0ba787f commit 6f7c9a5

File tree

1 file changed

+9
-3
lines changed
  • fuzz-testing/src/main/scala/org/apache/comet/fuzz

1 file changed

+9
-3
lines changed

fuzz-testing/src/main/scala/org/apache/comet/fuzz/Main.scala

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ class Conf(arguments: Seq[String]) extends ScallopConf(arguments) {
4747
addSubcommand(generateData)
4848
object generateQueries extends Subcommand("queries") {
4949
val numFiles: ScallopOption[Int] =
50-
opt[Int](required = false, descr = "Number of input files to use")
50+
opt[Int](required = true, descr = "Number of input files to use")
5151
val numQueries: ScallopOption[Int] =
5252
opt[Int](required = true, descr = "Number of queries to generate")
5353
val randomSeed: ScallopOption[Long] =
@@ -58,7 +58,9 @@ class Conf(arguments: Seq[String]) extends ScallopConf(arguments) {
5858
val filename: ScallopOption[String] =
5959
opt[String](required = true, descr = "File to write queries to")
6060
val numFiles: ScallopOption[Int] =
61-
opt[Int](required = false, descr = "Number of input files to use")
61+
opt[Int](required = true, descr = "Number of input files to use")
62+
val showFailedSparkQueries: ScallopOption[Boolean] =
63+
opt[Boolean](required = false, descr = "Whether to show failed Spark queries")
6264
}
6365
addSubcommand(runQueries)
6466
verify()
@@ -107,7 +109,11 @@ object Main {
107109
numFiles = conf.generateQueries.numFiles(),
108110
conf.generateQueries.numQueries())
109111
case Some(conf.runQueries) =>
110-
QueryRunner.runQueries(spark, conf.runQueries.numFiles(), conf.runQueries.filename())
112+
QueryRunner.runQueries(
113+
spark,
114+
conf.runQueries.numFiles(),
115+
conf.runQueries.filename(),
116+
conf.runQueries.showFailedSparkQueries())
111117
case _ =>
112118
// scalastyle:off println
113119
println("Invalid subcommand")

0 commit comments

Comments
 (0)