From 6c9d1b0e8409f5e983ea1e3cf640279aaf413a88 Mon Sep 17 00:00:00 2001
From: Sergey Cherepanov <cheggaaa@gmail.com>
Date: Wed, 7 Jun 2023 20:45:32 +0200
Subject: [PATCH] rpctest pool

---
 net/rpc/rpctest/pool.go | 60 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 net/rpc/rpctest/pool.go

diff --git a/net/rpc/rpctest/pool.go b/net/rpc/rpctest/pool.go
new file mode 100644
index 00000000..0e2b5d1b
--- /dev/null
+++ b/net/rpc/rpctest/pool.go
@@ -0,0 +1,60 @@
+package rpctest
+
+import (
+	"context"
+	"github.com/anyproto/any-sync/app"
+	"github.com/anyproto/any-sync/net"
+	"github.com/anyproto/any-sync/net/peer"
+	"github.com/anyproto/any-sync/net/pool"
+	"sync"
+)
+
+func NewTestPool() *TestPool {
+	return &TestPool{peers: map[string]peer.Peer{}}
+}
+
+type TestPool struct {
+	peers map[string]peer.Peer
+	mu    sync.Mutex
+}
+
+func (t *TestPool) Init(a *app.App) (err error) {
+	return nil
+}
+
+func (t *TestPool) Name() (name string) {
+	return pool.CName
+}
+
+func (t *TestPool) Run(ctx context.Context) (err error) {
+	return nil
+}
+
+func (t *TestPool) Close(ctx context.Context) (err error) {
+	return nil
+}
+
+func (t *TestPool) Get(ctx context.Context, id string) (peer.Peer, error) {
+	t.mu.Lock()
+	defer t.mu.Unlock()
+	if p, ok := t.peers[id]; ok {
+		return p, nil
+	}
+	return nil, net.ErrUnableToConnect
+}
+
+func (t *TestPool) GetOneOf(ctx context.Context, peerIds []string) (peer.Peer, error) {
+	for _, id := range peerIds {
+		if p, err := t.Get(ctx, id); err == nil {
+			return p, nil
+		}
+	}
+	return nil, net.ErrUnableToConnect
+}
+
+func (t *TestPool) AddPeer(ctx context.Context, p peer.Peer) (err error) {
+	t.mu.Lock()
+	defer t.mu.Unlock()
+	t.peers[p.Id()] = p
+	return nil
+}