1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
from __future__ import annotations
from collections.abc import Callable
from uuid import uuid4
import pytest
from generalresearch.managers.thl.user_manager.user_manager import UserManager
from generalresearch.models.thl.product import Product
from generalresearch.models.thl.user import User
class TestUserManagerFetch:
def test_fetch(
self,
user_factory: Callable[..., User],
product: Product,
user_manager: UserManager,
):
user1: User = user_factory(product=product)
user2: User = user_factory(product=product)
res = user_manager.fetch_by_bpuids(
product_id=product.uuid,
product_user_ids=[user1.product_user_id, user2.product_user_id],
)
assert len(res) == 2
res = user_manager.fetch(user_ids=[user1.user_id, user2.user_id])
assert len(res) == 2
res = user_manager.fetch(user_uuids=[user1.uuid, user2.uuid])
assert len(res) == 2
# filter including bogus values
res = user_manager.fetch(user_uuids=[user1.uuid, uuid4().hex])
assert len(res) == 1
res = user_manager.fetch(user_uuids=[uuid4().hex])
assert len(res) == 0
def test_fetch_invalid(self, user_manager: UserManager):
with pytest.raises(AssertionError) as e:
user_manager.fetch(user_uuids=[], user_ids=None)
assert "Must pass ONE of user_ids, user_uuids" in str(e.value)
with pytest.raises(AssertionError) as e:
user_manager.fetch(user_uuids=uuid4().hex)
assert "must pass a collection of user_uuids" in str(e.value)
with pytest.raises(AssertionError) as e:
user_manager.fetch(user_uuids=[uuid4().hex], user_ids=[1, 2, 3])
assert "Must pass ONE of user_ids, user_uuids" in str(e.value)
with pytest.raises(AssertionError) as e:
user_manager.fetch(user_ids=list(range(501)))
assert "limit 500 user_ids" in str(e.value)
|
