def team_for_nobody(temp_db_instance_helper):
"""
Create a team that not belongs to regural user
"""
from app.modules.teams.models import Team
for _ in temp_db_instance_helper(Team(title="Admin User's team")):
yield _
def team_for_nobody(db): # pylint: disable=invalid-name
"""
Create a team that not belongs regural user
"""
from app.modules.teams.models import Team
team = Team(title="Admin User's team")
db.session.add(team)
db.session.commit()
yield team
# Cleanup
db.session.delete(team)
db.session.commit()
def team_for_nobody(db):
"""
Create a team that not belongs to regural user
"""
from app.modules.teams.models import Team
team = Team(title="Admin User's team")
with db.session.begin():
db.session.add(team)
yield team
# Cleanup
with db.session.begin():
db.session.delete(team)
def team_for_regular_user(db, regular_user, readonly_user):
from app.modules.teams.models import Team, TeamMember
team = Team(title="Regular User's team")
regular_user_team_member = TeamMember(team=team,
user=regular_user,
is_leader=True)
readonly_user_team_member = TeamMember(team=team, user=readonly_user)
with db.session.begin():
db.session.add(team)
db.session.add(regular_user_team_member)
db.session.add(readonly_user_team_member)
yield team
# Cleanup
TeamMember.query.filter(TeamMember.team == team).delete()
Team.query.filter(Team.id == team.id).delete()
def team_for_regular_user(db, regular_user, readonly_user):
# pylint: disable=invalid-name
from app.modules.teams.models import Team, TeamMember
team = Team(title="Regular User's team")
db.session.add(team)
regular_user_team_member = TeamMember(team=team,
user=regular_user,
is_leader=True)
db.session.add(regular_user_team_member)
readonly_user_team_member = TeamMember(team=team, user=readonly_user)
db.session.add(readonly_user_team_member)
db.session.commit()
yield team
# Cleanup
db.session.delete(readonly_user_team_member)
db.session.delete(regular_user_team_member)
db.session.delete(team)
db.session.commit()