This commit is contained in:
Sergey Cherepanov 2023-02-02 21:16:59 +03:00 committed by Mikhail Iudin
parent 984dc91f14
commit 9a043e39a0
No known key found for this signature in database
GPG Key ID: FAAAA8BAABDFF1C0

View File

@ -48,10 +48,7 @@ func (t *TestPool) Get(ctx context.Context, id string) (peer.Peer, error) {
func (t *TestPool) Dial(ctx context.Context, id string) (peer.Peer, error) {
t.mu.Lock()
defer t.mu.Unlock()
if t.ts == nil {
return nil, ErrCantConnect
}
return &testPeer{id: id, Conn: t.ts.Dial(ctx)}, nil
return t.Get(ctx, id)
}
func (t *TestPool) GetOneOf(ctx context.Context, peerIds []string) (peer.Peer, error) {