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'
