projectdiscovery/httpx
1
1
Fork 0
mirror of https://github.com/projectdiscovery/httpx.git synced 2024-12-01 04:08:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

adding more tests

Browse Source
This commit is contained in:
mzack 2021-08-23 15:37:46 +02:00
parent 8f7aebef36
commit 4a2a6d31d1
3 changed files with 57 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
cmd/integration-test/http.go
View File

@ -11,21 +11,30 @@ import (
)
var httpTestcases = map[string]testutils.TestCase{
"Standard HTTP GET Request": &standardHttpGet{},
"Standard HTTPS GET Request": &standardHttpGet{tls: true},
"Regression test for: https://github.com/projectdiscovery/httpx/issues/276": &issue276{},
"Regression test for: https://github.com/projectdiscovery/httpx/issues/277": &issue277{},
"Regression test for: https://github.com/projectdiscovery/httpx/issues/303": &issue303{},
"Standard HTTP GET Request": &standardHttpGet{},
"Standard HTTPS GET Request": &standardHttpGet{tls: true},
"Raw HTTP GET Request": &standardHttpGet{unsafe: true},
"Raw request with non standard rfc path via stdin": &standardHttpGet{unsafe: true, stdinPath: "/%invalid"},
"Raw request with non standard rfc path via cli flag": &standardHttpGet{unsafe: true, path: "/%invalid"},
"Regression test for: https://github.com/projectdiscovery/httpx/issues/363": &issue363{}, // infinite redirect
"Regression test for: https://github.com/projectdiscovery/httpx/issues/276": &issue276{}, // full path with port in output
"Regression test for: https://github.com/projectdiscovery/httpx/issues/277": &issue277{}, // scheme://host:port via stdin
"Regression test for: https://github.com/projectdiscovery/httpx/issues/303": &issue303{}, // misconfigured gzip header with uncompressed body
}
type standardHttpGet struct {
tls bool
tls bool
unsafe bool
stdinPath string
path string
expectedOutput string
}
func (h *standardHttpGet) Execute() error {
router := httprouter.New()
router.GET("/", httprouter.Handle(func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
fmt.Fprintf(w, "This is a test")
r.Close = true
}))
var ts *httptest.Server
if h.tls {
@ -34,14 +43,31 @@ func (h *standardHttpGet) Execute() error {
ts = httptest.NewServer(router)
}
defer ts.Close()
var extra []string
if h.unsafe {
extra = append(extra, "-unsafe")
}
if h.path != "" {
extra = append(extra, "-path", "\""+h.path+"\"")
}
results, err := testutils.RunHttpxAndGetResults(ts.URL, debug)
URL := ts.URL
if h.stdinPath != "" {
URL += h.stdinPath
}
results, err := testutils.RunHttpxAndGetResults(URL, debug, extra...)
if err != nil {
return err
}
if len(results) != 1 {
return errIncorrectResultsCount(results)
}
if h.expectedOutput != "" && !strings.EqualFold(results[0], h.expectedOutput) {
return errIncorrectResult(results[0], h.expectedOutput)
}
return nil
}
@ -70,7 +96,6 @@ func (h *issue276) Execute() error {
// status code
// title
expected := ts.URL + "/redirect" + " [302] [Object moved]"
// log.Fatal(results[0], expected)
if !strings.EqualFold(results[0], expected) {
return errIncorrectResult(results[0], expected)
}
@ -136,3 +161,26 @@ func (h *issue303) Execute() error {
}
return nil
}
type issue363 struct{}
func (h *issue363) Execute() error {
var ts *httptest.Server
router := httprouter.New()
router.GET("/redirect", httprouter.Handle(func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
w.Header().Add("Location", ts.URL+"/redirect")
w.WriteHeader(302)
fmt.Fprintf(w, "<html><body><title>Object moved</title></body></html>")
}))
ts = httptest.NewServer(router)
defer ts.Close()
results, err := testutils.RunHttpxAndGetResults(ts.URL+"/redirect", debug, "-no-color", "-follow-redirects")
if err != nil {
return err
}
if len(results) != 1 {
return errIncorrectResultsCount(results)
}
return nil
}

1
integration_tests/run.sh
View File

@ -9,7 +9,6 @@ go build
mv integration-test ../../integration_tests/integration-test
cd ../../integration_tests
./integration-test
rm integration-test httpx 2>/dev/null
if [ $? -eq 0 ]
then
exit 0

2
internal/testutils/integration.go
View File

@ -9,7 +9,7 @@ import (
// RunNucleiAndGetResults returns a list of results for a template
func RunHttpxAndGetResults(url string, debug bool, extra ...string) ([]string, error) {
cmd := exec.Command("bash", "-c")
cmdLine := `echo ` + url + ` | ./httpx `
cmdLine := `echo "` + url + `" | ./httpx `
cmdLine += strings.Join(extra, " ")
if debug {
cmdLine += " -debug"