status-im
/
go-libp2p
mirror of https://github.com/status-im/go-libp2p.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

send much more data in integration test

This commit is contained in:
Yusef Napora 2020-01-29 09:05:19 -05:00
parent 8c0011194e
commit 297dd7fae9
1 changed files with 41 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
p2p/security/noise/integration_test.go
View File

@ -1,7 +1,6 @@
package noise package noise
import ( import (
"bufio"
"context" "context"
"crypto/rand" "crypto/rand"
"fmt" "fmt"
@ -12,6 +11,7 @@ import (
"github.com/libp2p/go-libp2p-core/peer" "github.com/libp2p/go-libp2p-core/peer"
ma "github.com/multiformats/go-multiaddr" ma "github.com/multiformats/go-multiaddr"
"io" "io"
"io/ioutil"
mrand "math/rand" mrand "math/rand"
"testing" "testing"
"time" "time"
@ -111,8 +111,8 @@ func TestLibp2pIntegration_NoPipes(t *testing.T) {
defer hb.Close() defer hb.Close()
ha.SetStreamHandler(testProtocolID, handleStream) ha.SetStreamHandler(testProtocolID, streamHandler(t))
hb.SetStreamHandler(testProtocolID, handleStream) hb.SetStreamHandler(testProtocolID, streamHandler(t))
addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", hb.Addrs()[0].String(), hb.ID())) addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", hb.Addrs()[0].String(), hb.ID()))
if err != nil { if err != nil {
@ -136,7 +136,7 @@ func TestLibp2pIntegration_NoPipes(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
_, err = stream.Write([]byte("hello\n")) err = writeRandomPayloadAndClose(t, stream)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -168,8 +168,8 @@ func TestLibp2pIntegration_WithPipes(t *testing.T) {
defer hb.Close() defer hb.Close()
ha.SetStreamHandler(testProtocolID, handleStream) ha.SetStreamHandler(testProtocolID, streamHandler(t))
hb.SetStreamHandler(testProtocolID, handleStream) hb.SetStreamHandler(testProtocolID, streamHandler(t))
addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", ha.Addrs()[0].String(), ha.ID())) addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", ha.Addrs()[0].String(), ha.ID()))
if err != nil { if err != nil {
@ -193,7 +193,7 @@ func TestLibp2pIntegration_WithPipes(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
_, err = stream.Write([]byte("hello\n")) err = writeRandomPayloadAndClose(t, stream)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -225,8 +225,8 @@ func TestLibp2pIntegration_XXFallback(t *testing.T) {
defer hb.Close() defer hb.Close()
ha.SetStreamHandler(testProtocolID, handleStream) ha.SetStreamHandler(testProtocolID, streamHandler(t))
hb.SetStreamHandler(testProtocolID, handleStream) hb.SetStreamHandler(testProtocolID, streamHandler(t))
addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", hb.Addrs()[0].String(), hb.ID())) addr, err := ma.NewMultiaddr(fmt.Sprintf("%s/p2p/%s", hb.Addrs()[0].String(), hb.ID()))
if err != nil { if err != nil {
@ -250,7 +250,7 @@ func TestLibp2pIntegration_XXFallback(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
_, err = stream.Write([]byte("hello\n")) err = writeRandomPayloadAndClose(t, stream)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -277,18 +277,38 @@ func TestConstrucingWithMaker(t *testing.T) {
_ = h.Close() _ = h.Close()
} }
func handleStream(stream net.Stream) { func writeRandomPayloadAndClose(t *testing.T, stream net.Stream) error {
defer func() { t.Helper()
if err := stream.Close(); err != nil { size := 1 << 24
log.Error("error closing stream", "err", err) r := mrand.New(mrand.NewSource(42))
} start := time.Now()
}() lr := io.LimitReader(r, int64(size))
rw := bufio.NewReadWriter(bufio.NewReader(stream), bufio.NewWriter(stream)) c, err := io.Copy(stream, lr)
msg, err := rw.Reader.ReadString('\n') elapsed := time.Since(start)
if err != nil { if err != nil {
fmt.Println("stream err", err) return fmt.Errorf("failed to write out bytes: %v", err)
return }
t.Logf("wrote %d bytes in %dms", c, elapsed.Milliseconds())
return stream.Close()
}
func streamHandler(t *testing.T) func(net.Stream) {
return func(stream net.Stream) {
t.Helper()
defer func() {
if err := stream.Close(); err != nil {
t.Error("error closing stream: ", err)
}
}()
start := time.Now()
c, err := io.Copy(ioutil.Discard, stream)
elapsed := time.Since(start)
if err != nil {
t.Error("error reading from stream: ", err)
return
}
t.Logf("read %d bytes in %dms", c, elapsed.Milliseconds())
} }
fmt.Printf("got msg: %s", msg)
} }