[Netlens]- fix observe calls use case

Author: sassiwalid

Tests/NetLensTests/Helpers/WaitForStream+Async.swift

@@ -33,4 +33,41 @@ func waitForStreamValue<T:Sendable> (from stream: AsyncStream<T>, timeout: TimeI
     }
 }
 
+func waitForStreamCondition<T: Sendable>(
+    from stream: AsyncStream<T>,
+    timeout: TimeInterval = 2.0,
+    condition: @Sendable @escaping (T) -> Bool
+) async throws -> T {
+    try await withThrowingTaskGroup(of: T?.self) { group in
+
+        group.addTask(operation: { @Sendable in
+                var iterator = stream.makeAsyncIterator()
+
+                while let value = await iterator.next() {
+                    if condition(value) {
+                        return value
+                    }
+                }
+                return nil
+            })
+
+        group.addTask {
+            try await Task.sleep(for: .seconds(timeout))
+            return nil
+        }
+
+        guard let result = try await group.next() else {
+            throw AsyncTestError.timedOut
+        }
+
+        group.cancelAll()
+
+        guard let value = result else {
+            throw AsyncTestError.timedOut
+        }
+
+        return value
+    }
+}
+
 

Tests/NetLensTests/Mocks/Repositories/MockNetworkCallRepository.swift

@@ -9,8 +9,7 @@ import Foundation
 
 @testable import NetLens
 
-@MainActor
-final class MockNetworkCallRepository: @preconcurrency NetworkCallRepository, @unchecked Sendable {
+final class MockNetworkCallRepository: NetworkCallRepository, @unchecked Sendable {
 
     private(set) var calls: [NetworkCall] = []
 
@@ -49,14 +48,16 @@ final class MockNetworkCallRepository: @preconcurrency NetworkCallRepository, @u
     }
 
     func reset() {
+        callsSubject.continuation.finish()
+
         calls.removeAll()
 
         addNetworkCallCallCount = 0
 
         getAllCallCallCount = 0
 
         clearAllCallsCount = 0
-        
+
         callsSubject = AsyncStream<[NetworkCall]>.makeStream()
     }
 

Tests/NetLensTests/UseCasesTests/ObserveNetworkCallsUseCaseSwiftTests.swift

@@ -24,30 +24,33 @@ struct ObserveNetworkCallsUseCaseSwiftTests {
     }
 
     @Test("Execute return stream from repository")
-    func execureReturnstream() async throws {
+    func executeReturnstream() async throws {
 
-        let call = NetworkCallFactory.makeSuccessCall()
+        let testCall = NetworkCallFactory.makeSuccessCall()
 
         let stream = observeNetworkCallsUseCase.execute()
 
         let streamTask = Task {
-            try await waitForStreamValue(
+            try await waitForStreamCondition(
                 from: stream,
                 timeout: 2.0
-            )
-            
+            ) { calls in
+                !calls.isEmpty
+            }
         }
 
-        await mockNetworkrepository.addNetworkCall(call)
+        try await Task.sleep(for: .milliseconds(50))
+
+        await mockNetworkrepository.addNetworkCall(testCall)
 
         let receivedCalls = try await streamTask.value
 
         #expect(receivedCalls.count == 1)
 
-        #expect(receivedCalls.first?.id == call.id)
-        
+        #expect(receivedCalls.first?.id == testCall.id)
+
     }
-    
-    
+
+
 }