Explorar o código

better error handling

Darien Raymond %!s(int64=9) %!d(string=hai) anos
pai
achega
90504fc077
Modificáronse 1 ficheiros con 7 adicións e 2 borrados
  1. 7 2
      testing/scenarios/shadowsocks_test.go

+ 7 - 2
testing/scenarios/shadowsocks_test.go

@@ -1,9 +1,9 @@
 package scenarios
 
 import (
+	"fmt"
 	"net"
 	"testing"
-
 	"v2ray.com/core/common/alloc"
 	v2net "v2ray.com/core/common/net"
 	"v2ray.com/core/testing/assert"
@@ -44,16 +44,21 @@ func TestShadowsocksTCP(t *testing.T) {
 
 		response := alloc.NewBuffer().Clear()
 		finished := false
+		expectedResponse := "Processed: " + payload
 		for {
 			_, err := response.FillFrom(conn)
 			assert.Error(err).IsNil()
 			if err != nil {
 				break
 			}
-			if response.String() == "Processed: "+payload {
+			if response.String() == expectedResponse {
 				finished = true
 				break
 			}
+			if response.Len() > len(expectedResponse) {
+				fmt.Printf("Unexpected response: %s\n", response.String())
+				break
+			}
 		}
 		assert.Bool(finished).IsTrue()