diff --git a/proxy/mtproto/auth_test.go b/proxy/mtproto/auth_test.go index e2778d0c..6f392fe1 100644 --- a/proxy/mtproto/auth_test.go +++ b/proxy/mtproto/auth_test.go @@ -6,21 +6,27 @@ import ( "testing" "v2ray.com/core/common" + "v2ray.com/core/common/compare" . "v2ray.com/core/proxy/mtproto" . "v2ray.com/ext/assert" ) func TestInverse(t *testing.T) { - assert := With(t) - + const size = 64 b := make([]byte, 64) - common.Must2(rand.Read(b)) + for b[0] == b[size-1] { + common.Must2(rand.Read(b)) + } bi := Inverse(b) - assert(b[0], NotEquals, bi[0]) + if b[0] == bi[0] { + t.Fatal("seems bytes are not inversed: ", b[0], "vs", bi[0]) + } bii := Inverse(bi) - assert(bii, Equals, b) + if err := compare.BytesEqualWithDetail(bii, b); err != nil { + t.Fatal(err) + } } func TestAuthenticationReadWrite(t *testing.T) {