From 955687efe3368daba85261c796649db1ea110607 Mon Sep 17 00:00:00 2001
From: George Lesica <george@lesica.com>
Date: Sat, 15 Feb 2025 12:32:00 -0700
Subject: [PATCH] Add sketch of a SeqContains function

---
 assert.go | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/assert.go b/assert.go
index 261beaf..8bd5db8 100644
--- a/assert.go
+++ b/assert.go
@@ -5,6 +5,7 @@ import (
 	"bytes"
 	"errors"
 	"fmt"
+	"iter"
 	"reflect"
 	"strconv"
 	"strings"
@@ -152,6 +153,39 @@ func NotSliceContains[T any](t testing.TB, haystack []T, needle T, msgAndArgs ..
 	}
 }
 
+// SeqContains asserts that "haystack" contains "needle", but only checks, at most,
+// maxChecked items (since a Seq can be infinite). If this is <= 0 there is no limit.
+func SeqContains[T any](t testing.TB, haystack iter.Seq[T], needle T, maxChecked int, msgAndArgs ...interface{}) {
+	t.Helper()
+
+	found := false
+	count := 0
+	haystack(func(item T) bool {
+		if objectsAreEqual(item, needle) {
+			found = true
+			return false
+		}
+
+		if maxChecked > 0 {
+			count++
+			if count >= maxChecked {
+				return false
+			}
+		}
+
+		return true
+	})
+
+	if found {
+		return
+	}
+
+	msg := formatMsgAndArgs("Haystack does not contain needle.", msgAndArgs...)
+	needleRepr := repr.String(needle, repr.Indent("  "))
+	haystackRepr := repr.String(haystack, repr.Indent("  "))
+	t.Fatalf("%s\nNeedle: %s\nHaystack: %s\n", msg, needleRepr, haystackRepr)
+}
+
 // Zero asserts that a value is its zero value.
 func Zero[T any](t testing.TB, value T, msgAndArgs ...any) {
 	var zero T