mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2025-05-03 21:34:48 -04:00
* Factor out a common TestHtmlParser Looks like I'm doing this in a few different places. * Improve OIDC login test Complete the OIDC login flow, rather than giving up halfway through. * Ensure that OIDC login works with multiple OIDC providers * Fix bugs in handling clientRedirectUrl - don't drop duplicate query-params, or params with no value - allow utf-8 in query-params
This commit is contained in:
parent
a8703819eb
commit
02070c69fa
9 changed files with 189 additions and 86 deletions
53
tests/test_utils/html_parsers.py
Normal file
53
tests/test_utils/html_parsers.py
Normal file
|
@ -0,0 +1,53 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# Copyright 2021 The Matrix.org Foundation C.I.C.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from html.parser import HTMLParser
|
||||
from typing import Dict, Iterable, List, Optional, Tuple
|
||||
|
||||
|
||||
class TestHtmlParser(HTMLParser):
|
||||
"""A generic HTML page parser which extracts useful things from the HTML"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
# a list of links found in the doc
|
||||
self.links = [] # type: List[str]
|
||||
|
||||
# the values of any hidden <input>s: map from name to value
|
||||
self.hiddens = {} # type: Dict[str, Optional[str]]
|
||||
|
||||
# the values of any radio buttons: map from name to list of values
|
||||
self.radios = {} # type: Dict[str, List[Optional[str]]]
|
||||
|
||||
def handle_starttag(
|
||||
self, tag: str, attrs: Iterable[Tuple[str, Optional[str]]]
|
||||
) -> None:
|
||||
attr_dict = dict(attrs)
|
||||
if tag == "a":
|
||||
href = attr_dict["href"]
|
||||
if href:
|
||||
self.links.append(href)
|
||||
elif tag == "input":
|
||||
input_name = attr_dict.get("name")
|
||||
if attr_dict["type"] == "radio":
|
||||
assert input_name
|
||||
self.radios.setdefault(input_name, []).append(attr_dict["value"])
|
||||
elif attr_dict["type"] == "hidden":
|
||||
assert input_name
|
||||
self.hiddens[input_name] = attr_dict["value"]
|
||||
|
||||
def error(_, message):
|
||||
raise AssertionError(message)
|
Loading…
Add table
Add a link
Reference in a new issue