def test_remove_token_from_headers_map2(): t = BaseTransform() t.enumerable_headers.append('ENUM_') header_map = {'A': 'ENUM_'} t.remove_token_from_headers_map(header_map, 'ENUM_') assert 'A' in header_map
def test_remove_token_from_headers_map(): t = BaseTransform() header_map = {'A': '1'} t.remove_token_from_headers_map(header_map, '1') assert 'A' not in header_map