test_in_memory_cache.py

# Copyright (c) 2023 - 2024, Owners of https://github.com/autogenhub
#
# SPDX-License-Identifier: Apache-2.0
#
# Portions derived from  https://github.com/microsoft/autogen are under the MIT License.
# SPDX-License-Identifier: MIT
from autogen.cache.in_memory_cache import InMemoryCache


def test_prefixed_key():
    cache = InMemoryCache(seed="test")
    assert cache._prefixed_key("key") == "test_key"


def test_get_with_default_value():
    cache = InMemoryCache()
    assert cache.get("key", "default_value") == "default_value"


def test_get_without_default_value():
    cache = InMemoryCache()
    assert cache.get("key") is None


def test_get_with_set_value():
    cache = InMemoryCache()
    cache.set("key", "value")
    assert cache.get("key") == "value"


def test_get_with_set_value_and_seed():
    cache = InMemoryCache(seed="test")
    cache.set("key", "value")
    assert cache.get("key") == "value"


def test_set():
    cache = InMemoryCache()
    cache.set("key", "value")
    assert cache._cache["key"] == "value"


def test_set_with_seed():
    cache = InMemoryCache(seed="test")
    cache.set("key", "value")
    assert cache._cache["test_key"] == "value"


def test_context_manager():
    with InMemoryCache() as cache:
        cache.set("key", "value")
        assert cache.get("key") == "value"