lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-27 01:45:23 +00:00
Code Issues Releases Wiki Activity

fix test break

Browse Source
This commit is contained in:
Darien Raymond 2018-04-17 00:40:27 +02:00
parent 5d23604713
commit ef1d9d24cd
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
3 changed files with 24 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
app/proxyman/mux/mux_test.go
View File

@ -1,7 +1,6 @@
package mux_test
import (
"context"
"io"
"testing"
@ -9,7 +8,7 @@ import (
"v2ray.com/core/common/buf"
"v2ray.com/core/common/net"
"v2ray.com/core/common/protocol"
"v2ray.com/core/transport/ray"
"v2ray.com/core/transport/pipe"
. "v2ray.com/ext/assert"
)
@ -31,16 +30,16 @@ func readAll(reader buf.Reader) (buf.MultiBuffer, error) {
func TestReaderWriter(t *testing.T) {
assert := With(t)
stream := ray.NewStream(context.Background())
pReader, pWriter := pipe.New()
dest := net.TCPDestination(net.DomainAddress("v2ray.com"), 80)
writer := NewWriter(1, dest, stream, protocol.TransferTypeStream)
writer := NewWriter(1, dest, pWriter, protocol.TransferTypeStream)
dest2 := net.TCPDestination(net.LocalHostIP, 443)
writer2 := NewWriter(2, dest2, stream, protocol.TransferTypeStream)
writer2 := NewWriter(2, dest2, pWriter, protocol.TransferTypeStream)
dest3 := net.TCPDestination(net.LocalHostIPv6, 18374)
writer3 := NewWriter(3, dest3, stream, protocol.TransferTypeStream)
writer3 := NewWriter(3, dest3, pWriter, protocol.TransferTypeStream)
writePayload := func(writer *Writer, payload ...byte) error {
b := buf.New()
@ -60,7 +59,7 @@ func TestReaderWriter(t *testing.T) {
assert(writePayload(writer2, 'y'), IsNil)
writer2.Close()
bytesReader := buf.NewBufferedReader(stream)
bytesReader := buf.NewBufferedReader(pReader)
meta, err := ReadMetadata(bytesReader)
assert(err, IsNil)
@ -133,7 +132,7 @@ func TestReaderWriter(t *testing.T) {
assert(meta.SessionID, Equals, uint16(2))
assert(byte(meta.Option), Equals, byte(0))
stream.Close()
pWriter.Close()
meta, err = ReadMetadata(bytesReader)
assert(err, IsNotNil)

25
common/buf/reader_test.go
View File

@ -2,12 +2,11 @@ package buf_test
import (
"bytes"
"context"
"io"
"testing"
. "v2ray.com/core/common/buf"
"v2ray.com/core/transport/ray"
"v2ray.com/core/transport/pipe"
. "v2ray.com/ext/assert"
)
@ -39,24 +38,24 @@ func TestAdaptiveReader(t *testing.T) {
func TestBytesReaderWriteTo(t *testing.T) {
assert := With(t)
stream := ray.NewStream(context.Background())
reader := NewBufferedReader(stream)
pReader, pWriter := pipe.New()
reader := NewBufferedReader(pReader)
b1 := New()
b1.AppendBytes('a', 'b', 'c')
b2 := New()
b2.AppendBytes('e', 'f', 'g')
assert(stream.WriteMultiBuffer(NewMultiBufferValue(b1, b2)), IsNil)
stream.Close()
assert(pWriter.WriteMultiBuffer(NewMultiBufferValue(b1, b2)), IsNil)
pWriter.Close()
stream2 := ray.NewStream(context.Background())
writer := NewBufferedWriter(stream2)
pReader2, pWriter2 := pipe.New()
writer := NewBufferedWriter(pWriter2)
writer.SetBuffered(false)
nBytes, err := io.Copy(writer, reader)
assert(err, IsNil)
assert(nBytes, Equals, int64(6))
mb, err := stream2.ReadMultiBuffer()
mb, err := pReader2.ReadMultiBuffer()
assert(err, IsNil)
assert(len(mb), Equals, 2)
assert(mb[0].String(), Equals, "abc")
@ -66,14 +65,14 @@ func TestBytesReaderWriteTo(t *testing.T) {
func TestBytesReaderMultiBuffer(t *testing.T) {
assert := With(t)
stream := ray.NewStream(context.Background())
reader := NewBufferedReader(stream)
pReader, pWriter := pipe.New()
reader := NewBufferedReader(pReader)
b1 := New()
b1.AppendBytes('a', 'b', 'c')
b2 := New()
b2.AppendBytes('e', 'f', 'g')
assert(stream.WriteMultiBuffer(NewMultiBufferValue(b1, b2)), IsNil)
stream.Close()
assert(pWriter.WriteMultiBuffer(NewMultiBufferValue(b1, b2)), IsNil)
pWriter.Close()
mbReader := NewReader(reader)
mb, err := mbReader.ReadMultiBuffer()

12
common/buf/writer_test.go
View File

@ -4,14 +4,12 @@ import (
"bufio"
"bytes"
"crypto/rand"
"testing"
"context"
"io"
"testing"
"v2ray.com/core/common"
. "v2ray.com/core/common/buf"
"v2ray.com/core/transport/ray"
"v2ray.com/core/transport/pipe"
. "v2ray.com/ext/assert"
)
@ -36,16 +34,16 @@ func TestWriter(t *testing.T) {
func TestBytesWriterReadFrom(t *testing.T) {
assert := With(t)
cache := ray.NewStream(context.Background())
pReader, pWriter := pipe.New()
const size = 50000
reader := bufio.NewReader(io.LimitReader(rand.Reader, size))
writer := NewBufferedWriter(cache)
writer := NewBufferedWriter(pWriter)
writer.SetBuffered(false)
nBytes, err := reader.WriteTo(writer)
assert(nBytes, Equals, int64(size))
assert(err, IsNil)
mb, err := cache.ReadMultiBuffer()
mb, err := pReader.ReadMultiBuffer()
assert(err, IsNil)
assert(mb.Len(), Equals, int32(size))
}