feat: add optional message for assertion

This commit is contained in:
debugtalk
2020-06-23 11:11:05 +08:00
parent 286dbbfe55
commit d11aca83c0
9 changed files with 158 additions and 91 deletions

View File

@@ -310,7 +310,12 @@ def make_teststep_chain_style(teststep: Dict) -> Text:
expect = validator["expect"]
if isinstance(expect, Text):
expect = f'"{expect}"'
step_info += f".assert_{assert_method}({check}, {expect})"
message = validator["message"]
if message:
step_info += f".assert_{assert_method}({check}, {expect}, '{message}')"
else:
step_info += f".assert_{assert_method}({check}, {expect})"
return f"Step({step_info})"