|  | @@ -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()
 | 
	
		
			
				|  |  |  
 |