def test_set_data(self, mock_return):
""" Verify get_results returns a DataFrame."""
mock_return.request("GET",
GAMEBENCH_CONFIG['url'] +
"/v1/sessions/session_id",
json=self.session_app_json["response"])
expected = pandas.DataFrame(
data=[self.session_app_json['response']['app']])
self.mediator = SessionDetailMediator(**DEFAULT_SESSION_DETAIL_PARAMS)
actual = self.mediator.get_results()
assert_frame_equal(expected, actual)
def test_set_data(self, mock_retriever, mock_utility):
""" Verify the proper methods are called."""
mock_retriever.get_response_json.return_value = SAMPLE_RESPONSE
mock_utility.return_value = {
'metric': NO_METRIC_REQUEST_PARAMS['metric'],
'response_json': SAMPLE_RESPONSE
}
mediator = SessionDetailMediator(**NO_METRIC_REQUEST_PARAMS)
mediator.get_results()
with self.subTest():
mock_retriever.assert_called_once_with(
**NO_METRIC_REQUEST_PARAMS['response'])
mock_utility.assert_called_once()
class TestSessionDetailMediator(TestCase):
""" Tests for the SessionDetailMediator."""
def setUp(self):
self.error_message = "\nTest: {} \nExpected: {}\nActual: {}\n"
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "sessionid.json")) as json_data:
self.test_json = json.load(json_data)
@requests_mock.Mocker()
def test_set_data(self, mock_return):
""" Verify get_results returns a DataFrame."""
adapter = RequestsAdapter(**NO_METRIC_EXPECTED_REQUEST_PARAMS)
mock_return.request(
"POST",
"https://api.production.gamebench.net/v1/sessions/session_id?test_params",
json=self.test_json["response"]
)
series = pandas.Series(adapter.request().json())
expected = pandas.DataFrame(
series['app'],
index=['app']
)
self.mediator = SessionDetailMediator(**NO_METRIC_REQUEST_PARAMS)
actual = self.mediator.get_results()
assert_frame_equal(
expected,
actual
)
def test_init_creates_retriever(self, mock_retriever):
""" Verify the init calls the ResponseRetriever and assigns it to a variable."""
mediator = SessionDetailMediator(**NO_METRIC_REQUEST_PARAMS)
with self.subTest():
mock_retriever.assert_called_once_with(
**NO_METRIC_REQUEST_PARAMS['response'])
self.assertIsNotNone(mediator.retriever)
def test_init_creates_retriever(self, mock_retriever):
""" Verify the init calls the ResponseRetriever and assigns it to a variable."""
mediator = SessionDetailMediator(**DEFAULT_SESSION_DETAIL_PARAMS)
with self.subTest():
mock_retriever.assert_called_once_with(
**DEFAULT_SESSION_DETAIL_PARAMS)
with self.subTest():
self.assertIsNotNone(mediator.retriever)
def __init__(self, **request_parameters):
""" Sets up the mediator and data attributes.
The mediator is set to the mediator that it will be contacting. The data
is set to the DataFrame returned from the get_results method.
:param request_parameters: Dictionary passed in from the model Creator. Contains information to
build and send a response.
"""
super().__init__()
self.mediator = SessionDetailMediator(**request_parameters)
self.data = self.get_data()
def test_set_data(self, mock_return):
""" Verify get_results returns a DataFrame."""
adapter = RequestsAdapter(**NO_METRIC_EXPECTED_REQUEST_PARAMS)
mock_return.request(
"POST",
"https://api.production.gamebench.net/v1/sessions/session_id?test_params",
json=self.test_json["response"]
)
series = pandas.Series(adapter.request().json())
expected = pandas.DataFrame(
series['app'],
index=['app']
)
self.mediator = SessionDetailMediator(**NO_METRIC_REQUEST_PARAMS)
actual = self.mediator.get_results()
assert_frame_equal(
expected,
actual
)
class TestSessionDetailMediator(TestCase):
""" Tests for the SessionDetailMediator."""
def setUp(self):
self.error_message = "\nTest: {} \nExpected: {}\nActual: {}\n"
with open(os.path.join(PARENT_DIR + API_SAMPLES +
"sessionid.json")) as json_data:
self.session_app_json = json.load(json_data)
@requests_mock.Mocker()
def test_set_data(self, mock_return):
""" Verify get_results returns a DataFrame."""
mock_return.request("GET",
GAMEBENCH_CONFIG['url'] +
"/v1/sessions/session_id",
json=self.session_app_json["response"])
expected = pandas.DataFrame(
data=[self.session_app_json['response']['app']])
self.mediator = SessionDetailMediator(**DEFAULT_SESSION_DETAIL_PARAMS)
actual = self.mediator.get_results()
assert_frame_equal(expected, actual)