from types import SimpleNamespace

from auto_reverse.proxy import flow_from_mitm


def _fake_mitm_flow():
    request = SimpleNamespace(
        method="POST", pretty_host="ex.com", path="/api/users?role=admin",
        headers={"content-type": "application/json"}, content=b'{"name": "Ada"}',
        query=SimpleNamespace(fields=[("role", "admin")]),
    )
    response = SimpleNamespace(
        status_code=201, headers={"content-type": "application/json"},
        content=b'{"id": 1}',
    )
    return SimpleNamespace(request=request, response=response, timestamp_start=1.5)


def test_flow_from_mitm_maps_fields():
    captured = flow_from_mitm(_fake_mitm_flow())
    assert captured.method == "POST"
    assert captured.host == "ex.com"
    assert captured.path == "/api/users"
    assert captured.query == {"role": ["admin"]}
    assert captured.status == 201
    assert captured.request_json() == {"name": "Ada"}
    assert captured.response_json() == {"id": 1}


def test_flow_from_mitm_handles_missing_response():
    flow = _fake_mitm_flow()
    flow.response = None
    captured = flow_from_mitm(flow)
    assert captured is None