aboutsummaryrefslogtreecommitdiff
path: root/tests/managers/thl
diff options
context:
space:
mode:
authorMax Nanis2026-08-16 01:38:14 -0700
committerMax Nanis2026-08-16 01:38:14 -0700
commit24bf43ca9643c1d13f4acd9f14435249b0839df1 (patch)
treecb19022f44554a3b23085073503dd938a7fabe9e /tests/managers/thl
parentfd77cd429cc6205f9a8f2f4168298fe82d461383 (diff)
downloadgeneralresearch-24bf43ca9643c1d13f4acd9f14435249b0839df1.tar.gz
generalresearch-24bf43ca9643c1d13f4acd9f14435249b0839df1.zip
Pandera supression attempt. Conftest assert for type validation (no none allowed). min_items to min_length
Diffstat (limited to 'tests/managers/thl')
-rw-r--r--tests/managers/thl/test_user_manager/test_base.py61
1 files changed, 31 insertions, 30 deletions
diff --git a/tests/managers/thl/test_user_manager/test_base.py b/tests/managers/thl/test_user_manager/test_base.py
index 3c7ee38..0d7ffef 100644
--- a/tests/managers/thl/test_user_manager/test_base.py
+++ b/tests/managers/thl/test_user_manager/test_base.py
@@ -12,77 +12,78 @@ from generalresearch.managers.thl.user_manager import (
from generalresearch.managers.thl.user_manager.rate_limit import (
RateLimitItemPerHourConstantKey,
)
-from generalresearch.models.thl.product import UserCreateConfig, Product
-from generalresearch.models.thl.user import User
-from test_utils.models.conftest import (
- user,
- product,
- user_manager,
- product_manager,
+from generalresearch.managers.thl.user_manager.user_manager import (
+ UserManager,
)
+from generalresearch.models.thl.product import Product, UserCreateConfig
+from generalresearch.models.thl.user import User
logger = logging.getLogger()
class TestUserManager:
+ @pytest.fixture(autouse=True)
+ def _setup(self, user_manager: UserManager, user: User):
+ self.user: User = user
+ self.user_manager: UserManager = user_manager
- def test_copying_lru_cache(self, user_manager, user):
+ def test_copying_lru_cache(self):
# Before adding the deepcopy_return decorator, this would fail b/c the returned user
# is mutable, and it would mutate in the cache
# user_manager = self.get_user_manager()
- user_manager.clear_user_inmemory_cache(user)
- u = user_manager.get_user(user_id=user.user_id)
+ self.user_manager.clear_user_inmemory_cache(self.user)
+ u = self.user_manager.get_user(user_id=self.user.user_id)
assert not u.blocked
u.blocked = True
- u = user_manager.get_user(user_id=user.user_id)
+ u = self.user_manager.get_user(user_id=self.user.user_id)
assert not u.blocked
- def test_get_user_no_inmemory(self, user, user_manager):
- user_manager.clear_user_inmemory_cache(user)
- user_manager.get_user.__wrapped__.cache_clear()
- u = user_manager.get_user(user_id=user.user_id)
+ def test_get_user_no_inmemory(self):
+ self.user_manager.clear_user_inmemory_cache(self.user)
+ self.user_manager.get_user.__wrapped__.cache_clear()
+ u = self.user_manager.get_user(user_id=self.user.user_id)
# this should hit mysql
- assert u == user
+ assert u == self.user
- cache_info = user_manager.get_user.__wrapped__.cache_info()
+ cache_info = self.user_manager.get_user.__wrapped__.cache_info()
assert cache_info.hits == 0, cache_info
assert cache_info.misses == 1, cache_info
# this should hit the lru cache
- u = user_manager.get_user(user_id=user.user_id)
- assert u == user
+ u = self.user_manager.get_user(user_id=self.user.user_id)
+ assert u == self.user
- cache_info = user_manager.get_user.__wrapped__.cache_info()
+ cache_info = self.user_manager.get_user.__wrapped__.cache_info()
assert cache_info.hits == 1, cache_info
assert cache_info.misses == 1, cache_info
- def test_get_user_with_inmemory(self, user_manager, user):
+ def test_get_user_with_inmemory(self):
# user_manager = self.get_user_manager()
- user_manager.set_user_inmemory_cache(user)
- user_manager.get_user.__wrapped__.cache_clear()
- u = user_manager.get_user(user_id=user.user_id)
+ self.user_manager.set_user_inmemory_cache(self.user)
+ self.user_manager.get_user.__wrapped__.cache_clear()
+ u = self.user_manager.get_user(user_id=self.user.user_id)
# this should hit inmemory cache
- assert u == user
+ assert u == self.user
- cache_info = user_manager.get_user.__wrapped__.cache_info()
+ cache_info = self.user_manager.get_user.__wrapped__.cache_info()
assert cache_info.hits == 0, cache_info
assert cache_info.misses == 1, cache_info
# this should hit the lru cache
- u = user_manager.get_user(user_id=user.user_id)
- assert u == user
+ u = self.user_manager.get_user(user_id=self.user.user_id)
+ assert u == self.user
- cache_info = user_manager.get_user.__wrapped__.cache_info()
+ cache_info = self.user_manager.get_user.__wrapped__.cache_info()
assert cache_info.hits == 1, cache_info
assert cache_info.misses == 1, cache_info
class TestBlockUserManager:
- def test_block_user(self, product, user_manager):
+ def test_block_user(self, product, user_manager: UserManager):
product_user_id = f"user-{uuid4().hex[:10]}"
# mysql_user_manager to skip user creation limit check