mirror of
https://github.com/ipfs/kubo.git
synced 2025-08-06 11:31:54 +08:00
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package mockrouting
|
|
|
|
import (
|
|
dht "github.com/ipfs/go-ipfs/routing/dht"
|
|
"github.com/ipfs/go-ipfs/thirdparty/testutil"
|
|
mocknet "gx/ipfs/QmZ8bCZpMWDbFSh6h2zgTYwrhnjrGM5c9WCzw72SU8p63b/go-libp2p/p2p/net/mock"
|
|
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
|
|
ds "gx/ipfs/QmfQzVugPq1w5shWRcLWSeiHF4a2meBX7yVD8Vw7GWJM9o/go-datastore"
|
|
sync "gx/ipfs/QmfQzVugPq1w5shWRcLWSeiHF4a2meBX7yVD8Vw7GWJM9o/go-datastore/sync"
|
|
)
|
|
|
|
type mocknetserver struct {
|
|
mn mocknet.Mocknet
|
|
}
|
|
|
|
func NewDHTNetwork(mn mocknet.Mocknet) Server {
|
|
return &mocknetserver{
|
|
mn: mn,
|
|
}
|
|
}
|
|
|
|
func (rs *mocknetserver) Client(p testutil.Identity) Client {
|
|
return rs.ClientWithDatastore(context.TODO(), p, ds.NewMapDatastore())
|
|
}
|
|
|
|
func (rs *mocknetserver) ClientWithDatastore(ctx context.Context, p testutil.Identity, ds ds.Datastore) Client {
|
|
|
|
// FIXME AddPeer doesn't appear to be idempotent
|
|
|
|
host, err := rs.mn.AddPeer(p.PrivateKey(), p.Address())
|
|
if err != nil {
|
|
panic("FIXME")
|
|
}
|
|
return dht.NewDHT(ctx, host, sync.MutexWrap(ds))
|
|
}
|
|
|
|
var _ Server = &mocknetserver{}
|