test_api_client.py

# -*- coding: utf-8 -*-
import mock
import pytest

from tests.base import TestBase
from tvdbrest import VERSION
from tvdbrest.client import TVDB, Unauthorized, APIError, NotFound


@pytest.fixture
def tvdb():
    return TVDB("myusername", "myuserkey", "myapikey")


@pytest.fixture
def empty_positive_response():
    m = mock.Mock()
    m.status_code = 200
    m.json = mock.Mock(return_value={"data": []})
    return m


class TestLoginLogout(object):
    
    def test_login_status(self):
        tvdb = TVDB("myusername", "myuserkey", "myapikey")
        assert not tvdb.logged_in

    @mock.patch('tvdbrest.client.requests.request')
    def test_login_status_after_login(self, request_mock):
        response_mock = mock.MagicMock()
        response_mock.status_code = 200
        response_mock.json = mock.MagicMock(return_value={
            'token': 'jwttoken'
        })
        
        request_mock.return_value = response_mock
        
        tvdb = TVDB("myusername", "myuserkey", "myapikey")
        tvdb.login()
        
        request_mock.assert_called_with('post', 'https://api.thetvdb.com/login', headers={
            'Accept-Language': 'en',
            'User-Agent': 'tvdb-rest %s' % VERSION
        }, json={
            'username': 'myusername',
            'userkey': 'myuserkey',
            'apikey': 'myapikey'
        })

        assert tvdb.logged_in

    @mock.patch('tvdbrest.client.requests.request')
    def test_failed_login(self, request_method_mock):
        response_mock = mock.MagicMock()
        response_mock.status_code = 401
        request_method_mock.return_value = response_mock
        
        tvdb = TVDB("myusername", "myuserkey", "myapikey")
        with pytest.raises(Unauthorized):
            tvdb.login()
        
        assert not tvdb.logged_in

    def test_logout(self, tvdb):
        tvdb.jwttoken = "abc"
        assert tvdb.logged_in
        tvdb.logout()
        assert tvdb.jwttoken is None
        assert not tvdb.logged_in

    @mock.patch('tvdbrest.client.requests.request')
    def test_decorator_login_before_api_call(self, request_method_mock, tvdb):
        response_mock = mock.MagicMock()
        response_mock.status_code = 200
        response_mock.json = mock.MagicMock(return_value={"data": []})

        request_method_mock.return_value = response_mock

        login_mock = mock.MagicMock()
        tvdb.login = login_mock
        tvdb.languages()
        
        assert login_mock.called

    @mock.patch('tvdbrest.client.requests.request')
    def test_authorization_for_api_call(self, request_mock, tvdb, empty_positive_response):
        request_mock.return_value = empty_positive_response
        tvdb.jwttoken = "test"
        tvdb.languages()
    
        request_mock.assert_called_with('get', 'https://api.thetvdb.com/languages', headers={
            'Authorization': 'Bearer test',
            'Accept-Language': 'en',
            'User-Agent': 'tvdb-rest %s' % VERSION
        })


class TestClientBasics(TestBase):

    @mock.patch('tvdbrest.client.requests.request')
    def test_accept_language(self, request_mock, empty_positive_response):
        tvdb = TVDB("myusername", "myuserkey", "myapikey", 'de')
        request_mock.return_value = empty_positive_response
        tvdb.jwttoken = "test"
        tvdb.languages()
    
        request_mock.assert_called_with('get', 'https://api.thetvdb.com/languages', headers={
            'Authorization': 'Bearer test',
            'Accept-Language': 'de',
            'User-Agent': 'tvdb-rest %s' % VERSION
        })

    @mock.patch('tvdbrest.client.requests.request')
    def test_accept_language_not_set(self, request_mock, tvdb, empty_positive_response):
        tvdb.accept_language = None
        request_mock.return_value = empty_positive_response
        tvdb.jwttoken = "test"
        tvdb.languages()
    
        request_mock.assert_called_with('get', 'https://api.thetvdb.com/languages', headers={
            'Authorization': 'Bearer test',
            'User-Agent': 'tvdb-rest %s' % VERSION
        })

    @mock.patch('tvdbrest.client.requests.request')
    def test_raise_apierror_on_4xx(self, request_mock, tvdb):
        m = mock.Mock()
        m.status_code = 405
        request_mock.return_value = m
        
        with pytest.raises(APIError):
            tvdb.login()

    @mock.patch('tvdbrest.client.requests.request')
    def test_raise_apierror_on_5xx(self, request_mock, tvdb):
        m = mock.Mock()
        m.status_code = 500
        request_mock.return_value = m
    
        with pytest.raises(APIError):
            tvdb.login()

    @mock.patch('tvdbrest.client.requests.request')
    def test_raise_not_found_on_404(self, request_mock, tvdb):
        # m = mock.Mock()
        # m.status_code = 404
        request_mock.return_value = self.api_response_404_mock()
        
        with pytest.raises(NotFound):
            tvdb.login()