106 lines
2.2 KiB
Go
106 lines
2.2 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/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
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
}
|