fix mux test

v2ray · Apr 5, 2018 · 7315b2c · 7315b2c
1 parent 435be61
commit 7315b2c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/app/proxyman/mux/mux_test.go b/app/proxyman/mux/mux_test.go
@@ -61,7 +61,6 @@ func TestReaderWriter(t *testing.T) {
 	writer2.Close()
 
 	bytesReader := buf.NewBufferedReader(stream)
-	streamReader := NewStreamReader(bytesReader)
 
 	meta, err := ReadMetadata(bytesReader)
 	assert(err, IsNil)
@@ -70,7 +69,7 @@ func TestReaderWriter(t *testing.T) {
 	assert(meta.Target, Equals, dest)
 	assert(byte(meta.Option), Equals, byte(OptionData))
 
-	data, err := readAll(streamReader)
+	data, err := readAll(NewStreamReader(bytesReader))
 	assert(err, IsNil)
 	assert(len(data), Equals, 1)
 	assert(data[0].String(), Equals, "abcd")
@@ -88,7 +87,7 @@ func TestReaderWriter(t *testing.T) {
 	assert(meta.SessionID, Equals, uint16(1))
 	assert(byte(meta.Option), Equals, byte(1))
 
-	data, err = readAll(streamReader)
+	data, err = readAll(NewStreamReader(bytesReader))
 	assert(err, IsNil)
 	assert(len(data), Equals, 1)
 	assert(data[0].String(), Equals, "efgh")
@@ -100,7 +99,7 @@ func TestReaderWriter(t *testing.T) {
 	assert(byte(meta.Option), Equals, byte(1))
 	assert(meta.Target, Equals, dest3)
 
-	data, err = readAll(streamReader)
+	data, err = readAll(NewStreamReader(bytesReader))
 	assert(err, IsNil)
 	assert(len(data), Equals, 1)
 	assert(data[0].String(), Equals, "x")
@@ -123,7 +122,7 @@ func TestReaderWriter(t *testing.T) {
 	assert(meta.SessionID, Equals, uint16(2))
 	assert(byte(meta.Option), Equals, byte(1))
 
-	data, err = readAll(streamReader)
+	data, err = readAll(NewStreamReader(bytesReader))
 	assert(err, IsNil)
 	assert(len(data), Equals, 1)
 	assert(data[0].String(), Equals, "y")