42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import unittest
|
|
|
|
from mitmproxy.test import tflow, tutils
|
|
|
|
from hippolyzer.lib.base.datatypes import UUID
|
|
from hippolyzer.lib.proxy.http_flow import HippoHTTPFlow
|
|
from hippolyzer.lib.proxy.http_proxy import SerializedCapData
|
|
from hippolyzer.lib.proxy.message_logger import HTTPMessageLogEntry
|
|
from hippolyzer.lib.proxy.sessions import SessionManager
|
|
|
|
|
|
class TestHTTPFlows(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.session_manager = SessionManager()
|
|
self.session = self.session = self.session_manager.create_session({
|
|
"session_id": UUID.random(),
|
|
"secure_session_id": UUID.random(),
|
|
"agent_id": UUID.random(),
|
|
"circuit_code": 0,
|
|
"sim_ip": "127.0.0.1",
|
|
"sim_port": "1",
|
|
"seed_capability": "https://test.localhost:4/foo",
|
|
})
|
|
|
|
def test_request_formatting(self):
|
|
req = tutils.treq(host="example.com", port=80)
|
|
resp = tutils.tresp()
|
|
fake_flow = tflow.tflow(req=req, resp=resp)
|
|
fake_flow.metadata["cap_data_ser"] = SerializedCapData(
|
|
cap_name="FakeCap",
|
|
session_id=str(self.session.id),
|
|
base_url="http://example.com",
|
|
)
|
|
flow = HippoHTTPFlow.from_state(fake_flow.get_state(), self.session_manager)
|
|
entry = HTTPMessageLogEntry(flow)
|
|
self.assertEqual(entry.request(beautify=True), """GET [[FakeCap]]/path HTTP/1.1\r
|
|
# http://example.com/path\r
|
|
header: qvalue\r
|
|
content-length: 7\r
|
|
\r
|
|
content""")
|