def test_auto_length(self):
data = """POST /action.php?action=login&code=2 HTTP/1.1
Content-Length: 11
Connection: close
username=1&pass=223"""
data = data.replace("\n", "\r\n")
http = parseHTTP(data)
http = parseHTTP(http.build())
self.assertEqual(int(http.headers["content-length"][1]), 19)
def test_post_request(self):
data = """POST /action.php?action=login&code=2 HTTP/1.1
Host: web.test.com
User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:70.0) Gecko/20100101 Firefox/70.0
Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8
Accept-Language: zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2
Content-Type: application/x-www-form-urlencoded
Content-Length: 19
Origin: http://web.test.com
Connection: close
Referer: http://web.test.com/action.php?action=login
Cookie: csrftoken=37a35c64931a61e2fb4051035a94f7ff
Upgrade-Insecure-Requests: 1
username=1&pass=223"""
data = data.replace("\n", "\r\n")
http = parseHTTP(data)
self.assertEqual(http.method, "POST")
self.assertEqual(http.url.path, "/action.php")
self.assertEqual(http.version, "HTTP/1.1")
self.assertEqual(http.build_url(), "/action.php?action=login&code=2")
self.assertEqual(http.body, "username=1&pass=223")
self.assertEqual(http.state, 6)
def test_wrong(self):
data = "test"
http = parseHTTP(data)
self.assertEqual(http.method, None)
self.assertEqual(http.state, 1)
def test_get_request_state2(self):
data = "GET / HTTP/1.1\r\n"
self.assertEqual(parseHTTP(data).state, 2)
def test_change_request(self):
data = """POST /action.php HTTP/1.1\r\n\r\n"""
data_new = """POST /test.php HTTP/1.1\r\n\r\n"""
http = parseHTTP(data)
http.url = http.url._replace(path="/test.php")
self.assertEqual(http.build(), data_new)