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