lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-01-17 14:57:44 -05:00
Code Issues Releases Wiki Activity

remove use of small buffer in tests

Browse Source
This commit is contained in:
v2ray 2016-08-01 22:59:18 +02:00
parent 3e4bdc4b0c
commit c42ed62fd5
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
4 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
proxy/freedom/freedom_test.go
View File

@ -52,7 +52,7 @@ func TestSinglePacket(t *testing.T) {
traffic := ray.NewRay() traffic := ray.NewRay()
data2Send := "Data to be sent to remote" data2Send := "Data to be sent to remote"
payload := alloc.NewSmallBuffer().Clear().Append([]byte(data2Send)) payload := alloc.NewLocalBuffer(2048).Clear().Append([]byte(data2Send))
go freedom.Dispatch(v2net.TCPDestination(v2net.LocalHostIP, port), payload, traffic) go freedom.Dispatch(v2net.TCPDestination(v2net.LocalHostIP, port), payload, traffic)
traffic.InboundInput().Close() traffic.InboundInput().Close()
@ -78,7 +78,7 @@ func TestUnreachableDestination(t *testing.T) {
}) })
traffic := ray.NewRay() traffic := ray.NewRay()
data2Send := "Data to be sent to remote" data2Send := "Data to be sent to remote"
payload := alloc.NewSmallBuffer().Clear().Append([]byte(data2Send)) payload := alloc.NewLocalBuffer(2048).Clear().Append([]byte(data2Send))
err := freedom.Dispatch(v2net.TCPDestination(v2net.IPAddress([]byte{127, 0, 0, 1}), 128), payload, traffic) err := freedom.Dispatch(v2net.TCPDestination(v2net.IPAddress([]byte{127, 0, 0, 1}), 128), payload, traffic)
assert.Error(err).IsNotNil() assert.Error(err).IsNotNil()

24
proxy/shadowsocks/protocol_test.go
View File

@ -15,7 +15,7 @@ import (
func TestNormalRequestParsing(t *testing.T) { func TestNormalRequestParsing(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
buffer.AppendBytes(1, 127, 0, 0, 1, 0, 80) buffer.AppendBytes(1, 127, 0, 0, 1, 0, 80)
request, err := ReadRequest(buffer, nil, false) request, err := ReadRequest(buffer, nil, false)
@ -28,7 +28,7 @@ func TestNormalRequestParsing(t *testing.T) {
func TestEmptyPayload(t *testing.T) { func TestEmptyPayload(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
_, err := ReadRequest(buffer, nil, false) _, err := ReadRequest(buffer, nil, false)
assert.Error(err).Equals(io.EOF) assert.Error(err).Equals(io.EOF)
} }
@ -36,7 +36,7 @@ func TestEmptyPayload(t *testing.T) {
func TestSingleBytePayload(t *testing.T) { func TestSingleBytePayload(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear().AppendBytes(1) buffer := alloc.NewLocalBuffer(2048).Clear().AppendBytes(1)
_, err := ReadRequest(buffer, nil, false) _, err := ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
} }
@ -44,7 +44,7 @@ func TestSingleBytePayload(t *testing.T) {
func TestWrongAddressType(t *testing.T) { func TestWrongAddressType(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear().AppendBytes(5) buffer := alloc.NewLocalBuffer(2048).Clear().AppendBytes(5)
_, err := ReadRequest(buffer, nil, false) _, err := ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
} }
@ -52,15 +52,15 @@ func TestWrongAddressType(t *testing.T) {
func TestInsufficientAddressRequest(t *testing.T) { func TestInsufficientAddressRequest(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear().AppendBytes(1, 1) buffer := alloc.NewLocalBuffer(2048).Clear().AppendBytes(1, 1)
_, err := ReadRequest(buffer, nil, false) _, err := ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
buffer = alloc.NewSmallBuffer().Clear().AppendBytes(4, 1) buffer = alloc.NewLocalBuffer(2048).Clear().AppendBytes(4, 1)
_, err = ReadRequest(buffer, nil, false) _, err = ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
buffer = alloc.NewSmallBuffer().Clear().AppendBytes(3, 255, 1) buffer = alloc.NewLocalBuffer(2048).Clear().AppendBytes(3, 255, 1)
_, err = ReadRequest(buffer, nil, false) _, err = ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
} }
@ -68,7 +68,7 @@ func TestInsufficientAddressRequest(t *testing.T) {
func TestInsufficientPortRequest(t *testing.T) { func TestInsufficientPortRequest(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear().AppendBytes(1, 1, 2, 3, 4, 5) buffer := alloc.NewLocalBuffer(2048).Clear().AppendBytes(1, 1, 2, 3, 4, 5)
_, err := ReadRequest(buffer, nil, false) _, err := ReadRequest(buffer, nil, false)
assert.Error(err).Equals(transport.ErrCorruptedPacket) assert.Error(err).Equals(transport.ErrCorruptedPacket)
} }
@ -76,7 +76,7 @@ func TestInsufficientPortRequest(t *testing.T) {
func TestOTARequest(t *testing.T) { func TestOTARequest(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
buffer.AppendBytes(0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0, 239, 115, 52, 212, 178, 172, 26, 6, 168, 0) buffer.AppendBytes(0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0, 239, 115, 52, 212, 178, 172, 26, 6, 168, 0)
auth := NewAuthenticator(HeaderKeyGenerator( auth := NewAuthenticator(HeaderKeyGenerator(
@ -91,7 +91,7 @@ func TestOTARequest(t *testing.T) {
func TestInvalidOTARequest(t *testing.T) { func TestInvalidOTARequest(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
buffer.AppendBytes(0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0, 239, 115, 52, 212, 178, 172, 26, 6, 168, 1) buffer.AppendBytes(0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0, 239, 115, 52, 212, 178, 172, 26, 6, 168, 1)
auth := NewAuthenticator(HeaderKeyGenerator( auth := NewAuthenticator(HeaderKeyGenerator(
@ -104,7 +104,7 @@ func TestInvalidOTARequest(t *testing.T) {
func TestUDPRequestParsing(t *testing.T) { func TestUDPRequestParsing(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
buffer.AppendBytes(1, 127, 0, 0, 1, 0, 80, 1, 2, 3, 4, 5, 6) buffer.AppendBytes(1, 127, 0, 0, 1, 0, 80, 1, 2, 3, 4, 5, 6)
request, err := ReadRequest(buffer, nil, true) request, err := ReadRequest(buffer, nil, true)
@ -118,7 +118,7 @@ func TestUDPRequestParsing(t *testing.T) {
func TestUDPRequestWithOTA(t *testing.T) { func TestUDPRequestWithOTA(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
buffer.AppendBytes( buffer.AppendBytes(
0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0, 0x13, 13, 119, 119, 119, 46, 118, 50, 114, 97, 121, 46, 99, 111, 109, 0, 0,
1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0,

2
proxy/socks/protocol/socks4_test.go
View File

@ -31,7 +31,7 @@ func TestSocks4AuthenticationResponseToBytes(t *testing.T) {
response := NewSocks4AuthenticationResponse(byte(0x10), 443, []byte{1, 2, 3, 4}) response := NewSocks4AuthenticationResponse(byte(0x10), 443, []byte{1, 2, 3, 4})
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
defer buffer.Release() defer buffer.Release()
response.Write(buffer) response.Write(buffer)

6
proxy/socks/protocol/socks_test.go
View File

@ -84,7 +84,7 @@ func TestResponseWrite(t *testing.T) {
[16]byte{}, [16]byte{},
v2net.Port(53), v2net.Port(53),
} }
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
defer buffer.Release() defer buffer.Release()
response.Write(buffer) response.Write(buffer)
@ -105,7 +105,7 @@ func TestSetIPv6(t *testing.T) {
response := NewSocks5Response() response := NewSocks5Response()
response.SetIPv6([]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}) response.SetIPv6([]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15})
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
defer buffer.Release() defer buffer.Release()
response.Write(buffer) response.Write(buffer)
assert.Bytes(buffer.Value).Equals([]byte{ assert.Bytes(buffer.Value).Equals([]byte{
@ -118,7 +118,7 @@ func TestSetDomain(t *testing.T) {
response := NewSocks5Response() response := NewSocks5Response()
response.SetDomain("v2ray.com") response.SetDomain("v2ray.com")
buffer := alloc.NewSmallBuffer().Clear() buffer := alloc.NewLocalBuffer(2048).Clear()
defer buffer.Release() defer buffer.Release()
response.Write(buffer) response.Write(buffer)
assert.Bytes(buffer.Value).Equals([]byte{ assert.Bytes(buffer.Value).Equals([]byte{