# 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
