def run_team_management(team_name, team_id, action) -> None:
"""Create, Delete Teams or Renew TeamToken."""
try:
if action == "create":
if not team_name:
click.echo("Missing argument: --team_name")
return None
else:
team_management.create_team(team_name)
elif action == "delete":
if not team_id:
click.echo("Missing argument: --team_id")
return None
else:
click.echo(
f"Do you want to delete the team with the id: {team_id}? [y/n] ",
nl=False,
)
click.echo()
c = click.getchar()
if c == "y":
click.echo("Start deletion")
team_management.delete_team(team_id)
elif c == "n":
click.echo("Abort!")
else:
click.echo("Invalid input")
elif action == "renew-team-token":
if not team_id:
click.echo("Missing argument: --team_id")
return None
else:
team_management.renew_team_token(team_id)
elif action == "remove-all-team-members":
if not team_id:
click.echo("Missing argument: --team_id")
return None
else:
click.echo(
f"Do you want to remove all users from "
f"the team with the id: {team_id}? [y/n] ",
nl=False,
)
click.echo()
c = click.getchar()
if c == "y":
click.echo("Start remove all team members")
team_management.remove_all_team_members(team_id)
elif c == "n":
click.echo("Abort!")
else:
click.echo("Invalid input")
except Exception:
logger.exception("team management failed")
sentry.capture_exception()
def test_create_team(self):
self.team_name = "unittest-team"
self.team_id, self.team_token = team_management.create_team(
self.team_name)
# check if returned values are plausible
self.assertIsNotNone(self.team_id)
self.assertGreaterEqual(len(self.team_token), 32)
# check data in Firebase
fb_db = firebaseDB()
ref = fb_db.reference(f"v2/teams/{self.team_id}")
team_data = ref.get()
self.assertEqual(len(team_data), 2)
self.assertEqual(team_data["teamName"], self.team_name)
self.assertEqual(team_data["teamToken"], self.team_token)