@@ -29,9 +29,8 @@ describe("StreamableHttpRunner", () => {
29
29
describe ( description , ( ) => {
30
30
beforeAll ( async ( ) => {
31
31
config . httpHeaders = headers ;
32
- runner = new StreamableHttpRunner (
33
- config ,
34
- ( { logger, deviceId } ) => new MCPConnectionManager ( config , driverOptions , logger , deviceId )
32
+ runner = new StreamableHttpRunner ( config , ( { logger, deviceId } ) =>
33
+ Promise . resolve ( new MCPConnectionManager ( config , driverOptions , logger , deviceId ) )
35
34
) ;
36
35
await runner . start ( ) ;
37
36
} ) ;
@@ -113,9 +112,8 @@ describe("StreamableHttpRunner", () => {
113
112
try {
114
113
for ( let i = 0 ; i < 3 ; i ++ ) {
115
114
config . httpPort = 0 ; // Use a random port for each runner
116
- const runner = new StreamableHttpRunner (
117
- config ,
118
- ( { logger, deviceId } ) => new MCPConnectionManager ( config , driverOptions , logger , deviceId )
115
+ const runner = new StreamableHttpRunner ( config , ( { logger, deviceId } ) =>
116
+ Promise . resolve ( new MCPConnectionManager ( config , driverOptions , logger , deviceId ) )
119
117
) ;
120
118
await runner . start ( ) ;
121
119
runners . push ( runner ) ;
@@ -147,7 +145,8 @@ describe("StreamableHttpRunner", () => {
147
145
const logger = new CustomLogger ( ) ;
148
146
const runner = new StreamableHttpRunner (
149
147
config ,
150
- ( { logger, deviceId } ) => new MCPConnectionManager ( config , driverOptions , logger , deviceId ) ,
148
+ ( { logger, deviceId } ) =>
149
+ Promise . resolve ( new MCPConnectionManager ( config , driverOptions , logger , deviceId ) ) ,
151
150
[ logger ]
152
151
) ;
153
152
await runner . start ( ) ;
0 commit comments