class TestRequestHandler(unittest.TestCase):
VALID_LEAGUE_CODE = "BL"
VALID_TEAM_CODE = "AFC"
VALID_TIME = 10
def setUp(self):
dummy_key = 12345678901234567890123456789012
headers = {'X-Auth-Token': dummy_key}
TEAM_DATA = get_team_data()["teams"]
TEAM_NAMES = {team["code"]: team["id"] for team in TEAM_DATA}
LEAGUE_IDS = leagueids.LEAGUE_IDS
self.dummy_url = "http://some_url"
writer = get_writer()
self.rq = RequestHandler(headers, LEAGUE_IDS, TEAM_NAMES, writer)
def tearDown(self):
pass
@mock.patch('requests.get')
def test_ok_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.ok,
response=json.dumps({'key': 'value'}))
try:
self.rq._get(self.dummy_url)
except APIErrorException:
self.fail("Threw exception erroneously")
@mock.patch('requests.get')
def test_bad_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.bad,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("Invalid request. "
"Check parameters." in context.exception)
@mock.patch('requests.get')
def test_forbidden_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.forbidden,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue('This resource is restricted' in context.exception)
@mock.patch('requests.get')
def test_not_found_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.not_found,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("This resource does not exist. "
"Check parameters" in context.exception)
@mock.patch('requests.get')
def test_too_many_requests_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.too_many_requests,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("You have exceeded your allowed "
"requests per minute/day" in context.exception)
@mock.patch('soccer.writers.Stdout.live_scores')
@mock.patch('requests.get')
def test_get_live_scores_ok(self, mock_request_call, mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'games': [1, 2]}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_live_scores(True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.live_scores')
@mock.patch('requests.get')
def test_get_live_scores_0_games(self,
mock_request_call, mock_writer,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'games': []}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_live_scores(True)
mock_click.assert_called_with("No live action "
"currently", fg="red", bold=True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.live_scores')
@mock.patch('requests.get')
def test_get_live_scores_error(self,
mock_request_call, mock_writer, mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'games': [1, 2]}, 400)]
mock_writer.return_value = mock.Mock()
self.rq.get_live_scores(True)
mock_click.assert_called_with("There was problem getting "
"live scores", fg="red", bold=True)
mock_writer.assert_called_once()
@mock.patch('soccer.writers.Stdout.team_scores')
@mock.patch('requests.get')
def test_get_team_scores_ok(self,
mock_request_call, mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': [1, 2]}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_team_scores(TestRequestHandler.VALID_TEAM_CODE,
6, True, True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.team_scores')
@mock.patch('requests.get')
def test_get_team_scores_0_fixtures(self,
mock_request_call,
mock_writer, mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': []}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_team_scores(TestRequestHandler.VALID_TEAM_CODE,
6, True, True)
mock_click.assert_called_with("No action during"
" past week. Change the time "
"parameter to get "
"more fixtures.", fg="red", bold=True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.team_scores')
@mock.patch('requests.get')
def test_get_team_scores_bad_id(self,
mock_request_call, mock_writer, mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': [1, 2]}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_team_scores("AdkljdfkljkdlFC", 6, True, True)
mock_click.assert_called_with("Team code is not "
"correct.", fg="red", bold=True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_standings_error(self,
mock_request_call, mock_click):
mock_request_call.side_effect = \
APIErrorException()
self.rq.get_standings(TestRequestHandler.VALID_LEAGUE_CODE)
mock_click.assert_called_with("No standings availble for "
"{league}.".format(
league=TestRequestHandler.VALID_LEAGUE_CODE),
fg="red", bold=True)
@mock.patch('soccer.writers.Stdout.standings')
@mock.patch('requests.get')
def test_get_standings_ok(self,
mock_request_call,
mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'standing': []}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_standings(TestRequestHandler.VALID_LEAGUE_CODE)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_league_scores_error(self,
mock_request_call,
mock_click):
mock_request_call.side_effect = \
APIErrorException()
self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
TestRequestHandler.VALID_TIME, False, False)
mock_click.assert_called_with("No data "
"for the given league.", fg="red", bold=True)
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_league_scores_no_fixtures(self,
mock_request_call,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': []}, 200)]
self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
TestRequestHandler.VALID_TIME, False, False)
mock_click.assert_called_with("No {league} matches in "
"the past week.".format(
league=TestRequestHandler.VALID_LEAGUE_CODE),
fg="red", bold=True)
@mock.patch('soccer.writers.Stdout.league_scores')
@mock.patch('requests.get')
def test_get_league_scores_multiple_fixtures(self,
mock_request_call,
mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': [1]}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
TestRequestHandler.VALID_TIME, False, False)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_team_players_error(self,
mock_request_call,
mock_click):
mock_request_call.side_effect = APIErrorException()
self.rq.get_team_players(TestRequestHandler.VALID_TEAM_CODE)
mock_click.assert_called_with("No data for the team. "
"Please check the team code.", bold=True, fg="red")
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_team_players_no_players(self,
mock_request_call,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'count': "0"}, 200)]
self.rq.get_team_players(TestRequestHandler.VALID_TEAM_CODE)
mock_click.assert_called_with("No players found "
"for this team", fg="red", bold=True)
@mock.patch('soccer.writers.Stdout.team_players')
@mock.patch('requests.get')
def test_get_team_players_no_players(self,
mock_request_call,
mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'count': "1"}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_team_players(TestRequestHandler.VALID_TEAM_CODE)
mock_writer.assert_called_once()
def main(league, time, standings, team, live, use12hour, players,
output_format, output_file, upcoming, lookup, listcodes, apikey):
"""
A CLI for live and past football scores from various football leagues.
League codes:
\b
- CL: Champions League
- PL: England Premier League
- EL1: England League One
- ELC: England Championship
- FL1: French Ligue 1
- FL2: French Ligue 2
- BL: German Bundesliga
- BL2: 2. Bundesliga
- SA: Serie A
- DED: Eredivisie
- PPL: Primeira Liga
- PD: Primera Division
- SD: Segunda Division
"""
headers = {'X-Auth-Token': apikey}
try:
if output_format == 'stdout' and output_file:
raise IncorrectParametersException(
'Printing output to stdout and '
'saving to a file are mutually exclusive')
writer = get_writer(output_format, output_file)
rh = RequestHandler(headers, LEAGUE_IDS, TEAM_NAMES, writer)
if listcodes:
list_team_codes()
return
if live:
rh.get_live_scores(use12hour)
return
if standings:
if not league:
raise IncorrectParametersException(
'Please specify a league. '
'Example --standings --league=EPL')
rh.get_standings(league)
return
if team:
if lookup:
map_team_id(team)
return
if players:
rh.get_team_players(team)
return
else:
rh.get_team_scores(team, time, upcoming, use12hour)
return
rh.get_league_scores(league, time, upcoming, use12hour)
except IncorrectParametersException as e:
click.secho(e.message, fg="red", bold=True)
class TestRequestHandler(unittest.TestCase):
VALID_LEAGUE_CODE = "BL"
VALID_TEAM_CODE = "AFC"
VALID_TIME = 10
def setUp(self):
dummy_key = 12345678901234567890123456789012
headers = {'X-Auth-Token': dummy_key}
TEAM_DATA = get_team_data()["teams"]
TEAM_NAMES = {team["code"]: team["id"] for team in TEAM_DATA}
LEAGUE_IDS = leagueids.LEAGUE_IDS
self.dummy_url = "http://some_url"
writer = get_writer()
self.rq = RequestHandler(headers, LEAGUE_IDS, TEAM_NAMES, writer)
def tearDown(self):
pass
@mock.patch('requests.get')
def test_ok_code(self, mock_call):
mock_call.return_value = mock.MagicMock(status_code=requests.codes.ok,
response=json.dumps(
{'key': 'value'}))
try:
self.rq._get(self.dummy_url)
except APIErrorException:
self.fail("Threw exception erroneously")
@mock.patch('requests.get')
def test_bad_code(self, mock_call):
mock_call.return_value = mock.MagicMock(status_code=requests.codes.bad,
response=json.dumps(
{'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("Invalid request. "
"Check parameters." in context.exception)
@mock.patch('requests.get')
def test_forbidden_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.forbidden,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue('This resource is restricted' in context.exception)
@mock.patch('requests.get')
def test_not_found_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.not_found,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("This resource does not exist. "
"Check parameters" in context.exception)
@mock.patch('requests.get')
def test_too_many_requests_code(self, mock_call):
mock_call.return_value = mock.MagicMock(
status_code=requests.codes.too_many_requests,
response=json.dumps({'key': 'value'}))
with self.assertRaises(APIErrorException) as context:
self.rq._get(self.dummy_url)
self.assertTrue("You have exceeded your allowed "
"requests per minute/day" in context.exception)
# @mock.patch('soccer.writers.Stdout.live_scores')
# @mock.patch('requests.get')
# def test_get_live_scores_ok(self, mock_request_call, mock_writer):
# mock_request_call.side_effect = \
# [mocked_requests_get({'games': [1, 2]}, 200)]
# mock_writer.return_value = mock.Mock()
# self.rq.get_live_scores(True)
# mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.live_scores')
@mock.patch('requests.get')
def test_get_live_scores_0_games(self, mock_request_call, mock_writer,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'games': []}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_live_scores(True)
mock_click.assert_called_with("No live action "
"currently",
fg="red",
bold=True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.live_scores')
@mock.patch('requests.get')
def test_get_live_scores_error(self, mock_request_call, mock_writer,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'games': [1, 2]}, 400)]
mock_writer.return_value = mock.Mock()
self.rq.get_live_scores(True)
mock_click.assert_called_with(
"There was problem getting "
"live scores", fg="red", bold=True)
mock_writer.assert_called_once()
# @mock.patch('soccer.writers.Stdout.team_scores')
# @mock.patch('requests.get')
# def test_get_team_scores_ok(self,
# mock_request_call, mock_writer):
# mock_request_call.side_effect = \
# [mocked_requests_get({'fixtures': [1, 2]}, 200)]
# mock_writer.return_value = mock.Mock()
# self.rq.get_team_scores(TestRequestHandler.VALID_TEAM_CODE,
# 6, True, True)
# mock_writer.assert_called_once()
# @mock.patch('click.secho')
# @mock.patch('soccer.writers.Stdout.team_scores')
# @mock.patch('requests.get')
# def test_get_team_scores_0_fixtures(self,
# mock_request_call,
# mock_writer, mock_click):
# mock_request_call.side_effect = \
# [mocked_requests_get({'fixtures': []}, 200)]
# mock_writer.return_value = mock.Mock()
# self.rq.get_team_scores(TestRequestHandler.VALID_TEAM_CODE,
# 6, True, True)
# mock_click.assert_called_with("No action during"
# " past week. Change the time "
# "parameter to get "
# "more fixtures.", fg="red", bold=True)
# mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('soccer.writers.Stdout.team_scores')
@mock.patch('requests.get')
def test_get_team_scores_bad_id(self, mock_request_call, mock_writer,
mock_click):
mock_request_call.side_effect = \
[mocked_requests_get({'fixtures': [1, 2]}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_team_scores("AdkljdfkljkdlFC", 6, True, True)
mock_click.assert_called_with("Team code is not "
"correct.",
fg="red",
bold=True)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_standings_error(self, mock_request_call, mock_click):
mock_request_call.side_effect = \
APIErrorException()
self.rq.get_standings(TestRequestHandler.VALID_LEAGUE_CODE)
mock_click.assert_called_with(
"No standings availble for "
"{league}.".format(league=TestRequestHandler.VALID_LEAGUE_CODE),
fg="red",
bold=True)
@mock.patch('soccer.writers.Stdout.standings')
@mock.patch('requests.get')
def test_get_standings_ok(self, mock_request_call, mock_writer):
mock_request_call.side_effect = \
[mocked_requests_get({'standing': []}, 200)]
mock_writer.return_value = mock.Mock()
self.rq.get_standings(TestRequestHandler.VALID_LEAGUE_CODE)
mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_league_scores_error(self, mock_request_call, mock_click):
mock_request_call.side_effect = \
APIErrorException()
self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
TestRequestHandler.VALID_TIME, False, False)
mock_click.assert_called_with("No data "
"for the given league.",
fg="red",
bold=True)
# @mock.patch('click.secho')
# @mock.patch('requests.get')
# def test_get_league_scores_no_fixtures(self,
# mock_request_call,
# mock_click):
# mock_request_call.side_effect = \
# [mocked_requests_get({'fixtures': []}, 200)]
# self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
# TestRequestHandler.VALID_TIME, False, False)
# mock_click.assert_called_with("No {league} matches in "
# "the past week.".format(
# league=TestRequestHandler.VALID_LEAGUE_CODE),
# fg="red", bold=True)
# @mock.patch('soccer.writers.Stdout.league_scores')
# @mock.patch('requests.get')
# def test_get_league_scores_multiple_fixtures(self,
# mock_request_call,
# mock_writer):
# mock_request_call.side_effect = \
# [mocked_requests_get({'fixtures': [1]}, 200)]
# mock_writer.return_value = mock.Mock()
# self.rq.get_league_scores(TestRequestHandler.VALID_LEAGUE_CODE,
# TestRequestHandler.VALID_TIME, False, False)
# mock_writer.assert_called_once()
@mock.patch('click.secho')
@mock.patch('requests.get')
def test_get_team_players_error(self, mock_request_call, mock_click):
mock_request_call.side_effect = APIErrorException()
self.rq.get_team_players(TestRequestHandler.VALID_TEAM_CODE)
mock_click.assert_called_with(
"No data for the team. "
"Please check the team code.",
bold=True,
fg="red")
