mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-08 22:57:50 +08:00
d7dab3afea
For the rest of the packages in util, move them to thirdparty and update the references. util is gone! License: MIT Signed-off-by: Jeromy <jeromyj@gmail.com>
133 lines
3.3 KiB
Go
133 lines
3.3 KiB
Go
package integrationtest
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
|
|
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
|
|
|
|
core "github.com/ipfs/go-ipfs/core"
|
|
coreunix "github.com/ipfs/go-ipfs/core/coreunix"
|
|
mock "github.com/ipfs/go-ipfs/core/mock"
|
|
testutil "github.com/ipfs/go-ipfs/thirdparty/testutil"
|
|
"github.com/ipfs/go-ipfs/thirdparty/unit"
|
|
mocknet "gx/ipfs/QmUBogf4nUefBjmYjn6jfsfPJRkmDGSeMhNj4usRKq69f4/go-libp2p/p2p/net/mock"
|
|
"gx/ipfs/QmUBogf4nUefBjmYjn6jfsfPJRkmDGSeMhNj4usRKq69f4/go-libp2p/p2p/peer"
|
|
)
|
|
|
|
func TestThreeLeggedCatTransfer(t *testing.T) {
|
|
conf := testutil.LatencyConfig{
|
|
NetworkLatency: 0,
|
|
RoutingLatency: 0,
|
|
BlockstoreLatency: 0,
|
|
}
|
|
if err := RunThreeLeggedCat(RandomBytes(100*unit.MB), conf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestThreeLeggedCatDegenerateSlowBlockstore(t *testing.T) {
|
|
SkipUnlessEpic(t)
|
|
conf := testutil.LatencyConfig{BlockstoreLatency: 50 * time.Millisecond}
|
|
if err := RunThreeLeggedCat(RandomBytes(1*unit.KB), conf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestThreeLeggedCatDegenerateSlowNetwork(t *testing.T) {
|
|
SkipUnlessEpic(t)
|
|
conf := testutil.LatencyConfig{NetworkLatency: 400 * time.Millisecond}
|
|
if err := RunThreeLeggedCat(RandomBytes(1*unit.KB), conf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestThreeLeggedCatDegenerateSlowRouting(t *testing.T) {
|
|
SkipUnlessEpic(t)
|
|
conf := testutil.LatencyConfig{RoutingLatency: 400 * time.Millisecond}
|
|
if err := RunThreeLeggedCat(RandomBytes(1*unit.KB), conf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestThreeLeggedCat100MBMacbookCoastToCoast(t *testing.T) {
|
|
SkipUnlessEpic(t)
|
|
conf := testutil.LatencyConfig{}.NetworkNYtoSF().BlockstoreSlowSSD2014().RoutingSlow()
|
|
if err := RunThreeLeggedCat(RandomBytes(100*unit.MB), conf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func RunThreeLeggedCat(data []byte, conf testutil.LatencyConfig) error {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
const numPeers = 3
|
|
|
|
// create network
|
|
mn := mocknet.New(ctx)
|
|
mn.SetLinkDefaults(mocknet.LinkOptions{
|
|
Latency: conf.NetworkLatency,
|
|
// TODO add to conf. This is tricky because we want 0 values to be functional.
|
|
Bandwidth: math.MaxInt32,
|
|
})
|
|
|
|
bootstrap, err := core.NewNode(ctx, &core.BuildCfg{
|
|
Online: true,
|
|
Host: mock.MockHostOption(mn),
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer bootstrap.Close()
|
|
|
|
adder, err := core.NewNode(ctx, &core.BuildCfg{
|
|
Online: true,
|
|
Host: mock.MockHostOption(mn),
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer adder.Close()
|
|
|
|
catter, err := core.NewNode(ctx, &core.BuildCfg{
|
|
Online: true,
|
|
Host: mock.MockHostOption(mn),
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer catter.Close()
|
|
mn.LinkAll()
|
|
|
|
bis := bootstrap.Peerstore.PeerInfo(bootstrap.PeerHost.ID())
|
|
bcfg := core.BootstrapConfigWithPeers([]peer.PeerInfo{bis})
|
|
if err := adder.Bootstrap(bcfg); err != nil {
|
|
return err
|
|
}
|
|
if err := catter.Bootstrap(bcfg); err != nil {
|
|
return err
|
|
}
|
|
|
|
added, err := coreunix.Add(adder, bytes.NewReader(data))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
readerCatted, err := coreunix.Cat(ctx, catter, added)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// verify
|
|
bufout := new(bytes.Buffer)
|
|
io.Copy(bufout, readerCatted)
|
|
if 0 != bytes.Compare(bufout.Bytes(), data) {
|
|
return errors.New("catted data does not match added data")
|
|
}
|
|
return nil
|
|
}
|