vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

test: make redis query count util work with any redis instance

Browse source
This commit is contained in:
Ankush Menat 2025-01-06 14:29:37 +05:30
parent f74ada155e
commit 8f2f94b7e5
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
frappe/tests/classes/integration_test_case.py
View file

@ -130,25 +130,27 @@ class IntegrationTestCase(UnitTestCase):
@contextmanager
def assertRedisCallCounts(self, count: int) -> AbstractContextManager[None]:
from frappe.utils.redis_wrapper import RedisWrapper
commands = []
def execute_command_and_count(*args, **kwargs):
ret = orig_execute(*args, **kwargs)
key_len = 2
if "H" in args[0]:
if "H" in args[1]:
key_len = 3
commands.append((args)[:key_len])
commands.append((args)[1 : key_len + 1])
return ret
try:
orig_execute = frappe.cache.execute_command
frappe.cache.execute_command = execute_command_and_count
orig_execute = RedisWrapper.execute_command
RedisWrapper.execute_command = execute_command_and_count
yield
self.assertLessEqual(
len(commands), count, msg="commands executed: \n" + "\n".join(str(c) for c in commands)
)
finally:
frappe.cache.execute_command = orig_execute
RedisWrapper.execute_command = orig_execute
@contextmanager
def assertRowsRead(self, count: int) -> AbstractContextManager[None]: