Add hasSourceLocation assertion for FindingAssert

detekt · Jun 5, 2022 · da57cef · da57cef
1 parent 9146e4d
commit da57cef
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/detekt-test/api/detekt-test.api b/detekt-test/api/detekt-test.api
@@ -2,6 +2,7 @@ public final class io/gitlab/arturbosch/detekt/test/FindingAssert : org/assertj/
 	public fun <init> (Lio/gitlab/arturbosch/detekt/api/Finding;)V
 	public final fun getActual ()Lio/gitlab/arturbosch/detekt/api/Finding;
 	public final fun hasMessage (Ljava/lang/String;)Lio/gitlab/arturbosch/detekt/test/FindingAssert;
+	public final fun hasSourceLocation (II)Lio/gitlab/arturbosch/detekt/test/FindingAssert;
 }
 
 public final class io/gitlab/arturbosch/detekt/test/FindingsAssert : org/assertj/core/api/AbstractListAssert {

diff --git a/detekt-test/src/main/kotlin/io/gitlab/arturbosch/detekt/test/FindingsAssertions.kt b/detekt-test/src/main/kotlin/io/gitlab/arturbosch/detekt/test/FindingsAssertions.kt
@@ -92,6 +92,15 @@ class FindingsAssert(actual: List<Finding>) :
 }
 
 class FindingAssert(val actual: Finding?) : AbstractAssert<FindingAssert, Finding>(actual, FindingAssert::class.java) {
+
+    fun hasSourceLocation(line: Int, column: Int) = apply {
+        val expectedLocation = SourceLocation(line, column)
+        val actualLocation = actual.location.source
+        if (actualLocation != expectedLocation) {
+            failWithMessage("Expected source location to be $expectedLocation but was $actualLocation")
+        }
+    }
+
     fun hasMessage(expectedMessage: String) = apply {
         if (expectedMessage.isNotBlank() && actual.message.isBlank()) {
             failWithMessage("Expected message <$expectedMessage> but finding has no message")