def test_fps_stability(self, mock_return):
""" FPS Stability gets the correct data back
The FPS Stability needs to go through the 'to_dataframe' utility method.
Regression test for https://github.com/bigfishgames/gamebenchapi-pyclient/issues/67
"""
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "fps_stability.json")) as \
json_data:
self.fps_stability_json = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request('GET',
FPS_STABILITY_URL,
json=self.fps_stability_json['response'])
creator = ModelCreator('FpsStability', **FPS_STABILITY_PARAMS)
model = creator.get_model()
expected = pandas.DataFrame([self.fps_stability_json['response']])
actual = model.data
assert_frame_equal(actual, expected)
def test_keyword_search(self, mock_return):
""" Keyword search should return the proper DataFrames.
Regression test for https://github.com/bigfishgames/gamebenchapi-pyclient/issues/56
"""
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "keyword_search.json")) as \
json_data:
self.keyword_search_json = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request(
'POST',
'https://production.node.gce.gamebench.net/v1/sessions/typeahead',
json=self.keyword_search_json['response'])
creator = ModelCreator('Keyword', **KEYWORD_SEARCH_PARAMS)
model = creator.get_model()
expected = pandas.DataFrame([self.keyword_search_json['response']])
actual = model.data
assert_frame_equal(actual, expected)
def test_filter_using_apps(self, mock_return):
""" Keyword search using 'apps' should return the proper DataFrames.
Regression test for https://github.com/bigfishgames/gamebenchapi-pyclient/issues/59
"""
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "apps_filter.json")) as \
json_data:
self.app_filter_json = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request(
'POST',
'https://production.node.gce.gamebench.net/v1/sessions',
json=self.app_filter_json['response'])
creator = ModelCreator('SessionSummary', **APPS_FILTER_PARAMS)
model = creator.get_model()
expected = pandas.DataFrame([self.app_filter_json['response']])
actual = model.data
assert_frame_equal(actual, expected)
def test_error_raise_when_module_not_found(self, mock_import, mock_object):
""" Verify the ModuleNotFound exception is raised if the given model doesn't exist."""
mock_class_object = Mock
mock_import.return_value = 'mocking.a.module.path'
mock_object.return_value = mock_class_object
expected = "The not_valid model does not exist."
with self.assertRaises(ModelNotFound) as module_error:
self.creator = ModelCreator("not_valid")
actual = str(module_error.exception)
self.assertEqual(actual, expected)
def test_generic_model_creation(self, mock_return):
""" The ModelCreator returns the proper DataFrame."""
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "notes.json")) as \
json_data:
self.generic_json = json.load(json_data)
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request('GET',
GENERIC_SESSION_URL,
json=self.generic_json['response'])
creator = ModelCreator('SessionNotes', **DEFAULT_GENERIC_PARAMS)
model = creator.get_model()
expected = pandas.DataFrame([self.generic_json['response']])
actual = model.data
assert_frame_equal(actual, expected)
def test_time_series_model_creation(self, mock_return):
""" The ModelCreator returns the proper DataFrame."""
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "cpu_multiple_sessions.json")) as \
json_data:
self.time_series_json = json.load(json_data)
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request('GET',
DEFAULT_SESSION_URL,
json=self.time_series_json['response'])
creator = ModelCreator('Cpu', **DEFAULT_REQUEST_PARAMS)
model = creator.get_model()
expected = json_normalize(self.time_series_json['response'], 'samples',
['id', 'sessionId'])
actual = model.data
assert_frame_equal(actual, expected)
def test_session_summary_details(self, mock_return):
""" SessionSummary attributes should return the proper DataFrames."""
with open(os.path.join(
PARENT_DIR + API_SAMPLES + "sessionid.json")) as \
json_data:
self.session_app_json = json.load(json_data)
with open(os.path.join(
PARENT_DIR + '/fixtures/' + "authentication_token.json")) as \
json_data:
self.auth_token = json.load(json_data)
mock_return.request('POST', AUTH_URL, json=self.auth_token)
mock_return.request('GET',
BASE_SESSION_URL,
json=self.session_app_json['response'])
creator = ModelCreator('SessionSummary',
**DEFAULT_SESSION_DETAIL_PARAMS)
model = creator.get_model()
df = pandas.DataFrame([self.session_app_json['response']])
expected = df.filter(['app'])
actual = model.app
assert_frame_equal(actual, expected)
class TestFindSessionDetail(TestCase):
""" Module path for Session Detail models can be found."""
@patch(MODEL_CREATOR_GET_CLASS)
@patch(MODEL_CREATOR_IMPORT_MODULE)
@patch(MODEL_CREATOR_SET_MODULE)
def setUp(self, mock_module, mock_import, mock_object):
mock_module.return_value = "Test.Path"
mock_import.return_value = "Test Module"
mock_object.return_value = stub_function
self.creator = ModelCreator("App")
def test_find_session_detail_model(self):
""" Verify a session detail model can be found and the model name returned."""
actual = self.creator._set_module_name_by_model()
expected = "gamebench_api_client.models.dataframes.session_detail.session_detail_models"
self.assertEqual(actual, expected)
def setUp(self, mock_module, mock_import, mock_object):
mock_module.return_value = "Test.Path"
mock_import.return_value = "Test Module"
mock_object.return_value = stub_function
self.creator = ModelCreator("App")