harmacist/healthchecks
1
0
Fork 0
forked from GithubBackups/healthchecks
Code Issues Pull Requests Projects Releases Wiki Activity

Prepare for the removal of Profile.current_team_id

Browse Source
This commit is contained in:
Pēteris Caune 2019-01-19 17:24:54 +02:00
parent 0994006603
commit 77e3212956
No known key found for this signature in database
GPG Key ID: E28D7679E9A9EDE2
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
hc/accounts/models.py
View File

@ -180,13 +180,13 @@ class Profile(models.Model):
return self.member_set.count() < self.team_limit return self.member_set.count() < self.team_limit
def invite(self, user): def invite(self, user):
for project in self.user.project_set.all(): project = self.get_own_project()
member = Member(team=self, user=user, project=project) Member.objects.create(team=self, user=user, project=project)
member.save()
# Switch the invited user over to the new team so they # Switch the invited user over to the new team so they
# notice the new team on next visit: # notice the new team on next visit:
user.profile.current_team = self user.profile.current_team = self
user.profile.current_project = project
user.profile.save() user.profile.save()
user.profile.send_instant_login_link(self) user.profile.send_instant_login_link(self)

10
hc/accounts/tests/test_profile.py
View File

@ -142,13 +142,11 @@ class ProfileTestCase(BaseTestCase):
members = self.project.member_set.all() members = self.project.member_set.all()
self.assertEqual(members.count(), 2) self.assertEqual(members.count(), 2)
frank_found = False member = Member.objects.get(project=self.project,
for member in members.all(): user__email="frank@example.org")
self.assertEqual(member.project, self.project)
if member.user.email == "frank@example.org":
frank_found = True
self.assertTrue(frank_found) profile = member.user.profile
self.assertEqual(profile.current_project, self.project)
# And an email should have been sent # And an email should have been sent
subj = ('You have been invited to join' subj = ('You have been invited to join'