mypy: Improve typing of dialogflow and salesforce tests.

This commit is contained in:
neiljp (Neil Pilgrim) 2018-06-09 20:25:28 -07:00 committed by showell
parent 509991f26f
commit cd76a3cf5a
2 changed files with 15 additions and 15 deletions

View file

@ -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']