mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-16 04:02:05 +08:00
2c8cb9fc75
humanize bandwidth output instrument conn.Conn for bandwidth metrics add poll command for continuous bandwidth reporting move bandwidth tracking onto multiaddr net connections another mild refactor of recording locations address concerns from PR lower mock nodes in race test due to increased goroutines per connection
75 lines
1.3 KiB
Go
75 lines
1.3 KiB
Go
package meterstream
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
inet "github.com/ipfs/go-ipfs/p2p/net"
|
|
peer "github.com/ipfs/go-ipfs/p2p/peer"
|
|
protocol "github.com/ipfs/go-ipfs/p2p/protocol"
|
|
u "github.com/ipfs/go-ipfs/util"
|
|
)
|
|
|
|
type FakeStream struct {
|
|
ReadBuf io.Reader
|
|
inet.Stream
|
|
}
|
|
|
|
func (fs *FakeStream) Read(b []byte) (int, error) {
|
|
return fs.ReadBuf.Read(b)
|
|
}
|
|
|
|
func (fs *FakeStream) Write(b []byte) (int, error) {
|
|
return len(b), nil
|
|
}
|
|
|
|
func TestCallbacksWork(t *testing.T) {
|
|
fake := new(FakeStream)
|
|
|
|
var sent int64
|
|
var recv int64
|
|
|
|
sentCB := func(n int64, proto protocol.ID, p peer.ID) {
|
|
sent += n
|
|
}
|
|
|
|
recvCB := func(n int64, proto protocol.ID, p peer.ID) {
|
|
recv += n
|
|
}
|
|
|
|
ms := newMeteredStream(fake, protocol.ID("TEST"), peer.ID("PEER"), recvCB, sentCB)
|
|
|
|
toWrite := int64(100000)
|
|
toRead := int64(100000)
|
|
|
|
fake.ReadBuf = io.LimitReader(u.NewTimeSeededRand(), toRead)
|
|
writeData := io.LimitReader(u.NewTimeSeededRand(), toWrite)
|
|
|
|
n, err := io.Copy(ms, writeData)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if n != toWrite {
|
|
t.Fatal("incorrect write amount")
|
|
}
|
|
|
|
if toWrite != sent {
|
|
t.Fatal("incorrectly reported writes", toWrite, sent)
|
|
}
|
|
|
|
n, err = io.Copy(ioutil.Discard, ms)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if n != toRead {
|
|
t.Fatal("incorrect read amount")
|
|
}
|
|
|
|
if toRead != recv {
|
|
t.Fatal("incorrectly reported reads")
|
|
}
|
|
}
|