113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
# Copyright (C) 2016-2024 RhodeCode GmbH
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License, version 3
|
|
# (only), as published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# This program is dual-licensed. If you wish to learn more about the
|
|
# RhodeCode Enterprise Edition, including its added features, Support services,
|
|
# and proprietary license terms, please see https://rhodecode.com/licenses/
|
|
|
|
import time
|
|
|
|
import pytest
|
|
|
|
from rhodecode.lib import rc_cache
|
|
|
|
|
|
@pytest.mark.usefixtures("app")
|
|
class TestCaches(object):
|
|
def test_cache_decorator_init_not_configured(self):
|
|
with pytest.raises(EnvironmentError):
|
|
rc_cache.get_or_create_region("dontexist")
|
|
|
|
@pytest.mark.parametrize(
|
|
"region_name",
|
|
[
|
|
"cache_perms",
|
|
"cache_perms",
|
|
],
|
|
)
|
|
def test_cache_decorator_init(self, region_name):
|
|
namespace = region_name
|
|
cache_region = rc_cache.get_or_create_region(region_name, region_namespace=namespace)
|
|
assert cache_region
|
|
|
|
@pytest.mark.parametrize(
|
|
"example_input",
|
|
[
|
|
("",),
|
|
("/ac",),
|
|
("/ac", 1, 2, object()),
|
|
("/ęćc", 1, 2, object()),
|
|
("/ąac",),
|
|
("/ac",),
|
|
],
|
|
)
|
|
def test_cache_manager_create_key(self, example_input):
|
|
key = rc_cache.utils.compute_key_from_params(*example_input)
|
|
assert key
|
|
|
|
@pytest.mark.parametrize("example_namespace", ["namespace", None])
|
|
@pytest.mark.parametrize(
|
|
"example_input",
|
|
[
|
|
("",),
|
|
("/ac",),
|
|
("/ac", 1, 2, object()),
|
|
("/ęćc", 1, 2, object()),
|
|
("/ąac",),
|
|
("/ac",),
|
|
],
|
|
)
|
|
def test_cache_keygen(self, example_input, example_namespace):
|
|
def func_wrapped():
|
|
return 1
|
|
|
|
func = rc_cache.utils.custom_key_generator(None, example_namespace, func_wrapped)
|
|
key = func(*example_input)
|
|
assert key
|
|
|
|
def test_store_value_in_cache(self):
|
|
cache_region = rc_cache.get_or_create_region("cache_perms", "test_cache")
|
|
# make sure we empty the cache now
|
|
cache_region.delete_multi(cache_region.backend.list_keys())
|
|
|
|
assert cache_region.backend.list_keys() == []
|
|
|
|
@cache_region.conditional_cache_on_arguments(expiration_time=5)
|
|
def compute(key):
|
|
return time.time()
|
|
|
|
for x in range(10):
|
|
compute(x)
|
|
|
|
assert len(set(cache_region.backend.list_keys())) == 10
|
|
|
|
def test_store_and_get_value_from_region(self):
|
|
cache_region = rc_cache.get_or_create_region("cache_perms", "test_cache")
|
|
# make sure we empty the cache now
|
|
for key in cache_region.backend.list_keys():
|
|
cache_region.delete(key)
|
|
assert cache_region.backend.list_keys() == []
|
|
|
|
@cache_region.conditional_cache_on_arguments(expiration_time=5)
|
|
def compute(key):
|
|
return time.time()
|
|
|
|
result = set()
|
|
for x in range(10):
|
|
ret = compute("x")
|
|
result.add(ret)
|
|
|
|
# once computed we have only one value (the same from cache)
|
|
# after executing it 10x
|
|
assert len(result) == 1
|