fix compilation of NativeProtocol seekInputStream func

Author: albertaleksieiev

Foundation/URLSession/NativeProtocol.swift

@@ -262,27 +262,38 @@ internal class _NativeProtocol: URLProtocol, _EasyHandleDelegate {
     func seekInputStream(to position: UInt64) throws {
         // We will reset the body source and seek forward.
         guard let session = task?.session as? URLSession else { fatalError() }
+        
+        var currentInputStream: InputStream?
+        
         if let delegate = session.delegate as? URLSessionTaskDelegate {
-            delegate.urlSession(session, task: task!, needNewBodyStream: { [weak self] inputStream in
-                if let strongSelf = self, let url = strongSelf.request.url, let inputStream = inputStream {
-                    switch strongSelf.internalState {
-                    case .transferInProgress(let currentTransferState):
-                        switch currentTransferState.requestBodySource {
-                        case is _BodyStreamSource:
-                            try inputStream.seek(to: position)
-                            let drain = strongSelf.createTransferBodyDataDrain()
-                            let source = _BodyStreamSource(inputStream: inputStream)
-                            let transferState = _TransferState(url: url, bodyDataDrain: drain, bodySource: source)
-                            strongSelf.internalState = .transferInProgress(transferState)
-                        default:
-                            NSUnimplemented()
-                        }
-                    default:
-                        //TODO: it's possible?
-                        break
-                    }
-                }
+            let dispatchGroup = DispatchGroup()
+            dispatchGroup.enter()
+            
+            delegate.urlSession(session, task: task!, needNewBodyStream: { inputStream in
+                currentInputStream = inputStream
+                dispatchGroup.leave()
             })
+            
+            _ = dispatchGroup.wait(timeout: .now() + 7)
+        }
+      
+        if let url = self.request.url, let inputStream = currentInputStream {
+            switch self.internalState {
+            case .transferInProgress(let currentTransferState):
+                switch currentTransferState.requestBodySource {
+                case is _BodyStreamSource:
+                    try inputStream.seek(to: position)
+                    let drain = self.createTransferBodyDataDrain()
+                    let source = _BodyStreamSource(inputStream: inputStream)
+                    let transferState = _TransferState(url: url, bodyDataDrain: drain, bodySource: source)
+                    self.internalState = .transferInProgress(transferState)
+                default:
+                    NSUnimplemented()
+                }
+            default:
+                //TODO: it's possible?
+                break
+            }
         }
     }
 

TestFoundation/TestStream.swift

@@ -6,6 +6,13 @@
 // See http://swift.org/LICENSE.txt for license information
 // See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
 //
+#if NS_FOUNDATION_ALLOWS_TESTABLE_IMPORT
+    #if (os(Linux) || os(Android))
+        @testable import Foundation
+    #else
+        @testable import SwiftFoundation
+    #endif
+#endif
 
 private extension Data {
     init(reading input: InputStream) {
@@ -18,7 +25,7 @@ private extension Data {
             let read = input.read(buffer, maxLength: bufferSize)
             self.append(buffer, count: read)
         }
-        buffer.deallocate(capacity: bufferSize)
+        buffer.deallocate()
 
         input.close()
     }
@@ -150,7 +157,7 @@ class TestStream : XCTestCase {
             try stream.seek(to: pos)
             let streamData = Data(reading: stream)
 
-            let subdata = data.subdata(in: Range(Int(pos)..<data.count))
+            let subdata = data[Int(pos)..<data.count]
             XCTAssertEqual(streamData, subdata)
 
             return subdata