def add_user(self, user, identity):
r = TeamUserRelationship.get(team_id=self.id,
user_id=user.name)
if not r:
TeamUserRelationship.create(team_id=self.id,
user_id=user.name,
identity=identity)
return True
def ttest_at_team(self): # FIXME
mention = "test_team"
team_name = "测试team"
description = "测试"
team = Team.add(mention, team_name, description)
ok_(team.uid == mention)
content = "@test_team code"
users = get_mentions_from_text(content)
# ok_(len(users) == 0)
team_id = team.id
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
ok_(rl.user_id == user_id)
users = get_mentions_from_text(content)
ok_(users[0] == user_id)
rl.delete()
users = get_mentions_from_text(content)
ok_(len(users) == 0)
team.delete()
users = get_mentions_from_text(content)
ok_(users[0] == mention)
def test_at_team(self):
mention = "test_team"
team_name = "测试team"
description = "测试"
team = Team.add(mention, team_name, description)
ok_(team.uid == mention)
content = "@test_team code"
users = get_mentions_from_text(content)
ok_(len(users) == 0)
team_id = team.id
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
ok_(rl.user_id == user_id)
users = get_mentions_from_text(content)
ok_(users[0] == user_id)
rl.delete()
users = get_mentions_from_text(content)
ok_(len(users) == 0)
team.delete()
users = get_mentions_from_text(content)
ok_(users[0] == mention)
def test_add_and_delete_team_user_relationship(self):
team_id = 1111
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
relationship = TeamUserRelationship.get(team_id=team_id,
user_id=user_id)
ok_(rl.id == relationship.id)
rl.delete()
def test_add_and_delete_team_user_relationship(self):
team_id = 1111
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
relationship = TeamUserRelationship.get(team_id=team_id,
user_id=user_id)
ok_(rl.id == relationship.id)
rl.delete()
def add_member(self, username):
r = TeamUserRelationship.get(team_id=self.id,
user_id=username)
if not r:
r = TeamUserRelationship.create(team_id=self.id,
user_id=username,
identity=TEAM_MEMBER)
return True
if r.identity != TEAM_MEMBER:
r.identity = TEAM_MEMBER
r.save()
return True