183 lines
4.0 KiB
Go
183 lines
4.0 KiB
Go
package request
|
|
|
|
import (
|
|
"io"
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
// /* (1) Parse request */
|
|
// req, _ := request.Parse(s)
|
|
|
|
// /* (3) Build response */
|
|
// res := req.BuildResponse()
|
|
|
|
// /* (4) Write into socket */
|
|
// _, err := res.Send(s)
|
|
// if err != nil {
|
|
// return nil, fmt.Errorf("Upgrade write error: %s", err)
|
|
// }
|
|
|
|
// if res.GetStatusCode() != 101 {
|
|
// s.Close()
|
|
// return nil, fmt.Errorf("Upgrade error (HTTP %d)\n", res.GetStatusCode())
|
|
// }
|
|
|
|
func TestEOFSocket(t *testing.T){
|
|
|
|
socket := new(bytes.Buffer)
|
|
|
|
_, err := Parse(socket)
|
|
|
|
if err == nil {
|
|
t.Fatalf("Empty socket expected EOF, got no error")
|
|
} else if err != io.ErrUnexpectedEOF {
|
|
t.Fatalf("Empty socket expected EOF, got '%s'", err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestInvalidRequestLine(t *testing.T){
|
|
|
|
socket := new(bytes.Buffer)
|
|
cases := []struct{
|
|
Reqline string
|
|
HasError bool
|
|
}{
|
|
{ "abc", true },
|
|
{ "a c", true },
|
|
{ "a c", true },
|
|
{ "a c", true },
|
|
{ "a b c", true },
|
|
|
|
{ "GET invaliduri HTTP/1.1", true },
|
|
{ "GET /validuri HTTP/1.1", false },
|
|
|
|
{ "POST /validuri HTTP/1.1", true },
|
|
{ "PUT /validuri HTTP/1.1", true },
|
|
{ "DELETE /validuri HTTP/1.1", true },
|
|
{ "OPTIONS /validuri HTTP/1.1", true },
|
|
{ "UNKNOWN /validuri HTTP/1.1", true },
|
|
|
|
{ "GET / HTTP", true },
|
|
{ "GET / HTTP/", true },
|
|
{ "GET / 1.1", true },
|
|
{ "GET / 1", true },
|
|
{ "GET / HTTP/52", true },
|
|
{ "GET / HTTP/1.", true },
|
|
{ "GET / HTTP/.1", true },
|
|
{ "GET / HTTP/1.1", false },
|
|
{ "GET / HTTP/2", false },
|
|
}
|
|
|
|
for ti, tc := range cases {
|
|
|
|
socket.Reset()
|
|
socket.Write( []byte(tc.Reqline) )
|
|
socket.Write( []byte("\r\n\r\n") )
|
|
|
|
_, err := Parse(socket)
|
|
|
|
if !tc.HasError {
|
|
|
|
// no error -> ok
|
|
if err == nil {
|
|
continue
|
|
// error for the end of the request -> ok
|
|
} else if _, ok := err.(*IncompleteRequest); ok {
|
|
continue
|
|
}
|
|
|
|
t.Errorf("[%d] Expected no error", ti)
|
|
}
|
|
|
|
// missing required error -> error
|
|
if tc.HasError && err == nil {
|
|
t.Errorf("[%d] Expected error", ti)
|
|
continue
|
|
}
|
|
|
|
ir, ok := err.(*InvalidRequest);
|
|
|
|
// not InvalidRequest err -> error
|
|
if !ok || ir.Field != "Request-Line" {
|
|
t.Errorf("[%d] expected InvalidRequest", ti)
|
|
continue
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func TestInvalidHost(t *testing.T){
|
|
|
|
requestLine := []byte( "GET / HTTP/1.1\r\n" )
|
|
|
|
socket := new(bytes.Buffer)
|
|
cases := []struct{
|
|
Host string
|
|
HasError bool
|
|
}{
|
|
{ "1", true },
|
|
{ "12", true },
|
|
{ "123", true },
|
|
{ "1234", false },
|
|
|
|
{ "singlevalue", false },
|
|
{ "multi value", true },
|
|
|
|
{ "singlevalue:1", false },
|
|
{ "singlevalue:", true },
|
|
{ "singlevalue:x", true },
|
|
{ "xx:x", true },
|
|
{ ":xxx", true },
|
|
{ "xxx:", true },
|
|
{ "a:12", false },
|
|
|
|
{ "google.com", false },
|
|
{ "8.8.8.8", false },
|
|
{ "google.com:8080", false },
|
|
{ "8.8.8.8:8080", false },
|
|
}
|
|
|
|
for ti, tc := range cases {
|
|
|
|
socket.Reset()
|
|
socket.Write(requestLine)
|
|
socket.Write( []byte("Host: ") )
|
|
socket.Write( []byte(tc.Host) )
|
|
socket.Write( []byte("\r\n\r\n") )
|
|
|
|
_, err := Parse(socket)
|
|
|
|
if !tc.HasError {
|
|
|
|
|
|
// no error -> ok
|
|
if err == nil {
|
|
continue
|
|
// error for the end of the request -> ok
|
|
} else if _, ok := err.(*IncompleteRequest); ok {
|
|
continue
|
|
}
|
|
|
|
t.Errorf("[%d] Expected no error; %s", ti, err)
|
|
}
|
|
|
|
// missing required error -> error
|
|
if tc.HasError && err == nil {
|
|
t.Errorf("[%d] Expected error", ti)
|
|
continue
|
|
}
|
|
|
|
// check if InvalidRequest
|
|
ir, ok := err.(*InvalidRequest);
|
|
|
|
// not InvalidRequest err -> error
|
|
if ok && ir.Field != "Host" {
|
|
t.Errorf("[%d] expected InvalidRequest", ti)
|
|
continue
|
|
}
|
|
|
|
}
|
|
|
|
} |