mirror of https://github.com/daffainfo/nuclei.git
86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
|
package offlinehttp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestReadResponseFromString(t *testing.T) {
|
||
|
|
expectedBody := `<!DOCTYPE html>
|
||
|
|
<html>
|
||
|
|
<head>
|
||
|
|
<title>Firing Range</title>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<h1>Version 0.48</h1>
|
||
|
|
<h1>What is the Firing Range?</h1>
|
||
|
|
<p>
|
||
|
|
</body>
|
||
|
|
</body>
|
||
|
|
</html>`
|
||
|
|
|
||
|
|
t.Run("response", func(t *testing.T) {
|
||
|
|
data := `HTTP/1.1 200 OK
|
||
|
|
Age: 0
|
||
|
|
Cache-Control: public, max-age=600
|
||
|
|
Content-Type: text/html
|
||
|
|
Server: Google Frontend
|
||
|
|
|
||
|
|
<!DOCTYPE html>
|
||
|
|
<html>
|
||
|
|
<head>
|
||
|
|
<title>Firing Range</title>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<h1>Version 0.48</h1>
|
||
|
|
<h1>What is the Firing Range?</h1>
|
||
|
|
<p>
|
||
|
|
</body>
|
||
|
|
</body>
|
||
|
|
</html>`
|
||
|
|
resp, err := readResponseFromString(data)
|
||
|
|
require.Nil(t, err, "could not read response from string")
|
||
|
|
|
||
|
|
respData, err := ioutil.ReadAll(resp.Body)
|
||
|
|
require.Nil(t, err, "could not read response body")
|
||
|
|
require.Equal(t, expectedBody, string(respData), "could not get correct parsed body")
|
||
|
|
require.Equal(t, "Google Frontend", resp.Header.Get("Server"), "could not get correct headers")
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("request-response", func(t *testing.T) {
|
||
|
|
data := `GET http://public-firing-range.appspot.com/ HTTP/1.1
|
||
|
|
Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9
|
||
|
|
Accept-Encoding: gzip, deflate
|
||
|
|
Upgrade-Insecure-Requests: 1
|
||
|
|
User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 11_1_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36
|
||
|
|
|
||
|
|
HTTP/1.1 200 OK
|
||
|
|
Age: 0
|
||
|
|
Cache-Control: public, max-age=600
|
||
|
|
Content-Type: text/html
|
||
|
|
Server: Google Frontend
|
||
|
|
|
||
|
|
<!DOCTYPE html>
|
||
|
|
<html>
|
||
|
|
<head>
|
||
|
|
<title>Firing Range</title>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<h1>Version 0.48</h1>
|
||
|
|
<h1>What is the Firing Range?</h1>
|
||
|
|
<p>
|
||
|
|
</body>
|
||
|
|
</body>
|
||
|
|
</html>`
|
||
|
|
resp, err := readResponseFromString(data)
|
||
|
|
require.Nil(t, err, "could not read response from string")
|
||
|
|
|
||
|
|
respData, err := ioutil.ReadAll(resp.Body)
|
||
|
|
require.Nil(t, err, "could not read response body")
|
||
|
|
require.Equal(t, expectedBody, string(respData), "could not get correct parsed body")
|
||
|
|
require.Equal(t, "Google Frontend", resp.Header.Get("Server"), "could not get correct headers")
|
||
|
|
})
|
||
|
|
}
|