mypy: Improve typing of dialogflow and salesforce tests.
This commit is contained in:
parent
509991f26f
commit
cd76a3cf5a
2 changed files with 15 additions and 15 deletions
|
@ -4,19 +4,10 @@ from contextlib import contextmanager
|
|||
|
||||
from unittest.mock import patch
|
||||
|
||||
from typing import Any, ByteString
|
||||
from typing import Iterator, ByteString
|
||||
|
||||
import json
|
||||
|
||||
class MockTextRequest():
|
||||
def __init__(self) -> None:
|
||||
self.session_id = ""
|
||||
self.query = ""
|
||||
self.response = ""
|
||||
|
||||
def getresponse(self) -> Any:
|
||||
return MockHttplibRequest(self.response)
|
||||
|
||||
class MockHttplibRequest():
|
||||
def __init__(self, response: str) -> None:
|
||||
self.response = response
|
||||
|
@ -24,8 +15,17 @@ class MockHttplibRequest():
|
|||
def read(self) -> ByteString:
|
||||
return json.dumps(self.response).encode()
|
||||
|
||||
class MockTextRequest():
|
||||
def __init__(self) -> None:
|
||||
self.session_id = ""
|
||||
self.query = ""
|
||||
self.response = ""
|
||||
|
||||
def getresponse(self) -> MockHttplibRequest:
|
||||
return MockHttplibRequest(self.response)
|
||||
|
||||
@contextmanager
|
||||
def mock_dialogflow(test_name: str, bot_name: str) -> Any:
|
||||
def mock_dialogflow(test_name: str, bot_name: str) -> Iterator[None]:
|
||||
response_data = read_bot_fixture_data(bot_name, test_name)
|
||||
try:
|
||||
df_request = response_data['request']
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue