33import pytest
44
55from fastapi_cache_plus import close_caches
6- from fastapi_cache_plus .backends .memory import CACHE_KEY , BaseCacheBackend
6+ from fastapi_cache_plus .backends .memory import CACHE_KEY , InMemoryCacheBackend
77from fastapi_cache_plus .registry import CacheRegistry
88
99
@@ -23,15 +23,15 @@ def test_get_from_registry_should_return_none_if_cache_not_registered(
2323def test_get_from_registry_should_return_cache_instance (
2424 cache_registry : CacheRegistry ,
2525) -> None :
26- cache = BaseCacheBackend ()
26+ cache = InMemoryCacheBackend ()
2727 cache_registry .set (CACHE_KEY , cache )
2828 assert cache_registry .get (CACHE_KEY ) == cache
2929
3030
3131def test_retrieve_all_registered_caches_from_registry (
3232 cache_registry : CacheRegistry ,
3333) -> None :
34- cache = BaseCacheBackend ()
34+ cache = InMemoryCacheBackend ()
3535 cache_registry .set (CACHE_KEY , cache )
3636 cache_registry .set ("OTHER_CACHE_KEY" , cache )
3737 assert cache_registry .all () == (cache , cache )
@@ -40,14 +40,14 @@ def test_retrieve_all_registered_caches_from_registry(
4040def test_registry_should_raise_error_on_dublicate_cache_key (
4141 cache_registry : CacheRegistry ,
4242) -> None :
43- cache = BaseCacheBackend ()
43+ cache = InMemoryCacheBackend ()
4444 cache_registry .set (CACHE_KEY , cache )
4545 with pytest .raises (NameError , match = "Cache with the same name already registered" ):
4646 cache_registry .set (CACHE_KEY , cache )
4747
4848
4949def test_remove_cache_from_registry (cache_registry : CacheRegistry ) -> None :
50- cache = BaseCacheBackend ()
50+ cache = InMemoryCacheBackend ()
5151 cache_registry .set (CACHE_KEY , cache )
5252 cache_registry .remove (CACHE_KEY )
5353 assert cache_registry .get (CACHE_KEY ) is None
@@ -63,7 +63,7 @@ def test_remove_cache_from_registry_should_raise_error_if_cache_not_register(
6363def test_flush_should_remove_all_registered_cashes (
6464 cache_registry : CacheRegistry ,
6565) -> None :
66- cache = BaseCacheBackend ()
66+ cache = InMemoryCacheBackend ()
6767 cache_registry .set (CACHE_KEY , cache )
6868 cache_registry .set ("OTHER_CACHE_KEY" , cache )
6969 cache_registry .flush ()
@@ -75,7 +75,7 @@ def test_flush_should_remove_all_registered_cashes(
7575async def test_close_caches_should_not_raise_exception (
7676 cache_registry : CacheRegistry ,
7777) -> None :
78- cache = BaseCacheBackend ()
78+ cache = InMemoryCacheBackend ()
7979 cache_registry .set (CACHE_KEY , cache )
8080 cache_registry .set ("OTHER_CACHE_KEY" , cache )
8181 await close_caches ()
0 commit comments