Fix bufferingPolicy not passed in.

liuliu · Jun 10, 2022 · 4e4c747 · 4e4c747
1 parent f79fc09
commit 4e4c747
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/sqlite/SQLiteWorkspace.swift b/src/sqlite/SQLiteWorkspace.swift
@@ -526,7 +526,7 @@ public final class SQLiteWorkspace: Workspace {
     public func subscribe<Element: Atom & Equatable>(
       object: Element, bufferingPolicy: AsyncStream<Element>.Continuation.BufferingPolicy
     ) -> AsyncStream<Element> {
-      AsyncStream { continuation in
+      AsyncStream(bufferingPolicy: bufferingPolicy) { continuation in
         let cancellable = self.subscribe(object: object) { object in
           switch object {
           case .deleted:
@@ -546,7 +546,7 @@ public final class SQLiteWorkspace: Workspace {
       fetchedResult: FetchedResult<Element>,
       bufferingPolicy: AsyncStream<FetchedResult<Element>>.Continuation.BufferingPolicy
     ) -> AsyncStream<FetchedResult<Element>> {
-      AsyncStream { continuation in
+      AsyncStream(bufferingPolicy: bufferingPolicy) { continuation in
         let cancellable = self.subscribe(fetchedResult: fetchedResult) { fetchedResult in
           continuation.yield(fetchedResult)
         }

diff --git a/src/tests/AsyncTests.swift b/src/tests/AsyncTests.swift
@@ -144,14 +144,13 @@ class AsyncTests: XCTestCase {
         MyGame.Sample.Monster.mana <= 50, orderBy: [MyGame.Sample.Monster.mana.ascending])
       XCTAssertEqual(fetchedResult.count, 6)
       let subscribeTask = Task { () -> FetchedResult<MyGame.Sample.Monster> in
-        var updateCount = 0
         var updatedFetchedResult = fetchedResult
         for await newFetchedResult in dflat.subscribe(
           fetchedResult: fetchedResult, bufferingPolicy: .unbounded)
         {
           updatedFetchedResult = newFetchedResult
-          updateCount += 1
-          if updateCount == 4 {
+          if newFetchedResult.first(where: { $0.name == "name 3" }) == nil {
+            // We've deleted this one, that's the last mutation, exit.
             break
           }
         }