diff --git a/test/types/balanced-pool.test-d.ts b/test/types/balanced-pool.test-d.ts index d4628b9487a..d7ccf7be8d4 100644 --- a/test/types/balanced-pool.test-d.ts +++ b/test/types/balanced-pool.test-d.ts @@ -10,6 +10,7 @@ expectAssignable(new BalancedPool('', { factory: () => new Dispatc expectAssignable(new BalancedPool('', { factory: (origin, opts) => new Client(origin, opts) })) expectAssignable(new BalancedPool('', { connections: 1 })) expectAssignable(new BalancedPool(['http://localhost:4242', 'http://www.nodejs.org'])) +expectAssignable(new BalancedPool([new URL('http://localhost:4242'),new URL('http://www.nodejs.org')], {})) { const pool = new BalancedPool('', {}) @@ -21,6 +22,8 @@ expectAssignable(new BalancedPool(['http://localhost:4242', 'http: // upstreams expectAssignable(pool.addUpstream('http://www.nodejs.org')) expectAssignable(pool.removeUpstream('http://www.nodejs.org')) + expectAssignable(pool.addUpstream(new URL('http://www.nodejs.org'))) + expectAssignable(pool.removeUpstream(new URL('http://www.nodejs.org'))) expectAssignable(pool.upstreams) diff --git a/types/balanced-pool.d.ts b/types/balanced-pool.d.ts index b5de726eeca..d1e9375875f 100644 --- a/types/balanced-pool.d.ts +++ b/types/balanced-pool.d.ts @@ -5,10 +5,10 @@ import { URL } from 'url' export default BalancedPool declare class BalancedPool extends Dispatcher { - constructor(url: string | URL | string[], options?: Pool.Options); + constructor(url: string | string[] | URL | URL[], options?: Pool.Options); - addUpstream(upstream: string): BalancedPool; - removeUpstream(upstream: string): BalancedPool; + addUpstream(upstream: string | URL): BalancedPool; + removeUpstream(upstream: string | URL): BalancedPool; upstreams: Array; /** `true` after `pool.close()` has been called. */