Skip to content
Prev Previous commit
Next Next commit
Fix unit tests.
  • Loading branch information
vbabanin committed Aug 20, 2025
commit 370b8e64ebd94bbf3a9d72b1105579705b8f2d4e
Original file line number Diff line number Diff line change
Expand Up @@ -130,12 +130,13 @@ class GridFSFindIterableSpecification extends Specification {
, null),
]
def cursor = {
def batchToReturn = cannedResults.collect();
Stub(BatchCursor) {
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -160,8 +160,8 @@ class ChangeStreamIterableSpecification extends Specification {
def cannedResults = ['{_id: {_data: 1}}', '{_id: {_data: 2}}', '{_id: {_data: 3}}'].collect {
RawBsonDocument.parse(it)
}
def executor = new TestOperationExecutor([cursor(cannedResults), cursor(cannedResults), cursor(cannedResults),
cursor(cannedResults)])
def executor = new TestOperationExecutor([cursor(cannedResults.collect()), cursor(cannedResults.collect()), cursor(cannedResults.collect()),
cursor(cannedResults.collect())])
def mongoIterable = new ChangeStreamIterableImpl(null, namespace, codecRegistry, readPreference, readConcern, executor, [],
Document, ChangeStreamLevel.COLLECTION, true, TIMEOUT_SETTINGS)

Expand Down Expand Up @@ -208,8 +208,8 @@ class ChangeStreamIterableSpecification extends Specification {
given:
def count = 0
def cannedResults = ['{_id: { _data: 1}}', '{_id: {_data: 2}}', '{_id: {_data: 3}}'].collect { RawBsonDocument.parse(it) }
def executor = new TestOperationExecutor([cursor(cannedResults), cursor(cannedResults), cursor(cannedResults),
cursor(cannedResults)])
def executor = new TestOperationExecutor([cursor(cannedResults.collect()), cursor(cannedResults.collect()), cursor(cannedResults.collect()),
cursor(cannedResults.collect())])
def mongoIterable = new ChangeStreamIterableImpl(null, namespace, codecRegistry, readPreference, readConcern, executor, [],
Document, ChangeStreamLevel.COLLECTION, true, TIMEOUT_SETTINGS).withDocumentClass(RawBsonDocument)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -128,12 +128,13 @@ class DistinctIterableSpecification extends Specification {
given:
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
def batchToReturn = cannedResults.collect()
Stub(BatchCursor) {
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -221,11 +221,12 @@ class FindIterableSpecification extends Specification {
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
Stub(BatchCursor) {
def batchToReturn = cannedResults.collect()
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -127,12 +127,13 @@ class ListCollectionsIterableSpecification extends Specification {
given:
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
def batchToReturn = cannedResults.collect()
Stub(BatchCursor) {
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,12 +83,13 @@ class ListDatabasesIterableSpecification extends Specification {
given:
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
def batchToReturn = cannedResults.collect()
Stub(BatchCursor) {
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -105,11 +105,12 @@ class ListIndexesIterableSpecification extends Specification {
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
Stub(BatchCursor) {
def batchToReturn = cannedResults.collect()
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -255,12 +255,13 @@ class MapReduceIterableSpecification extends Specification {
given:
def cannedResults = [new Document('_id', 1), new Document('_id', 2), new Document('_id', 3)]
def cursor = {
def batchToReturn = cannedResults.collect()
Stub(BatchCursor) {
def count = 0
def results
def getResult = {
count++
results = count == 1 ? cannedResults : null
results = count == 1 ? batchToReturn : null
results
}
next() >> {
Expand Down