github.com/GeniusesGroup/libgo@v0.0.0-20220929090155-5ff932cb408e/http/request_test.go

/* For license and copyright information please see the LEGAL file in the code repository */

package http

import (
	"bytes"
	"fmt"
	"reflect"
	"testing"

	"github.com/GeniusesGroup/libgo/compress/raw"
)

type RequestTest struct {
	name   string
	packet []byte
	req    Request // expected parse
	out    Request // parsed one
}

var requestTests = []RequestTest{
	{
		name:   "simple1",
		packet: []byte("GET / HTTP/1.1\r\n" + "\r\n"),
		req: Request{
			method: "GET",
			// uri: uri.URI{
			// 	uri:       "/",
			// 	uriAsByte: []byte("/"),
			// 	scheme:    "",
			// 	authority: "",
			// 	path:      "/",
			// 	query:     "",
			// 	fragment:  "",
			// },
			version: "HTTP/1.1",
			H:       header{},
			body: body{
				Codec: nil,
			},
		},
	}, {
		name: "full1",
		packet: []byte("POST /m?2586547852 HTTP/1.1\r\n" +
			"Accept: text/html\r\n" +
			"Accept-Encoding: gzip, deflate\r\n" +
			"Accept-Language: en,fa;q=0.9\r\n" +
			"Cache-Control: max-age=0\r\n" +
			"Connection: keep-alive\r\n" +
			"Content-Length: 15\r\n" +
			"Content-Type: application/json\r\n" +
			"Host: www.sabz.city\r\n" +
			"Set-Cookie: test\r\n" +
			"Upgrade-Insecure-Requests: 1\r\n" +
			"User-Agent: Mozilla\r\n" +
			"\r\n" +
			`{"Omid":"OMID"}`),
		req: Request{
			method: "POST",
			// uri: uri.URI{
			// 	uri:       "/m?2586547852",
			// 	uriAsByte: []byte("/m?2586547852"),
			// 	scheme:    "",
			// 	authority: "",
			// 	path:      "/m",
			// 	query:     "2586547852",
			// 	fragment:  "",
			// },
			version: "HTTP/1.1",
			H: header{
				headers: map[string][]string{
					"Accept":                    {"text/html"},
					"Accept-Encoding":           {"gzip, deflate"},
					"Accept-Language":           {"en,fa;q=0.9"},
					"Cache-Control":             {"max-age=0"},
					"Connection":                {"keep-alive"},
					"Content-Length":            {"15"},
					"Content-Type":              {"application/json"},
					"Host":                      {"www.sabz.city"},
					"Set-Cookie":                {"test"},
					"Upgrade-Insecure-Requests": {"1"},
					"User-Agent":                {"Mozilla"},
				},
			},
			body: body{
				Codec: &raw.ComDecom{
					Data: []byte(`{"Omid":"OMID"}`),
				},
			},
		},
	}, {
		name: "without-body",
		packet: []byte("POST /m?2586547852 HTTP/1.1\r\n" +
			"Accept: text/html\r\n" +
			"Accept-Encoding: gzip, deflate\r\n" +
			"Accept-Language: en,fa;q=0.9\r\n" +
			"Cache-Control: max-age=0\r\n" +
			"Connection: keep-alive\r\n" +
			"Content-Length: 15\r\n" +
			"Content-Type: application/json\r\n" +
			"Host: www.sabz.city\r\n" +
			"Set-Cookie: test\r\n" +
			"Upgrade-Insecure-Requests: 1\r\n" +
			"User-Agent: Mozilla\r\n" +
			"\r\n"),
		req: Request{
			method: "POST",
			// uri: uri.URI{
			// 	uri:       "/m?2586547852",
			// 	uriAsByte: []byte("/m?2586547852"),
			// 	scheme:    "",
			// 	authority: "",
			// 	path:      "/m",
			// 	query:     "2586547852",
			// 	fragment:  "",
			// },
			version: "HTTP/1.1",
			H: header{
				headers: map[string][]string{
					"Accept":                    {"text/html"},
					"Accept-Encoding":           {"gzip, deflate"},
					"Accept-Language":           {"en,fa;q=0.9"},
					"Cache-Control":             {"max-age=0"},
					"Connection":                {"keep-alive"},
					"Content-Length":            {"15"},
					"Content-Type":              {"application/json"},
					"Host":                      {"www.sabz.city"},
					"Set-Cookie":                {"test"},
					"Upgrade-Insecure-Requests": {"1"},
					"User-Agent":                {"Mozilla"},
				},
			},
			body: body{
				Codec: nil,
			},
		},
	},
}

func init() {
	requestTests[0].req.uri.Set("", "", "/", "", "")
	requestTests[1].req.uri.Set("", "", "/m", "2586547852", "")
	requestTests[2].req.uri.Set("", "", "/m", "2586547852", "")
}

func TestRequest_Unmarshal(t *testing.T) {
	for _, tt := range requestTests {
		t.Run(tt.name, func(t *testing.T) {
			tt.out.Init()
			var _, err = tt.out.Unmarshal(tt.packet)
			if err != nil {
				t.Errorf("Request.Unmarshal() error = %v", err)
			}
			if tt.out.method != tt.req.method {
				t.Errorf("Request.Unmarshal(%q) - Method:\n\tgot  %v\n\twant %v\n", tt.packet, tt.out.method, tt.req.method)
			}
			if tt.out.uri.URI() != tt.req.uri.URI() {
				t.Errorf("Request.Unmarshal(%q) - URI:\n\tgot  %v\n\twant %v\n", tt.packet, tt.out.uri.URI(), tt.req.uri.URI())
			}
			if tt.out.version != tt.req.version {
				t.Errorf("Request.Unmarshal(%q) - Version:\n\tgot  %v\n\twant %v\n", tt.packet, tt.out.version, tt.req.version)
			}
			// if !reflect.DeepEqual(tt.out.header.headers, tt.req.header.headers) {
			// 	t.Errorf("Request.Unmarshal(%q):\n\tgot  %v\n\twant %v\n", tt.packet, tt.out.header.headers, tt.req.header.headers)
			// }
			if !reflect.DeepEqual(tt.out.body, tt.req.body) {
				t.Errorf("Request.Unmarshal(%q):\n\tgot  %v\n\twant %v\n", tt.packet, tt.out.body, tt.req.body)
			}
		})
	}

	// s := req.header.GetSetCookies()[0]
	// err := s.CheckAndSanitize()
	// fmt.Fprintf(os.Stderr, "%v\n", err)
	// fmt.Fprintf(os.Stderr, "%v\n", s)
}

func TestRequest_Marshal(t *testing.T) {
	for _, tt := range requestTests {
		t.Run(tt.name, func(t *testing.T) {
			var httpPacket, _ = tt.req.Marshal()
			fmt.Println("cap--len of httpPacket:", cap(httpPacket), "--", len(httpPacket))
			fmt.Println("cap--len of tt.packet:", cap(tt.packet), "--", len(tt.packet))
			if httpPacket == nil {
				t.Errorf("Request.Marshal() return nil!")
			}
			if !bytes.Equal(tt.packet, httpPacket) {
				fmt.Println("encoded not same with original or just encode headers in not same order! ", tt.name)
				fmt.Println(string(httpPacket))
				// t.Errorf("Request.Marshal(%q):\n\tgot  %v\n\twant %v\n", tt.req, httpPacket, tt.packet)
			}
		})
	}
}

func BenchmarkRequest_Unmarshal(b *testing.B) {
	for n := 0; n < b.N; n++ {
		var httpReq Request
		httpReq.Init()
		httpReq.Unmarshal(requestTests[1].packet)
	}
}

func BenchmarkRequest_Marshal(b *testing.B) {
	for n := 0; n < b.N; n++ {
		requestTests[1].req.Marshal()
	}
}