test_api.py

from .test_setup import TestSetUp
from utils.exception import ValidationException
from codeforces.api import (user_info, user_rating, contest_list,
                            contest_standings, contest_ratingChanges,
                            user_status)


class TestAPI(TestSetUp):

    def test_user_info_api(self):
        handles = ['shivamsinghal1012', 'shivam011']
        response = user_info(handles)
        self.assertEqual(len(response), 2)
        self.assertEqual(response[0]['handle'], 'shivamsinghal1012')

    def test_user_rating_api(self):
        handle = 'Fefer_Ivan'
        response = user_rating(handle)
        self.assertTrue(len(response) > 0)
        self.assertEqual(response[0]['handle'], handle)

    def test_contest_list(self):
        response = contest_list()
        self.assertLess(response[0]['id'], 100001)
        response = contest_list(gym=True)
        self.assertGreaterEqual(response[0]['id'], 100001)

    def test_contest_standings(self):
        response = contest_standings(566, count=5)
        self.assertEqual(response['contest']['id'], 566)
        self.assertEqual(len(response['problems']), 7)
        self.assertEqual(len(response['rows']), 5)
        self.assertEqual(response['rows'][0]['party']['participantType'],
                         'CONTESTANT')

    def test_contest_ratingChanges(self):
        response = contest_ratingChanges(566)
        self.assertEqual(response[0]['rank'], 1)

    def test_user_status(self):
        handle = 'shivamsinghal1012'
        wrong_handle = 'er'
        starting_from = 1
        count = 10
        response = user_status(handle=handle,
                               starting_from=starting_from,
                               count=count)
        self.assertEqual(len(response[0]), 12)
        self.assertEqual(len(response), 10)
        self.assertRaises(ValidationException, user_status, wrong_handle)