tests: Test that docs & completions don't crash

This commit is contained in:
James Westman 2022-03-19 17:57:49 -05:00
parent 6576e02837
commit 3f37380c25
No known key found for this signature in database
GPG key ID: CE2DBA0ADB654EA6
4 changed files with 21 additions and 5 deletions

View file

@ -112,10 +112,10 @@ class Flag(AstNode):
@docs()
def docs(self):
type = self.parent.parent.value_type
if not isinstance(type, Enumeration):
return
if member := type.members.get(self.tokens["value"]):
return member.doc
else:
return type.doc
@validate()
def validate_for_type(self):