lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-10-07 17:44:02 -04:00
Code Issues Releases Wiki Activity

fix test break

Browse Source
This commit is contained in:
v2ray
2016-02-04 13:27:35 +01:00
parent 3212337aa3
commit 4f59d6847f
2 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
proxy/dokodemo/dokodemo_test.go
View File

@@ -16,7 +16,7 @@ import (
func TestDokodemoTCP(t *testing.T) {
v2testing.Current(t)
testPacketDispatcher := &testdispatcher.TestPacketDispatcher{}
testPacketDispatcher := testdispatcher.NewTestPacketDispatcher(nil)
data2Send := "Data to be sent to remote."
@@ -42,21 +42,23 @@ func TestDokodemoTCP(t *testing.T) {
tcpClient.Write([]byte(data2Send))
tcpClient.CloseWrite()
lastPacket := <-testPacketDispatcher.LastPacket
response := make([]byte, 1024)
nBytes, err := tcpClient.Read(response)
assert.Error(err).IsNil()
tcpClient.Close()
assert.StringLiteral("Processed: " + data2Send).Equals(string(response[:nBytes]))
assert.Bool(testPacketDispatcher.LastPacket.Destination().IsTCP()).IsTrue()
netassert.Address(testPacketDispatcher.LastPacket.Destination().Address()).Equals(v2net.IPAddress([]byte{1, 2, 3, 4}))
netassert.Port(testPacketDispatcher.LastPacket.Destination().Port()).Equals(128)
assert.Bool(lastPacket.Destination().IsTCP()).IsTrue()
netassert.Address(lastPacket.Destination().Address()).Equals(v2net.IPAddress([]byte{1, 2, 3, 4}))
netassert.Port(lastPacket.Destination().Port()).Equals(128)
}
func TestDokodemoUDP(t *testing.T) {
v2testing.Current(t)
testPacketDispatcher := &testdispatcher.TestPacketDispatcher{}
testPacketDispatcher := testdispatcher.NewTestPacketDispatcher(nil)
data2Send := "Data to be sent to remote."
@@ -80,14 +82,12 @@ func TestDokodemoUDP(t *testing.T) {
assert.Error(err).IsNil()
udpClient.Write([]byte(data2Send))
response := make([]byte, 1024)
nBytes, err := udpClient.Read(response)
assert.Error(err).IsNil()
udpClient.Close()
assert.StringLiteral("Processed: " + data2Send).Equals(string(response[:nBytes]))
assert.Bool(testPacketDispatcher.LastPacket.Destination().IsUDP()).IsTrue()
netassert.Address(testPacketDispatcher.LastPacket.Destination().Address()).Equals(v2net.IPAddress([]byte{5, 6, 7, 8}))
netassert.Port(testPacketDispatcher.LastPacket.Destination().Port()).Equals(256)
lastPacket := <-testPacketDispatcher.LastPacket
assert.StringLiteral(data2Send).Equals(string(lastPacket.Chunk().Value))
assert.Bool(lastPacket.Destination().IsUDP()).IsTrue()
netassert.Address(lastPacket.Destination().Address()).Equals(v2net.IPAddress([]byte{5, 6, 7, 8}))
netassert.Port(lastPacket.Destination().Port()).Equals(256)
}