<filename>tests/test_module_api.py
# -*- coding: utf-8 -*-

import shelve
import unittest
from unittest.mock import patch

from hypothesis import given
from hypothesis.strategies import text, sampled_from, integers, none, assume, dictionaries
from requests.exceptions import HTTPError

from pokebase import api
from pokebase.common import ENDPOINTS, cache_uri_build
from pokebase import cache
from pokebase.cache import set_cache, save


class TestFunction__call_api(unittest.TestCase):

    # _call_api(endpoint, resource_id)

    def setUp(self):
        set_cache('testing')

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=(integers(min_value=1)))
    @patch('pokebase.api.requests.get')
    def testArgs(self, mock_get, endpoint, resource_id):

        mock_get.return_value.json.return_value = {'id': resource_id}

        self.assertEqual(api._call_api(endpoint, resource_id)['id'],
                         resource_id)

    @given(endpoint=text(),
           resource_id=(integers(min_value=1)))
    def testArg_endpoint_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api._call_api(endpoint, resource_id)

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=none())
    @patch('pokebase.api.requests.get')
    def testArg_resource_id_None(self, mock_get, endpoint, resource_id):

        mock_get.return_value.json.return_value = {'count': 100, 'results': ['some', 'reults']}

        self.assertIsNotNone(api._call_api(endpoint, resource_id).get('count'))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1),
           subresource=text())
    @patch('pokebase.api.requests.get')
    def testArg_subresource_Text(self, mock_get, endpoint, resource_id, subresource):
        mock_get.return_value.json.return_value = {'version_details': 'foo'}

        self.assertIsNotNone(api._call_api(endpoint, resource_id, subresource).get('version_details'))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=(integers(min_value=1)))
    @patch('pokebase.api.requests.get')
    def testEnv_ErrorResponse(self, mock_get, endpoint, resource_id):
        mock_get.return_value.raise_for_status.side_effect = HTTPError()

        with self.assertRaises(HTTPError):
            api._call_api(endpoint, resource_id)


class TestFunction_get_data(unittest.TestCase):

    # get_data(endpoint, resource_id)

    def setUp(self):
        set_cache('testing')

    @given(data=dictionaries(text(), text()),
           endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1))
    def testArgs_GettingCachedData(self, data, endpoint, resource_id):
        assume(data != dict())
        # save some data to the cache
        save(data, endpoint, resource_id)
        self.assertEqual(data, api.get_data(endpoint, resource_id))

    @given(data=dictionaries(text(), text()),
           endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1))
    @patch('pokebase.api.requests.get')
    def testArgs_GettingNoncachedData(self, mock_get, data, endpoint, resource_id):

        mock_get.return_value.json.return_value = data

        # assert that the data is not in the cache
        with shelve.open(cache.API_CACHE) as cache_file:
            key = cache_uri_build(endpoint, resource_id)
            if key in cache_file.keys():
                del cache_file[key]

        self.assertEqual(data, api.get_data(endpoint, resource_id))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1),
           subresource=text())
    @patch('pokebase.api.requests.get')
    def testArg_subresource_Text(self, mock_get, endpoint, resource_id, subresource):
        mock_get.return_value.json.return_value = {'version_details': 'foo'}

        self.assertIsNotNone(api.get_data(endpoint, resource_id, subresource).get('version_details'))

    @given(endpoint=text(),
           resource_id=integers(min_value=1))
    def testArg_endpoint_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api.get_data(endpoint, resource_id)

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=text())
    def testArg_resource_id_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api.get_data(endpoint, resource_id)

<filename>tests/test_module_api.py
# -*- coding: utf-8 -*-

import shelve
import unittest
from unittest.mock import patch

from hypothesis import given
from hypothesis.strategies import text, sampled_from, integers, none, assume, dictionaries
from requests.exceptions import HTTPError

from pokebase import api
from pokebase.common import ENDPOINTS, cache_uri_build
from pokebase import cache
from pokebase.cache import set_cache, save


class TestFunction__call_api(unittest.TestCase):

    # _call_api(endpoint, resource_id)

    def setUp(self):
        set_cache('testing')

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=(integers(min_value=1)))
    @patch('pokebase.api.requests.get')
    def testArgs(self, mock_get, endpoint, resource_id):

        mock_get.return_value.json.return_value = {'id': resource_id}

        self.assertEqual(api._call_api(endpoint, resource_id)['id'],
                         resource_id)

    @given(endpoint=text(),
           resource_id=(integers(min_value=1)))
    def testArg_endpoint_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api._call_api(endpoint, resource_id)

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=none())
    @patch('pokebase.api.requests.get')
    def testArg_resource_id_None(self, mock_get, endpoint, resource_id):

        mock_get.return_value.json.return_value = {'count': 100, 'results': ['some', 'reults']}

        self.assertIsNotNone(api._call_api(endpoint, resource_id).get('count'))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1),
           subresource=text())
    @patch('pokebase.api.requests.get')
    def testArg_subresource_Text(self, mock_get, endpoint, resource_id, subresource):
        mock_get.return_value.json.return_value = {'version_details': 'foo'}

        self.assertIsNotNone(api._call_api(endpoint, resource_id, subresource).get('version_details'))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=(integers(min_value=1)))
    @patch('pokebase.api.requests.get')
    def testEnv_ErrorResponse(self, mock_get, endpoint, resource_id):
        mock_get.return_value.raise_for_status.side_effect = HTTPError()

        with self.assertRaises(HTTPError):
            api._call_api(endpoint, resource_id)


class TestFunction_get_data(unittest.TestCase):

    # get_data(endpoint, resource_id)

    def setUp(self):
        set_cache('testing')

    @given(data=dictionaries(text(), text()),
           endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1))
    def testArgs_GettingCachedData(self, data, endpoint, resource_id):
        assume(data != dict())
        # save some data to the cache
        save(data, endpoint, resource_id)
        self.assertEqual(data, api.get_data(endpoint, resource_id))

    @given(data=dictionaries(text(), text()),
           endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1))
    @patch('pokebase.api.requests.get')
    def testArgs_GettingNoncachedData(self, mock_get, data, endpoint, resource_id):

        mock_get.return_value.json.return_value = data

        # assert that the data is not in the cache
        with shelve.open(cache.API_CACHE) as cache_file:
            key = cache_uri_build(endpoint, resource_id)
            if key in cache_file.keys():
                del cache_file[key]

        self.assertEqual(data, api.get_data(endpoint, resource_id))

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=integers(min_value=1),
           subresource=text())
    @patch('pokebase.api.requests.get')
    def testArg_subresource_Text(self, mock_get, endpoint, resource_id, subresource):
        mock_get.return_value.json.return_value = {'version_details': 'foo'}

        self.assertIsNotNone(api.get_data(endpoint, resource_id, subresource).get('version_details'))

    @given(endpoint=text(),
           resource_id=integers(min_value=1))
    def testArg_endpoint_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api.get_data(endpoint, resource_id)

    @given(endpoint=sampled_from(ENDPOINTS),
           resource_id=text())
    def testArg_resource_id_Text(self, endpoint, resource_id):
        with self.assertRaises(ValueError):
            api.get_data(endpoint, resource_id)

# -*- coding: utf-8 -*- # _call_api(endpoint, resource_id) # get_data(endpoint, resource_id) # save some data to the cache # assert that the data is not in the cache