test: fix record iteration tests

gnolang · Feb 22, 2025 · b3909fe · b3909fe
1 parent 0eab92a
commit b3909fe
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/examples/gno.land/p/jeronimoalbi/datastore/record.gno b/examples/gno.land/p/jeronimoalbi/datastore/record.gno
@@ -182,6 +182,10 @@ func (rs recordset) Iterate(fn RecordIterFn) (stopped bool) {
 
 	offset := rs.query.Offset()
 	count := rs.query.Size()
+	if count == 0 {
+		count = rs.records.Size()
+	}
+
 	return rs.records.IterateByOffset(offset, count, func(_ string, v interface{}) bool {
 		return fn(v.(Record))
 	})
@@ -205,8 +209,12 @@ func (rs recordset) ReverseIterate(fn RecordIterFn) (stopped bool) {
 	}
 
 	offset := rs.query.Offset()
-	size := rs.query.Size()
-	return rs.records.ReverseIterateByOffset(offset, size, func(_ string, v interface{}) bool {
+	count := rs.query.Size()
+	if count == 0 {
+		count = rs.records.Size()
+	}
+
+	return rs.records.ReverseIterateByOffset(offset, count, func(_ string, v interface{}) bool {
 		return fn(v.(Record))
 	})
 }