diff --git a/proxy/shadowsocks/protocol_test.go b/proxy/shadowsocks/protocol_test.go index f2cb3c0a..6c28370f 100644 --- a/proxy/shadowsocks/protocol_test.go +++ b/proxy/shadowsocks/protocol_test.go @@ -39,6 +39,7 @@ func TestUDPEncoding(t *testing.T) { assert(decodedData.Bytes(), Equals, data.Bytes()) assert(decodedRequest.Address, Equals, request.Address) assert(decodedRequest.Port, Equals, request.Port) + assert(decodedRequest.Command, Equals, request.Command) } func TestTCPRequest(t *testing.T) { @@ -118,6 +119,7 @@ func TestTCPRequest(t *testing.T) { assert(err, IsNil) assert(decodedRequest.Address, Equals, request.Address) assert(decodedRequest.Port, Equals, request.Port) + assert(decodedRequest.Command, Equals, request.Command) decodedData, err := reader.ReadMultiBuffer() assert(err, IsNil)