def test_header_modifier_sets_specified_header_to_given_value(mock_session):
header_modifier = HeaderModifier(CUSTOM_NAME)
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
assert mock_session.headers.get(CUSTOM_NAME) == ORIGINAL_VALUE
def create_header_modifier(value_format):
return HeaderModifier(value_format=value_format)
def test_header_modifier_adds_specified_header(mock_session):
header_modifier = HeaderModifier(CUSTOM_NAME)
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
assert CUSTOM_NAME in mock_session.headers
def test_header_modifier_sets_default_header_to_given_value(mock_session):
header_modifier = HeaderModifier()
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
assert mock_session.headers.get(DEFAULT_HEADER) == ORIGINAL_VALUE
def test_header_modifier_adds_default_header_by_default(mock_session):
header_modifier = HeaderModifier()
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
assert DEFAULT_HEADER in mock_session.headers
def test_header_modifier_constructs_successfully():
assert HeaderModifier()
def test_auth_handler_constructs_successfully():
assert AuthHandler(TokenProvider(), HeaderModifier())
def test_header_modifier_updates_specified_header_if_present(mock_session):
header_modifier = HeaderModifier(CUSTOM_NAME)
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
header_modifier.modify_session(mock_session, UPDATED_VALUE)
assert mock_session.headers.get(CUSTOM_NAME) == UPDATED_VALUE
def test_header_modifier_updates_default_header_if_present(mock_session):
header_modifier = HeaderModifier()
header_modifier.modify_session(mock_session, ORIGINAL_VALUE)
header_modifier.modify_session(mock_session, UPDATED_VALUE)
assert mock_session.headers.get(DEFAULT_HEADER) == UPDATED_VALUE
