Use section end logic when checking assertNoFurtherElements

robfletcher · Aug 5, 2024 · 9d78529 · 9d78529
1 parent c6c4fe2
commit 9d78529
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/strikt-core/src/main/kotlin/strikt/assertions/Iterable.kt b/strikt-core/src/main/kotlin/strikt/assertions/Iterable.kt
@@ -692,8 +692,8 @@ fun <T: Iterable<E>, E> Builder<T>.containsWithOrderingConstraints(
     }
 
     val assertNoFurtherElements = builder.expectsNoFurtherElements ||
-      // Check if the last section has a `last` constraint
-      allSections.last().elementsWithConstraints.any { it.constraints.any { it == OrderingConstraint.Last } }
+      // Check if the last section explicitly defines the end element
+      allSections.last().endDefinedBy is SectionAssertionSpec.EndDefinition.DeclaredElement<*>
     if (assertNoFurtherElements) {
       assert("contains no further elements", expected = emptyList<E>()) {
         if (elementsConsumed == it.count()) {