def teams(app) -> List[Team]:
# Two departments, four teams
# Department 1 has 3 teams
# Department 2 has 1 team
department1 = Team(parent_team=None,
code="department1",
name="Department 1")
department2 = Team(parent_team=None,
code="department2",
name="Department 2")
department1.sub_teams = [
Team(code="11", name="Team 11"),
Team(code="12", name="Team 12"),
Team(code="13", name="Team 13"),
]
department2.sub_teams = [Team(code="21", name="Team 21")]
db.session.add(department1)
db.session.add(department2)
db.session.add_all(department1.sub_teams)
db.session.add_all(department2.sub_teams)
db.session.commit()
return [department1, department2
] + department1.sub_teams + department2.sub_teams
def add_teams(self):
department = Team(parent_team=None, name="Department", code="Dptmnt")
db.session.add(department)
team_codes_and_names = {
"CodeA": "Team A",
"CodeB": "Team B",
"CodeNameChange": "Team with name change",
"CodeOneMember": "Team with only one member",
}
for code, name in team_codes_and_names.items():
db.session.add(Team(parent_team=department, name=name, code=code))
db.session.commit()
def setup_required_objects(self, base_date: datetime):
team = Team(parent_team=None, code="ABC", name="Team ABC")
db.session.add(team)
db.session.commit()
activity = Activity(team_id=team.team_id, activity_name="ABC")
db.session.add(activity)
db.session.commit()
sprint = Sprint(
activity_id=activity.activity_id,
last_updated=datetime(2020, 4, 5, 6),
name="ABC 1",
state=Sprint.State.CLOSED.value,
start_date=base_date,
end_date=base_date + timedelta(days=13),
complete_date=base_date + timedelta(days=13),
)
db.session.add(sprint)
db.session.commit()
status_mapping_dict = {
"To Do": "To Do",
"In Progress": "In Progress",
"In Review": "In Progress",
"Done": "Done",
"Closed": "Done",
}
for status, status_category_str in status_mapping_dict.items():
status_category = StatusCategory(status_category_str)
status_mapping_obj = StatusCategoryStatusMapping(
status=status, status_category=status_category)
db.session.add(status_mapping_obj)
db.session.commit()
return activity, sprint
def test_earliest_and_latest_date(self, ot_visual, mocker):
# Create some OT measurements
dates = ({date(2019, 3, 10),
date(2019, 5, 5),
date(2019, 5, 10)}
| {date(2019, 8, day)
for day in range(1, 32)}
| {date(2019, 9, day)
for day in range(1, 10)}
| {date(2019, 10, 28)})
team = Team(parent_team=None, code="ABC", name="Team ABC")
for measurement_date in dates:
db.session.add(
OTMeasurement(
measurement_date=measurement_date,
team=team,
workdays_fix=20,
workdays_actual=20,
overtime=timedelta(hours=10, minutes=15),
))
db.session.commit()
mocker.patch("visuals.work_time.current_user", UserMock())
# Check if dates in overtime chart match
assert ot_visual.get_earliest_date() == min(dates)
assert ot_visual.get_latest_date() == max(dates)
def test_load_team_tree_permissions_no_access(self):
u = User(email="test@localhost")
db.session.add(u)
db.session.add(Team(name="A", code="A"))
db.session.commit()
result = load_team_tree_permissions(u)
assert len(result.team_roles) == 0
assert len(result.team_parents) == 0
assert len(result.team_children) == 0
def test_create_team_top_level(self):
team = Team(parent_team=None, code="ABC", name="Team ABC")
assert team is not None
db.session.add(team)
db.session.commit()
# Check if the team was actually added
assert (db.session.query(Team).filter(
Team.code == "ABC").one_or_none() is not None)
def _create_teams(l, parent_id=None):
for team_name, role, children in l:
t = Team(name=team_name, code=team_name, parent_id=parent_id)
db.session.add(t)
db.session.flush()
if role:
ut = UserTeam(user=u, team=t, role=role)
db.session.add(ut)
db.session.flush()
team_id_by_name[team_name] = t.team_id
_create_teams(children, parent_id=t.team_id)
def test_delete_records_team(self, db_session):
# Create some records in the database
for i in range(5):
db.session.add(
Team(code=str(i), name=f"Team {i}", parent_team=None))
db.session.commit()
# Delete records
success = action_delete_all_for_model(model_name="Team", output=print)
# Assert DB has no records in it for model
assert db.session.query(Team).count() == 0
# Assert function returned True
assert success
def setup_required_objects(self):
team = Team(parent_team=None, code="ABC", name="Team ABC")
db.session.add(team)
project = JiraProject(project_key="TP-1", project_name="Test Project 1")
db.session.add(project)
db.session.commit()
activity = Activity(
team_id=team.team_id,
activity_name="ABC Activity",
jira_project_id=project.id,
)
db.session.add(activity)
db.session.commit()
def team(db_session):
team = Team(parent_team=None, code="ABC", name="Team ABC")
flask_app_db.session.add(team)
flask_app_db.session.commit()
return team
def add_teams(self):
team_names = {"Team A", "Team B", "Team C", "Team D"}
for team_name in team_names:
db.session.add(
Team(parent_team=None, name=team_name, code=team_name))
db.session.commit()
def thc_data(
db_session
) -> Tuple[List[Team], List[THCQuestion], List[THCMeasurement]]:
# Set up team health check test data
# Teams
team = Team(name="Team A", code="TeamA") # 6 members
teams = [team]
# Questions
DECK = "Test-deck"
q1 = THCQuestion(deck=DECK,
topic="Topic 1",
answer_green="Good",
answer_red="Bad")
q2 = THCQuestion(deck=DECK,
topic="Topic 2",
answer_green="Good",
answer_red="Bad")
q_new = THCQuestion(deck=DECK,
topic="New topic",
answer_green="Great",
answer_red="Oh no!")
questions = [q1, q2, q_new]
# Clear result
m11 = THCMeasurement(
measurement_date=LAST_YEAR_DATE,
session_name=LAST_YEAR,
team=team,
question=q1,
result_red=0,
result_yellow=0,
result_green=6,
)
# Tie
m12 = THCMeasurement(
measurement_date=LAST_YEAR_DATE,
session_name=LAST_YEAR,
team=team,
question=q2,
result_red=2,
result_yellow=2,
result_green=2,
)
# One member missing
m21 = THCMeasurement(
measurement_date=PAST_DATE,
session_name=PAST,
team=team,
question=q1,
result_red=0,
result_yellow=0,
result_green=5,
)
# Result changed
m22 = THCMeasurement(
measurement_date=PAST_DATE,
session_name=PAST,
team=team,
question=q2,
result_red=1,
result_yellow=3,
result_green=2,
)
m31 = THCMeasurement(
measurement_date=RECENT_DATE,
session_name=RECENT,
team=team,
question=q1,
result_red=0,
result_yellow=0,
result_green=5,
)
# Result changed
m32 = THCMeasurement(
measurement_date=RECENT_DATE,
session_name=RECENT,
team=team,
question=q2,
result_red=1,
result_yellow=3,
result_green=2,
)
# New question
m33 = THCMeasurement(
measurement_date=RECENT_DATE,
session_name=RECENT,
team=team,
question=q_new,
result_red=2,
result_yellow=1,
result_green=3,
)
measurements = [m11, m12, m21, m22, m31, m32, m33]
db.session.add_all(teams)
db.session.add_all(questions)
db.session.add_all(measurements)
db.session.commit()
return teams, questions, measurements