Merge pull request #281 from sbarow/fix_test_coverage

damian-kolakowski · web-flow · commit fdcafa2eb641 · 2018-05-12T09:14:30.000+03:00
Move while loop read to own function
diff --git a/Sources/WebSockets.swift b/Sources/WebSockets.swift
@@ -98,11 +98,15 @@ public func websocket(
                 }
             }
             
-            do {
+            func read() throws {
                 while true {
                     let frame = try session.readFrame()
                     try handleOperationCode(frame)
                 }
+            }
+            
+            do {
+                try read()
             } catch let error {
                 switch error {
                 case WebSocketSession.Control.close:

Original file line number	Diff line number	Diff line change
`@@ -98,11 +98,15 @@ public func websocket(`
`98`	`98`	`}`
`99`	`99`	`}`
`100`	`100`
`101`		`- do {`
	`101`	`+ func read() throws {`
`102`	`102`	`while true {`
`103`	`103`	`let frame = try session.readFrame()`
`104`	`104`	`try handleOperationCode(frame)`
`105`	`105`	`}`
	`106`	`+ }`
	`107`	`+`
	`108`	`+ do {`
	`109`	`+ try read()`
`106`	`110`	`} catch let error {`
`107`	`111`	`switch error {`
`108`	`112`	`case WebSocketSession.Control.close:`