Fix broken unit test

Author: westey-m

dotnet/src/Connectors/Connectors.CosmosNoSql.UnitTests/CosmosNoSqlCollectionTests.cs

@@ -137,6 +137,28 @@ public async Task EnsureCollectionExistsUsesValidContainerPropertiesAsync(Indexi
         // Arrange
         const string CollectionName = "collection";
 
+        var mockFeedResponse = new Mock<FeedResponse<string>>();
+        mockFeedResponse
+            .Setup(l => l.Resource)
+            .Returns([]);
+
+        var mockFeedIterator = new Mock<FeedIterator<string>>();
+        mockFeedIterator
+            .SetupSequence(l => l.HasMoreResults)
+            .Returns(true)
+            .Returns(false);
+
+        mockFeedIterator
+            .Setup(l => l.ReadNextAsync(It.IsAny<CancellationToken>()))
+            .ReturnsAsync(mockFeedResponse.Object);
+
+        this._mockDatabase
+            .Setup(l => l.GetContainerQueryIterator<string>(
+                It.IsAny<QueryDefinition>(),
+                It.IsAny<string>(),
+                It.IsAny<QueryRequestOptions>()))
+            .Returns(mockFeedIterator.Object);
+
         var sut = new CosmosNoSqlCollection<string, TestIndexingModel>(
             this._mockDatabase.Object,
             CollectionName,

dotnet/src/Connectors/Connectors.Memory.CosmosNoSql/CosmosNoSqlCollection.cs

@@ -151,9 +151,9 @@ public override async Task<bool> CollectionExistsAsync(CancellationToken cancell
 
         using var errorHandlingFeedIterator = new ErrorHandlingFeedIterator<string>(feedIterator, this._collectionMetadata, OperationName);
 
-        while (feedIterator.HasMoreResults)
+        while (errorHandlingFeedIterator.HasMoreResults)
         {
-            var next = await feedIterator.ReadNextAsync(cancellationToken).ConfigureAwait(false);
+            var next = await errorHandlingFeedIterator.ReadNextAsync(cancellationToken).ConfigureAwait(false);
 
             foreach (var containerName in next.Resource)
             {