Port #2463 to v2

xunit · Jan 18, 2022 · 19618ef · 19618ef
1 parent b88fee5
commit 19618ef
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/src/xunit.assert/Asserts b/src/xunit.assert/Asserts
diff --git a/test/test.xunit.assert/Asserts/CollectionAssertsTests.cs b/test/test.xunit.assert/Asserts/CollectionAssertsTests.cs
@@ -925,6 +925,14 @@ public static void Equivalence()
 
 			Assert.Equal(expected, actual);
 		}
+
+		[Fact]
+		public static void EnumeratesOnlyOnce()
+		{
+			var expected = new[] { 1, 2, 3, 4, 5 };
+			var actual = new RunOnceEnumerable<int>(expected);
+			Assert.Equal(expected, actual);
+		}
 	}
 
 	public class EqualDictionary
@@ -1350,6 +1358,26 @@ public static void PredicateTooManyMatches()
 		}
 	}
 
+	sealed class RunOnceEnumerable<T> : IEnumerable<T>
+	{
+		private readonly IEnumerable<T> _source;
+		private bool _called;
+
+		public RunOnceEnumerable(IEnumerable<T> source)
+		{
+			_source = source;
+		}
+
+		public IEnumerator<T> GetEnumerator()
+		{
+			Assert.False(_called, "GetEnumerator() was called more than once");
+			_called = true;
+			return _source.GetEnumerator();
+		}
+
+		IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
+	}
+
 	sealed class SpyEnumerator<T> : IEnumerable<T>, IEnumerator<T>
 	{
 		IEnumerator<T>? innerEnumerator;