@@ -17,14 +17,14 @@ import NIOEmbedded
17
17
import NIOHTTP1
18
18
import Testing
19
19
20
+ @testable import AWSLambdaRuntime
21
+
20
22
#if canImport(FoundationEssentials)
21
23
import FoundationEssentials
22
24
#else
23
25
import Foundation
24
26
#endif
25
27
26
- @testable import AWSLambdaRuntime
27
-
28
28
struct ControlPlaneRequestEncoderTests {
29
29
let host = " 192.168.0.1 "
30
30
@@ -44,7 +44,7 @@ struct ControlPlaneRequestEncoderTests {
44
44
_ = try ? client. finish ( acceptAlreadyClosed: false )
45
45
_ = try ? server. finish ( acceptAlreadyClosed: false )
46
46
}
47
-
47
+
48
48
let request = try sendRequest ( . next, client: client, server: server)
49
49
50
50
#expect( request? . head. isKeepAlive == true )
@@ -64,7 +64,7 @@ struct ControlPlaneRequestEncoderTests {
64
64
_ = try ? client. finish ( acceptAlreadyClosed: false )
65
65
_ = try ? server. finish ( acceptAlreadyClosed: false )
66
66
}
67
-
67
+
68
68
let requestID = UUID ( ) . uuidString
69
69
let request = try sendRequest ( . invocationResponse( requestID, nil ) , client: client, server: server)
70
70
@@ -86,7 +86,7 @@ struct ControlPlaneRequestEncoderTests {
86
86
_ = try ? client. finish ( acceptAlreadyClosed: false )
87
87
_ = try ? server. finish ( acceptAlreadyClosed: false )
88
88
}
89
-
89
+
90
90
let requestID = UUID ( ) . uuidString
91
91
let payload = ByteBuffer ( string: " hello swift lambda! " )
92
92
@@ -111,7 +111,7 @@ struct ControlPlaneRequestEncoderTests {
111
111
_ = try ? client. finish ( acceptAlreadyClosed: false )
112
112
_ = try ? server. finish ( acceptAlreadyClosed: false )
113
113
}
114
-
114
+
115
115
let requestID = UUID ( ) . uuidString
116
116
let error = ErrorResponse ( errorType: " SomeError " , errorMessage: " An error happened " )
117
117
let request = try sendRequest ( . invocationError( requestID, error) , client: client, server: server)
@@ -139,7 +139,7 @@ struct ControlPlaneRequestEncoderTests {
139
139
_ = try ? client. finish ( acceptAlreadyClosed: false )
140
140
_ = try ? server. finish ( acceptAlreadyClosed: false )
141
141
}
142
-
142
+
143
143
let error = ErrorResponse ( errorType: " StartupError " , errorMessage: " Urgh! Startup failed. 😨 " )
144
144
let request = try sendRequest ( . initializationError( error) , client: client, server: server)
145
145
@@ -165,21 +165,29 @@ struct ControlPlaneRequestEncoderTests {
165
165
_ = try ? client. finish ( acceptAlreadyClosed: false )
166
166
_ = try ? server. finish ( acceptAlreadyClosed: false )
167
167
}
168
-
168
+
169
169
for _ in 0 ..< 1000 {
170
170
let nextRequest = try sendRequest ( . next, client: client, server: server)
171
171
#expect( nextRequest? . head. method == . GET)
172
172
#expect( nextRequest? . head. uri == " /2018-06-01/runtime/invocation/next " )
173
173
174
174
let requestID = UUID ( ) . uuidString
175
175
let payload = ByteBuffer ( string: " hello swift lambda! " )
176
- let successRequest = try sendRequest ( . invocationResponse( requestID, payload) , client: client, server: server)
176
+ let successRequest = try sendRequest (
177
+ . invocationResponse( requestID, payload) ,
178
+ client: client,
179
+ server: server
180
+ )
177
181
#expect( successRequest? . head. method == . POST)
178
182
#expect( successRequest? . head. uri == " /2018-06-01/runtime/invocation/ \( requestID) /response " )
179
183
}
180
184
}
181
185
182
- func sendRequest( _ request: ControlPlaneRequest , client: EmbeddedChannel , server: EmbeddedChannel ) throws -> NIOHTTPServerRequestFull ? {
186
+ func sendRequest(
187
+ _ request: ControlPlaneRequest ,
188
+ client: EmbeddedChannel ,
189
+ server: EmbeddedChannel
190
+ ) throws -> NIOHTTPServerRequestFull ? {
183
191
try client. writeOutbound ( request)
184
192
while let part = try client. readOutbound ( as: ByteBuffer . self) {
185
193
try server. writeInbound ( part)
0 commit comments