Skip to content

Commit 794d6d8

Browse files
committed
fix: use in memory backend for registry tests
1 parent 9ce9241 commit 794d6d8

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

tests/registry_tests.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import pytest
44

55
from 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
77
from fastapi_cache_plus.registry import CacheRegistry
88

99

@@ -23,15 +23,15 @@ def test_get_from_registry_should_return_none_if_cache_not_registered(
2323
def 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

3131
def 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(
4040
def 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

4949
def 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(
6363
def 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(
7575
async 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

Comments
 (0)