(t *testing.T, version, command uint8, ipStr string, port uint16, shouldFail bool)
| 31 | } |
| 32 | |
| 33 | func createRequest(t *testing.T, version, command uint8, ipStr string, port uint16, shouldFail bool) *Request { |
| 34 | ip := net.ParseIP(ipStr) |
| 35 | data := createRequestData(version, command, ip, port) |
| 36 | reader := bytes.NewReader(data) |
| 37 | req, err := NewRequest(reader) |
| 38 | if shouldFail { |
| 39 | assert.Error(t, err) |
| 40 | return nil |
| 41 | } |
| 42 | assert.NoError(t, err) |
| 43 | assert.True(t, req.Version == socks5Version, "version doesn't match expectation: %v", req.Version) |
| 44 | assert.True(t, req.Command == command, "command doesn't match expectation: %v", req.Command) |
| 45 | assert.True(t, req.DestAddr.Port == int(port), "port doesn't match expectation: %v", req.DestAddr.Port) |
| 46 | assert.True(t, req.DestAddr.IP.String() == ipStr, "ip doesn't match expectation: %v", req.DestAddr.IP.String()) |
| 47 | |
| 48 | return req |
| 49 | } |
| 50 | |
| 51 | func TestValidConnectRequest(t *testing.T) { |
| 52 | createRequest(t, socks5Version, connectCommand, "127.0.0.1", 1337, false) |
no test coverage detected