Updated unit tests

PureSwift · Aug 21, 2023 · fbbce91 · fbbce91
1 parent e46f139
commit fbbce91
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Tests/CoreModelMongoDBTests/MongoDBModelTests.swift b/Tests/CoreModelMongoDBTests/MongoDBModelTests.swift
@@ -82,5 +82,11 @@ final class MongoDBModelTests: XCTestCase {
         XCTAssertEqual(campgroundData.relationships[PropertyKey(Campground.CodingKeys.units)], .toMany([ObjectID(rentalUnit.id)]))
         let fetchedRentalUnit = try await store.fetch(Campground.RentalUnit.self, for: rentalUnit.id)
         XCTAssertEqual(fetchedRentalUnit, rentalUnit)
+        let rentalUnitFetchRequest = FetchRequest(
+            entity: Campground.RentalUnit.entityName,
+            predicate: Campground.RentalUnit.CodingKeys.campground.stringValue.compare(.equalTo, .relationship(.toOne(ObjectID(campground.id))))
+        )
+        let rentalUnitIDs = try await store.fetchID(rentalUnitFetchRequest)
+        XCTAssertEqual(rentalUnitIDs, campground.units.map { ObjectID($0) })
     }
 }