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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
from uuid import uuid4
class TestMembershipManager:
def test_init(self, membership_manager, gr_db):
assert membership_manager.pg_config == gr_db
class TestTeamManager:
def test_init(self, team_manager, gr_db):
assert team_manager.pg_config == gr_db
def test_get_or_create(self, team_manager):
from generalresearch.models.gr.team import Team
new_uuid = uuid4().hex
team: Team = team_manager.get_or_create(uuid=new_uuid)
assert isinstance(team, Team)
assert isinstance(team.id, int)
assert team.uuid == new_uuid
assert team.name == "< Unknown >"
def test_get_all(self, team_manager):
res1 = team_manager.get_all()
assert isinstance(res1, list)
team_manager.create_dummy()
res2 = team_manager.get_all()
assert len(res1) == len(res2) - 1
def test_create(self, team_manager):
from generalresearch.models.gr.team import Team
team: Team = team_manager.create_dummy()
assert isinstance(team, Team)
assert isinstance(team.id, int)
def test_add_user(self, team, team_manager, gr_um, gr_db, gr_redis_config):
from generalresearch.models.gr.authentication import GRUser
from generalresearch.models.gr.team import Membership
user: GRUser = gr_um.create_dummy()
instance = team_manager.add_user(team=team, gr_user=user)
assert isinstance(instance, Membership)
# assert team.gr_users is None
team.prefetch_gr_users(pg_config=gr_db, redis_config=gr_redis_config)
assert isinstance(team.gr_users, list)
assert len(team.gr_users)
assert team.gr_users == [user]
def test_get_by_uuid(self, team_manager):
from generalresearch.models.gr.team import Team
team: Team = team_manager.create_dummy()
instance = team_manager.get_by_uuid(team_uuid=team.uuid)
assert team.id == instance.id
def test_get_by_id(self, team_manager):
from generalresearch.models.gr.team import Team
team: Team = team_manager.create_dummy()
instance = team_manager.get_by_id(team_id=team.id)
assert team.uuid == instance.uuid
def test_get_by_user(self, team, team_manager, gr_um):
from generalresearch.models.gr.authentication import GRUser
from generalresearch.models.gr.team import Team
user: GRUser = gr_um.create_dummy()
team_manager.add_user(team=team, gr_user=user)
res = team_manager.get_by_user(gr_user=user)
assert isinstance(res, list)
assert len(res) == 1
instance = res[0]
assert isinstance(instance, Team)
assert instance.uuid == team.uuid
def test_get_by_user_duplicates(
self,
gr_user_token,
gr_user,
membership,
product_factory,
membership_factory,
team,
thl_web_rr,
gr_redis_config,
gr_db,
):
product_factory(team=team)
membership_factory(team=team, gr_user=gr_user)
gr_user.prefetch_teams(
pg_config=gr_db,
redis_config=gr_redis_config,
)
assert len(gr_user.teams) == 1
# def test_create_raise_on_duplicate(self):
# t_uuid = uuid4().hex
#
# # Make the first one
# team = TeamManager.create(
# uuid=t_uuid,
# name=f"test-{t_uuid[:6]}")
# assert isinstance(team, Team)
#
# # Try to make it again
# with pytest.raises(expected_exception=psycopg.errors.UniqueViolation):
# TeamManager.create(
# uuid=t_uuid,
# name=f"test-{t_uuid[:6]}")
#
|
