def test_get_request_id_from_downstream_header(): builder = EnvironBuilder() builder.headers['DOWNSTREAM-REQUEST-ID'] = 'from-downstream' request = CustomRequest(builder.get_environ()) request_id = request._get_request_id('DM-REQUEST-ID', 'DOWNSTREAM-REQUEST-ID') assert request_id == 'from-downstream'
def test_get_request_id_generates_id(uuid4_mock): builder = EnvironBuilder() request = CustomRequest(builder.get_environ()) uuid4_mock.return_value = 'generated' request_id = request._get_request_id('DM-REQUEST-ID', 'DOWNSTREAM-REQUEST-ID') uuid4_mock.assert_called_once() assert request_id == 'generated'
def test_get_request_id_with_no_downstream_header_configured(uuid4_mock): builder = EnvironBuilder() builder.headers[''] = 'from-downstream' request = CustomRequest(builder.get_environ()) uuid4_mock.return_value = 'generated' request_id = request._get_request_id('DM-REQUEST-ID', '') uuid4_mock.assert_called_once() assert request_id == 'generated'