LeStorage/LeStorageTests/StoredCollectionTests.swift

//
//  StoredCollectionTests.swift
//  LeStorageTests
//
//  Created by Laurent Morvillier on 16/10/2024.
//

import Testing
@testable import LeStorage

struct Error: Swift.Error, CustomStringConvertible {
    let description: String
    
    init(_ description: String) {
        self.description = description
    }
}

struct StoredCollectionTests {

    var collection: StoredCollection<MockStorable>
    
    init() {
        collection = StoreCenter.main.mainStore.registerCollection()
    }
    
    func ensureCollectionLoaded() async throws {
        // Wait for the collection to finish loading
        // Adjust the timeout as needed
        let timeout = 5.0 // seconds
        let startTime = Date()
        
        while !collection.hasLoaded {
            // Check for timeout
            if Date().timeIntervalSince(startTime) > timeout {
                throw Error("Collection loading timed out")
            }
            // Wait a bit before checking again
            try await Task.sleep(for: .milliseconds(100))
        }
        collection.reset()
    }
    
    @Test func testInitialization() async throws {
        try await ensureCollectionLoaded()
        #expect(collection.items.count == 0)
    }

    @Test func testAddOrUpdate() async throws {
        try await ensureCollectionLoaded()
        let item = MockStorable(id: "1", name: "Test")
        collection.addOrUpdate(instance: item)

        #expect(collection.items.count == 1)
        if let first = collection.items.first {
            #expect(first.id == "1")
        } else {
            Issue.record("missing record")
        }
        
    }

    @Test func testDelete() async throws {
        try await ensureCollectionLoaded()
        let item = MockStorable(id: "1", name: "Test")
        collection.addOrUpdate(instance: item)
        #expect(collection.items.count == 1)

        collection.delete(instance: item)
        #expect(collection.items.isEmpty)
    }

    @Test func testFindById() async throws {
        try await ensureCollectionLoaded()
        let item = MockStorable(id: "1", name: "Test")
        collection.addOrUpdate(instance: item)

        if let foundItem = collection.findById("1") {
            #expect(foundItem.id == "1")
        } else {
            Issue.record("missing item")
        }
    }

    @Test func testDeleteById() async throws {
        try await ensureCollectionLoaded()
        let item = MockStorable(id: "1", name: "Test")
        collection.addOrUpdate(instance: item)

        collection.deleteById("1")
        let search = collection.findById("1")
        #expect(search == nil)
    }

    @Test func testAddOrUpdateMultiple() async throws {
        try await ensureCollectionLoaded()
        let items = [
            MockStorable(id: "1", name: "Test1"),
            MockStorable(id: "2", name: "Test2"),
        ]

        collection.addOrUpdate(contentOfs: items)
        #expect(collection.items.count == 2)
    }

    @Test func testDeleteAll() async throws {
        try await ensureCollectionLoaded()
        let items = [
            MockStorable(id: "1", name: "Test1"),
            MockStorable(id: "2", name: "Test2"),
        ]

        collection.addOrUpdate(contentOfs: items)
        #expect(collection.items.count == 2)

        collection.clear()
        #expect(collection.items.isEmpty)
    }

    @Test func testRandomAccessCollection() async throws {
        try await ensureCollectionLoaded()
        let items = [
            MockStorable(id: "1", name: "Test1"),
            MockStorable(id: "2", name: "Test2"),
            MockStorable(id: "3", name: "Test3"),
        ]

        collection.addOrUpdate(contentOfs: items)

        #expect(collection.startIndex == 0)
        #expect(collection.endIndex == 3)
        
        if collection.count > 2 {
            #expect(collection[1].name == "Test2")
        } else {
            Issue.record("count not good")
        }

    }
}

// Mock Storable for testing purposes
class MockStorable: ModelObject, Storable {

    var id: String = Store.randomId()
    var name: String

    init(id: String, name: String) {
        self.id = id
        self.name = name
    }

    static func resourceName() -> String {
        return "mocks"
    }
    func copy(from other: any LeStorage.Storable) {
    }
    
    static func relationships() -> [LeStorage.Relationship] {
        return []
    }
    
}