diff options
Diffstat (limited to 'tests/integration/test_mongodb.py')
| -rw-r--r-- | tests/integration/test_mongodb.py | 37 |
1 files changed, 19 insertions, 18 deletions
diff --git a/tests/integration/test_mongodb.py b/tests/integration/test_mongodb.py index 05e61a2..a186ee7 100644 --- a/tests/integration/test_mongodb.py +++ b/tests/integration/test_mongodb.py @@ -1,12 +1,11 @@ import pytest -import unittest from unittest.mock import patch from pymongo import MongoClient -from requests_cache.backends import MongoDict, MongoPickleDict, get_valid_kwargs +from requests_cache.backends import MongoCache, MongoDict, MongoPickleDict, get_valid_kwargs from tests.conftest import fail_if_no_connection -from tests.integration.test_backends import BaseStorageTestCase +from tests.integration.test_backends import BaseCacheTest, BaseStorageTest @pytest.fixture(scope='module', autouse=True) @@ -19,22 +18,24 @@ def ensure_connection(): client.server_info() -class MongoDictTestCase(BaseStorageTestCase, unittest.TestCase): - def __init__(self, *args, **kwargs): - super().__init__(*args, storage_class=MongoDict, **kwargs) +class TestMongoDict(BaseStorageTest): + storage_class = MongoDict -class MongoPickleDictTestCase(BaseStorageTestCase, unittest.TestCase): - def __init__(self, *args, **kwargs): - super().__init__(*args, storage_class=MongoPickleDict, picklable=True, **kwargs) +class TestMongoPickleDict(BaseStorageTest): + storage_class = MongoPickleDict + picklable = True + @patch('requests_cache.backends.mongo.MongoClient') + @patch( + 'requests_cache.backends.mongo.get_valid_kwargs', + side_effect=lambda cls, kwargs: get_valid_kwargs(MongoClient, kwargs), + ) + def test_connection_kwargs(self, mock_get_valid_kwargs, mock_client): + """A spot check to make sure optional connection kwargs gets passed to connection""" + MongoDict('test', host='http://0.0.0.0', port=1234, invalid_kwarg='???') + mock_client.assert_called_with(host='http://0.0.0.0', port=1234) -@patch('requests_cache.backends.mongo.MongoClient') -@patch( - 'requests_cache.backends.mongo.get_valid_kwargs', - side_effect=lambda cls, kwargs: get_valid_kwargs(MongoClient, kwargs), -) -def test_connection_kwargs(mock_get_valid_kwargs, mock_client): - """A spot check to make sure optional connection kwargs gets passed to connection""" - MongoDict('test', host='http://0.0.0.0', port=1234, invalid_kwarg='???') - mock_client.assert_called_with(host='http://0.0.0.0', port=1234) + +class TestMongoCache(BaseCacheTest): + backend_class = MongoCache |