add got receive tests

This commit is contained in:
yanyongyu
2021-12-21 00:39:12 +08:00
parent d549cdf26a
commit e9b8515cf1
7 changed files with 111 additions and 52 deletions

View File

@ -318,16 +318,16 @@ class MatcherParam(Param):
return matcher
def Received(id: str, default: Any = None) -> Any:
def Received(id: Optional[str] = None, default: Any = None) -> Any:
def _received(matcher: "Matcher"):
return matcher.get_receive(id, default)
return matcher.get_receive(id or "", default)
return Depends(_received, use_cache=False)
def LastReceived(default: Any = None) -> Any:
def _last_received(matcher: "Matcher") -> Any:
return matcher.get_receive(None, default)
return matcher.get_last_receive(default)
return Depends(_last_received, use_cache=False)