Selaa lähdekoodia

better error handling

Darien Raymond 9 vuotta sitten
vanhempi
commit
90504fc077
1 muutettua tiedostoa jossa 7 lisäystä ja 2 poistoa
  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()