more tests

Author: tomerd

Tests/PackageCollectionsTests/PackageCollectionsTests.swift

@@ -94,6 +94,59 @@ final class PackageCollectionsTests: XCTestCase {
         }
     }
 
+    func testDeleteFromStorageWhenLast() throws {
+        let configuration = PackageCollections.Configuration()
+        let storage = makeMockStorage()
+        defer { XCTAssertNoThrow(try storage.close()) }
+
+        let mockCollection = makeMockCollections(count: 1).first!
+        let mockProfile1 = PackageCollectionsModel.Profile(name: "profile-\(UUID().uuidString)")
+        let mockProfile2 = PackageCollectionsModel.Profile(name: "profile-\(UUID().uuidString)")
+
+        let providers = [PackageCollectionsModel.CollectionSourceType.feed: MockProvider([mockCollection])]
+        let packageCollections = PackageCollections(configuration: configuration, storage: storage, providers: providers)
+
+        do {
+            let list = try await { callback in packageCollections.listCollections(callback: callback) }
+            XCTAssertEqual(list.count, 0, "list should be empty")
+        }
+
+        _ = try await { callback in packageCollections.addCollection(mockCollection.source, order: nil, to: mockProfile1, callback: callback) }
+        _ = try await { callback in packageCollections.addCollection(mockCollection.source, order: nil, to: mockProfile2, callback: callback) }
+
+        do {
+            let list1 = try await { callback in packageCollections.listCollections(in: mockProfile1, callback: callback) }
+            XCTAssertEqual(list1.count, 1, "list count should match")
+
+            let list2 = try await { callback in packageCollections.listCollections(in: mockProfile2, callback: callback) }
+            XCTAssertEqual(list2.count, 1, "list count should match")
+        }
+
+        do {
+            _ = try await { callback in packageCollections.removeCollection(mockCollection.source, from: mockProfile1, callback: callback) }
+            let list1 = try await { callback in packageCollections.listCollections(in: mockProfile1, callback: callback) }
+            XCTAssertEqual(list1.count, 0, "list count should match")
+
+            let list2 = try await { callback in packageCollections.listCollections(in: mockProfile2, callback: callback) }
+            XCTAssertEqual(list2.count, 1, "list count should match")
+
+            // check if exists in storage
+            XCTAssertNoThrow(try await { callback in storage.collections.get(identifier: mockCollection.identifier, callback: callback) })
+        }
+
+        do {
+            _ = try await { callback in packageCollections.removeCollection(mockCollection.source, from: mockProfile2, callback: callback) }
+            let list1 = try await { callback in packageCollections.listCollections(in: mockProfile1, callback: callback) }
+            XCTAssertEqual(list1.count, 0, "list count should match")
+
+            let list2 = try await { callback in packageCollections.listCollections(in: mockProfile2, callback: callback) }
+            XCTAssertEqual(list2.count, 0, "list count should match")
+
+            // check if exists in storage
+            XCTAssertThrowsError(try await { callback in storage.collections.get(identifier: mockCollection.identifier, callback: callback) }, "expected error")
+        }
+    }
+
     func testOrdering() throws {
         let configuration = PackageCollections.Configuration()
         let storage = makeMockStorage()