def test_heatmaps(self): proto, pandas, proto_game = write_proto_pandas_to_file( get_test_file("3_DRIBBLES_2_FLICKS.replay", is_replay=True)) f = download_replay_discord("3_DRIBBLES_2_FLICKS.replay") r = requests.post(LOCAL_URL + '/api/upload', files={'replays': ('fake_file.replay', f)}) r.raise_for_status() assert (r.status_code == 202) time.sleep(30) r = requests.get(LOCAL_URL + '/api/global/replay_count') result = json.loads(r.content) assert int(result) > 0 # actual test id = proto_game.game_metadata.match_guid r = requests.get(LOCAL_URL + '/api/replay/' + id + '/heatmaps') r.raise_for_status() assert r.status_code == 200 result = json.loads(r.content) assert 'data' in result assert 'maxs' in result def assert_keys(value): assert 'ball' in value assert proto_game.players[0].name in value assert_keys(result['data']) assert_keys(result['maxs'])
def test_proto_upload_with_tags(self, test_client): proto, pandas, proto_game = write_proto_pandas_to_file(get_test_file(get_complex_replay_list()[0], is_replay=True)) with open(proto, 'rb') as f: encoded_proto = base64.b64encode(zlib.compress(f.read())).decode() with open(pandas, 'rb') as f: encoded_pandas = base64.b64encode(zlib.compress(f.read())).decode() obj = { 'status': '200', 'proto': encoded_proto, 'pandas': encoded_pandas } r = Request('POST', LOCAL_URL + '/api/upload/proto', json=obj, params={'tags': [TAG_NAME, TAG_NAME + "hello"], 'player_id': proto_game.players[0].id.id}) response = test_client.send(r) assert(response.status_code == 200) fake_session = get_current_session() game = fake_session.query(Game).first() assert(game.hash == '70DDECEA4653AC55EA77DBA0DB497995') assert(game.name == '3 kickoffs 4 shots') assert(len(game.tags) == 2) assert(game.tags[0].name == TAG_NAME) assert(game.tags[0].owner == proto_game.players[0].id.id) assert(game.tags[0].games[0] == game) assert(game.tags[1].name == (TAG_NAME + "hello")) player = fake_session.query(Player.platformid == proto_game.players[0].id.id).first() assert(player is not None)
def test_proto_upload_with_privacy(self, test_client): proto, pandas, proto_game = write_proto_pandas_to_file(get_test_file(get_complex_replay_list()[0], is_replay=True)) with open(proto, 'rb') as f: encoded_proto = base64.b64encode(zlib.compress(f.read())).decode() with open(pandas, 'rb') as f: encoded_pandas = base64.b64encode(zlib.compress(f.read())).decode() obj = { 'status': '200', 'proto': encoded_proto, 'pandas': encoded_pandas } r = Request('POST', LOCAL_URL + '/api/upload/proto', json=obj, params={'visibility': GameVisibilitySetting.PRIVATE.name, 'player_id': proto_game.players[0].id.id}) response = test_client.send(r) assert(response.status_code == 200) fake_session = get_current_session() game = fake_session.query(Game).first() assert(game.hash == '70DDECEA4653AC55EA77DBA0DB497995') assert(game.name == '3 kickoffs 4 shots') assert(game.visibility == GameVisibilitySetting.PRIVATE) game_visiblity = fake_session.query(GameVisibility).first() assert(game_visiblity.game == game.hash) assert(game_visiblity.player == proto_game.players[0].id.id) assert(game_visiblity.visibility == GameVisibilitySetting.PRIVATE) assert(game_visiblity.release_date == datetime.max) player = fake_session.query(Player.platformid == proto_game.players[0].id.id).first() assert(player is not None)
def test_upload_proto(self): proto, pandas, proto_game = write_proto_pandas_to_file(get_test_file("3_KICKOFFS_4_SHOTS.replay", is_replay=True)) with open(proto, 'rb') as f: encoded_proto = base64.b64encode(zlib.compress(f.read())).decode() with open(pandas, 'rb') as f: encoded_pandas = base64.b64encode(zlib.compress(f.read())).decode() obj = { 'status': '200', 'proto': encoded_proto, 'pandas': encoded_pandas } r = requests.post(LOCAL_URL + '/api/upload/proto', json=obj, params={'tags': ['TAG'], 'visibility': GameVisibilitySetting.PRIVATE.name, 'player_id': proto_game.players[0].id.id}) r.raise_for_status() assert r.status_code == 200 r = requests.get(LOCAL_URL + '/api/global/replay_count') result = json.loads(r.content) assert int(result) == 1 response = requests.get(LOCAL_URL + '/api/tag') result = json.loads(response.content) assert result[0]['owner_id'] == "76561198018756583" assert result[0]['name'].startswith('TAG') assert len(result) == 1 response = requests.get(LOCAL_URL + '/api/player/76561198018756583/match_history?page=0&limit=10') assert response.status_code == 200 assert len(response.json()['replays']) >= 1
def test_heatmaps(self, use_test_paths, temp_file): use_test_paths.patch() test_path = use_test_paths.get_temp_path() proto, pandas, proto_game = write_proto_pandas_to_file( get_test_file("ALL_STAR.replay", is_replay=True)) with open(proto, 'rb') as f: encoded_proto = base64.b64encode(zlib.compress(f.read())).decode() obj = {'status': '200', 'proto': encoded_proto} r = requests.post(LOCAL_URL + '/api/upload/proto', json=obj) r.raise_for_status() assert (r.status_code == 200) save_replay(proto_game, temp_file, pandas[:-len(PANDAS_EXTENSION)], proto[:-len(PROTO_EXTENSION)]) r = requests.get(LOCAL_URL + '/api/global/replay_count') result = json.loads(r.content) assert int( result ) > 0, 'This test can not run without a replay in the database' # test default self.assert_heatmap(proto_game, has_ball=True) # test query params self.assert_heatmap( proto_game, query_params={"type": HeatMapType.POSITIONING.value}, has_ball=True) self.assert_heatmap(proto_game, query_params={"type": HeatMapType.BOOST.value}) self.assert_heatmap( proto_game, query_params={"type": HeatMapType.BOOST_COLLECT.value}) self.assert_heatmap( proto_game, query_params={"type": HeatMapType.BOOST_SPEED.value}) self.assert_heatmap( proto_game, query_params={"type": HeatMapType.SLOW_SPEED.value})
def test_heatmaps(self): proto, pandas, proto_game = write_proto_pandas_to_file(get_test_file("ALL_STAR.replay", is_replay=True)) f = download_replay_discord("ALL_STAR.replay") r = requests.post(LOCAL_URL + '/api/upload', files={'replays': ('fake_file.replay', f)}) r.raise_for_status() assert(r.status_code == 202) time.sleep(35) r = requests.get(LOCAL_URL + '/api/global/replay_count') result = json.loads(r.content) assert int(result) > 0, 'This test can not run without a replay in the database' # test default self.assert_heatmap(proto_game, has_ball=True) # test query params self.assert_heatmap(proto_game, query_params={"type": HeatMapType.POSITIONING.value}, has_ball=True) self.assert_heatmap(proto_game, query_params={"type": HeatMapType.BOOST.value}) self.assert_heatmap(proto_game, query_params={"type": HeatMapType.BOOST_COLLECT.value}) self.assert_heatmap(proto_game, query_params={"type": HeatMapType.BOOST_SPEED.value}) self.assert_heatmap(proto_game, query_params={"type": HeatMapType.SLOW_SPEED.value})