DataStoreTests.swift

import Foundation
import Testing

@testable import WordPress

@Suite(.timeLimit(.minutes(1)))
struct InMemoryDataStoreTests {

    @Test
    func testUpdatesAfterCreation() async {
        let store = InMemoryDataStore<DisplayUser>()
        let stream = await store.listStream(query: .all)

        await confirmation("The stream produces an update") { confirmation in
            for await _ in stream.prefix(1) {
                confirmation()
            }
        }
    }

    @Test
    func testUpdatesAfterStore() async {
        let store = InMemoryDataStore<DisplayUser>()
        let stream = await store.listStream(query: .all)

        Task.detached {
            try await Task.sleep(for: .milliseconds(50))
            try await store.store([.MockUser])
        }

        await confirmation("The stream produces an update", expectedCount: 2) { confirmation in
            for await _ in stream.prefix(2) {
                confirmation()
            }
        }
    }

    @Test
    func testUpdatesAfterDelete() async throws {
        let store = InMemoryDataStore<DisplayUser>()
        try await store.store([.MockUser])

        let stream = await store.listStream(query: .all)

        Task.detached {
            try await Task.sleep(for: .milliseconds(50))
            try await store.delete(query: .all)
        }

        await confirmation("The stream produces an update", expectedCount: 2) { confirmation in
            for await _ in stream.prefix(2) {
                confirmation()
            }
        }
    }

    @Test
    func testStreamTerminates() async {
        var store: InMemoryDataStore<DisplayUser>? = .init()
        let stream = await store!.listStream(query: .all)

        Task.detached {
            try await Task.sleep(for: .milliseconds(50))
            store = nil
        }

        await confirmation("The stream produces one update and then terminates", expectedCount: 1) { confirmation in
            for await _ in stream {
                // Do nothing
            }
            confirmation()
        }
    }

}