mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-10 17:20:41 +03:00
bf7a5520dc
* try to add exception handler * improve server error test * fix lint * add build_uri util * fix header file path --------- Co-authored-by: Dmitry Afanasyev <afanasiev@litres.ru>
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
import time
|
|
from typing import Callable
|
|
|
|
import pytest
|
|
|
|
from core.utils import build_uri, timed_lru_cache
|
|
|
|
|
|
class TestTimedLruCache:
|
|
call_count: int = 0
|
|
|
|
def sum_two_numbers(self, first: int, second: int) -> int:
|
|
self.call_count += 1
|
|
return first + second
|
|
|
|
def test_timed_lru_cache_cached_for_1_second(self) -> None:
|
|
self.call_count = 0
|
|
|
|
tested_func = timed_lru_cache(seconds=1)(self.sum_two_numbers)
|
|
|
|
self._call_function_many_times(call_times=2, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.5)
|
|
self._call_function_many_times(call_times=4, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(1)
|
|
self._call_function_many_times(call_times=3, func=tested_func, first=2, second=40, result=42)
|
|
assert tested_func(2, 2) == 4
|
|
assert self.call_count == 3
|
|
|
|
def test_timed_lru_cache_cached_for_long_time(self) -> None:
|
|
self.call_count = 0
|
|
|
|
tested_func = timed_lru_cache(minutes=5)(self.sum_two_numbers)
|
|
|
|
self._call_function_many_times(call_times=3, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.2)
|
|
self._call_function_many_times(call_times=4, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.2)
|
|
self._call_function_many_times(call_times=2, func=tested_func, first=2, second=40, result=42)
|
|
assert tested_func(2, 2) == 4
|
|
assert self.call_count == 2
|
|
|
|
def test_timed_lru_cache_cached_for_short_time(self) -> None:
|
|
self.call_count = 0
|
|
|
|
tested_func = timed_lru_cache(milliseconds=200)(self.sum_two_numbers)
|
|
|
|
self._call_function_many_times(call_times=2, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.3)
|
|
self._call_function_many_times(call_times=5, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.3)
|
|
self._call_function_many_times(call_times=7, func=tested_func, first=2, second=40, result=42)
|
|
time.sleep(0.3)
|
|
self._call_function_many_times(call_times=3, func=tested_func, first=2, second=40, result=42)
|
|
assert tested_func(2, 2) == 4
|
|
assert self.call_count == 5
|
|
|
|
@staticmethod
|
|
def _call_function_many_times(
|
|
call_times: int, func: Callable[[int, int], int], first: int, second: int, result: int
|
|
) -> None:
|
|
for _ in range(call_times):
|
|
assert func(first, second) == result
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"uri_parts, expected_result",
|
|
[
|
|
(["", "admin"], "/admin"),
|
|
(["/gpt", "admin"], "/gpt/admin"),
|
|
(["/gpt", "/chat"], "/gpt/chat"),
|
|
(["gpt", ""], "/gpt"),
|
|
(["gpt"], "/gpt"),
|
|
(["gpt", "chat"], "/gpt/chat"),
|
|
(["", ""], "/"),
|
|
(["gpt", "/chat", "/admin"], "/gpt/chat/admin"),
|
|
(["gpt/", "/chat/", "/admin"], "/gpt/chat/admin"),
|
|
],
|
|
)
|
|
def test_build_uri_with_slash_prefix(uri_parts: list[str], expected_result: str) -> None:
|
|
assert build_uri(uri_parts) == expected_result
|