html_branches: Fix strict_optional errors.

Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg 2020-07-04 17:59:37 -07:00 committed by Tim Abbott
parent 2fb96e94f1
commit 8a03f8a070
3 changed files with 9 additions and 4 deletions

View File

@ -40,8 +40,5 @@ strict_optional = True
# General exclusions to work on
[mypy-tools.lib.html_branches]
strict_optional = False
[mypy-zthumbor.loaders.helpers]
strict_optional = False

View File

@ -53,7 +53,7 @@ class HtmlTreeBranch:
class Node:
def __init__(self, token: Token, parent: "Optional[Node]") -> None:
def __init__(self, token: Optional[Token], parent: "Optional[Node]") -> None:
# FIXME parent parameter is not used!
self.token = token
self.children: List[Node] = []
@ -133,6 +133,7 @@ def html_branches(text: str, fn: Optional[str] = None) -> List[HtmlTreeBranch]:
branches: List[HtmlTreeBranch] = []
def walk(node: Node, tag_info_list: Sequence[TagInfo] = []) -> None:
assert node.token is not None
info = get_tag_info(node.token)
tag_info_list = [*tag_info_list, info]

View File

@ -49,24 +49,31 @@ class TestHtmlBranches(unittest.TestCase):
tree = html_tag_tree(html)
assert tree.children[0].token is not None
self.assertEqual(tree.children[0].token.kind, 'html_start')
self.assertEqual(tree.children[0].token.tag, 'html')
assert tree.children[0].children[0].token is not None
self.assertEqual(tree.children[0].children[0].token.kind, 'html_start')
self.assertEqual(tree.children[0].children[0].token.tag, 'head')
assert tree.children[0].children[0].children[0].token is not None
self.assertEqual(tree.children[0].children[0].children[0].token.kind, 'html_start')
self.assertEqual(tree.children[0].children[0].children[0].token.tag, 'title')
assert tree.children[0].children[1].token is not None
self.assertEqual(tree.children[0].children[1].token.kind, 'html_start')
self.assertEqual(tree.children[0].children[1].token.tag, 'body')
assert tree.children[0].children[1].children[0].token is not None
self.assertEqual(tree.children[0].children[1].children[0].token.kind, 'html_start')
self.assertEqual(tree.children[0].children[1].children[0].token.tag, 'p')
assert tree.children[0].children[1].children[0].children[0].token is not None
self.assertEqual(tree.children[0].children[1].children[0].children[0].token.kind, 'html_singleton')
self.assertEqual(tree.children[0].children[1].children[0].children[0].token.tag, 'br')
assert tree.children[0].children[1].children[1].token is not None
self.assertEqual(tree.children[0].children[1].children[1].token.kind, 'html_start')
self.assertEqual(tree.children[0].children[1].children[1].token.tag, 'p')