@@ -24,7 +24,9 @@ class HTTPConnectionPool_WaiterTests: XCTestCase {
24
24
let theRightEL = eventLoopGroup. next ( )
25
25
let theFalseEL = eventLoopGroup. next ( )
26
26
27
- let waiter = HTTPConnectionPool . Waiter ( request: MockScheduledRequest ( ) , eventLoopRequirement: theRightEL)
27
+ let mockRequest = MockScheduledRequest ( eventLoopPreference: . init( . testOnly_exact( channelOn: theRightEL, delegateOn: theFalseEL) ) )
28
+
29
+ let waiter = HTTPConnectionPool . Waiter ( request: mockRequest)
28
30
29
31
XCTAssertTrue ( waiter. canBeRun ( on: theRightEL) )
30
32
XCTAssertFalse ( waiter. canBeRun ( on: theFalseEL) )
@@ -33,7 +35,8 @@ class HTTPConnectionPool_WaiterTests: XCTestCase {
33
35
func testCanBeRunIfNoEventLoopIsSpecified( ) {
34
36
let eventLoopGroup = MultiThreadedEventLoopGroup ( numberOfThreads: 2 )
35
37
36
- let waiter = HTTPConnectionPool . Waiter ( request: MockScheduledRequest ( ) , eventLoopRequirement: nil )
38
+ let mockRequest = MockScheduledRequest ( eventLoopPreference: . indifferent)
39
+ let waiter = HTTPConnectionPool . Waiter ( request: mockRequest)
37
40
38
41
for el in eventLoopGroup. makeIterator ( ) {
39
42
XCTAssertTrue ( waiter. canBeRun ( on: el) )
@@ -42,11 +45,13 @@ class HTTPConnectionPool_WaiterTests: XCTestCase {
42
45
}
43
46
44
47
private class MockScheduledRequest : HTTPScheduledRequest {
45
- init ( ) { }
48
+ init ( eventLoopPreference: HTTPClient . EventLoopPreference ) {
49
+ self . eventLoopPreference = eventLoopPreference
50
+ }
46
51
47
52
var logger : Logger { preconditionFailure ( " Unimplemented " ) }
48
53
var connectionDeadline : NIODeadline { preconditionFailure ( " Unimplemented " ) }
49
- var eventLoopPreference : HTTPClient . EventLoopPreference { preconditionFailure ( " Unimplemented " ) }
54
+ let eventLoopPreference : HTTPClient . EventLoopPreference
50
55
51
56
func requestWasQueued( _: HTTPRequestScheduler ) {
52
57
preconditionFailure ( " Unimplemented " )
0 commit comments