Moving flag from constructor to member

eed3si9n · Sep 26, 2020 · 73f89d0 · 73f89d0
1 parent 435635a
commit 73f89d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/jvm/src/test/scala/org/expecty/ExpectyLocationSpec.scala b/jvm/src/test/scala/org/expecty/ExpectyLocationSpec.scala
@@ -19,7 +19,7 @@ import junit.framework.ComparisonFailure
 import com.eed3si9n.expecty.Expecty
 
 class ExpectyLocationSpec {
-  val assert = new Expecty(true) // (displayLoc = true)
+  val assert = new Expecty { override val showLocation = true }
 
   @Test
   def position(): Unit = {

diff --git a/shared/src/main/scala/com/eed3si9n/expecty/Expecty.scala b/shared/src/main/scala/com/eed3si9n/expecty/Expecty.scala
@@ -14,9 +14,10 @@
 
 package com.eed3si9n.expecty
 
-class Expecty(displayLoc: Boolean = false) extends Recorder[Boolean, Unit] {
+class Expecty extends Recorder[Boolean, Unit] {
   val failEarly: Boolean = true
   val showTypes: Boolean = false
+  val showLocation: Boolean = false
   // val printAsts: Boolean = false
   // val printExprs: Boolean = false
 
@@ -28,7 +29,7 @@ class Expecty(displayLoc: Boolean = false) extends Recorder[Boolean, Unit] {
       // if (printExprs) println(rendering)
       if (!recordedExpr.value && failEarly) {
         val loc = recordedExpr.location
-        val locStr = if(displayLoc) " (" + loc.relativePath + ":" + loc.line + ")" else ""
+        val locStr = if(showLocation) " (" + loc.relativePath + ":" + loc.line + ")" else ""
         val msg = recordedMessage()
         val header =
           "assertion failed" + locStr +