def test_request_id_policy(auto_request_id, request_id_init, request_id_set, request_id_req):
"""Test policy with no other policy and happy path"""
kwargs = {}
if auto_request_id is not None:
kwargs['auto_request_id'] = auto_request_id
if request_id_init != "_unset":
kwargs['request_id'] = request_id_init
request_id_policy = RequestIdPolicy(**kwargs)
if request_id_set != "_unset":
request_id_policy.set_request_id(request_id_set)
request = HttpRequest('GET', 'http://localhost/')
pipeline_request = PipelineRequest(request, PipelineContext(None))
if request_id_req != "_unset":
pipeline_request.context.options['request_id'] = request_id_req
with mock.patch('uuid.uuid1', return_value="VALUE"):
request_id_policy.on_request(pipeline_request)
assert all(v is not None for v in request.headers.values())
if request_id_req != "_unset" and request_id_req:
assert request.headers["x-ms-client-request-id"] == request_id_req
elif not request_id_req:
assert not "x-ms-client-request-id" in request.headers
elif request_id_set != "_unset" and request_id_set:
assert request.headers["x-ms-client-request-id"] == request_id_set
elif not request_id_set:
assert not "x-ms-client-request-id" in request.headers
elif request_id_init != "_unset" and request_id_init:
assert request.headers["x-ms-client-request-id"] == request_id_init
elif not request_id_init:
assert not "x-ms-client-request-id" in request.headers
elif auto_request_id or auto_request_id is None:
assert request.headers["x-ms-client-request-id"] == "VALUE"
else:
assert not "x-ms-client-request-id" in request.headers
def test_request_id_already_exists():
"""Test policy with no other policy and happy path"""
request_id_policy = RequestIdPolicy()
request = HttpRequest('GET', 'http://localhost/')
request.headers["x-ms-client-request-id"] = "VALUE"
pipeline_request = PipelineRequest(request, PipelineContext(None))
request_id_policy.on_request(pipeline_request)
assert request.headers["x-ms-client-request-id"] == "VALUE"
def test_request_id_policy_fix_id():
"""Test policy with no other policy and happy path"""
test_request_id = 'test_request_id'
request_id_policy = RequestIdPolicy(request_id=test_request_id)
request = HttpRequest('GET', 'http://127.0.0.1/')
pipeline_request = PipelineRequest(request, PipelineContext(None))
request_id_policy.on_request(pipeline_request)
assert request.headers["x-ms-client-request-id"] == test_request_id