def test_handle_remove(self):
"""Test team command remove parser."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
test_user.github_username = "******"
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
other_user = User("anotheruser")
other_user.github_id = "githubID"
other_user.github_username = "******"
self.db.retrieve.side_effect = [test_user, other_user,
test_user, other_user]
self.db.query.return_value = [team]
team_attach = [team.get_attachment()]
with self.app.app_context():
self.testcommand.handle("team add brs ID", user)
resp, code = self.testcommand.handle("team remove brs ID", user)
expect = json.loads(jsonify({'attachments': team_attach,
'text': 'Removed '
'User from brs'}).data)
resp = json.loads(resp.data)
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.store.assert_called_with(team)
self.gh.remove_team_member.assert_called_once_with("myuser",
"githubid")
def test_check_credentials_not_lead(self):
"""Test checking to see if user is lead for certain team."""
user = User("USFAS689")
user.github_id = "IDGithub"
team = Team("brussels", "team", "id")
team.add_member(user.github_id)
user.permissions_level = Permissions.team_lead
self.assertFalse(util.check_permissions(user, team))
def test_handle_view(self):
"""Test team command view parser."""
team = Team("BRS", "brs", "web")
team_attach = [team.get_attachment()]
self.db.query.return_value = [team]
with self.app.app_context():
resp, code = self.testcommand.handle("team view brs", user)
expect = {'attachments': team_attach}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
def org_get_teams(self) -> List[Team]:
"""Return array of teams associated with organization."""
teams = self.org.get_teams()
team_array = []
for team in teams:
# convert PaginatedList to List
team_model = ModelTeam(str(team.id), team.name, "")
team_model.members = set(
str(user.id) for user in self.list_team_members(team.id))
team_array.append(team_model)
return team_array
def test_handle_add_not_admin(self):
"""Test team command add parser with insufficient permission."""
test_user = User("userid")
test_user.github_username = "******"
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
self.assertTupleEqual(self.testcommand.handle("team add brs ID", user),
(self.testcommand.permission_error, 200))
self.db.store.assert_not_called()
self.gh.add_team_member.assert_not_called()
def test_handle_delete(self):
"""Test team command delete parser."""
team = Team("BRS", "brs", "web")
team.github_team_id = "12345"
test_user = User("userid")
test_user.github_id = "1234"
team.add_team_lead("1234")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
self.assertTupleEqual(self.testcommand.handle("team delete brs", user),
(f"Team brs deleted", 200))
self.db.delete.assert_called_once_with(Team, "12345")
self.gh.org_delete_team.assert_called_once_with(int("12345"))
def test_handle_add_github_error(self):
"""Test team command add parser with Github Exception."""
test_user = User("userid")
test_user.github_username = "******"
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
add_user = User("anotheruser")
add_user.github_username = "******"
self.db.retrieve.side_effect = [test_user, add_user]
self.db.query.return_value = [team]
self.gh.add_team_member.side_effect = GithubAPIException("error")
self.assertTupleEqual(self.testcommand.handle("team add brs ID", user),
("User added unsuccessfully with the"
" following error: error", 200))
self.db.store.assert_not_called()
def test_handle_refresh_changed(self):
"""Test team command refresh parser if team edited in github."""
test_user = User(user)
test_user.permissions_level = Permissions.admin
team = Team("TeamID", "TeamName", "android")
team_update = Team("TeamID", "new team name", "android")
team_update.add_member(test_user.github_id)
team2 = Team("OTEAM", "other team2", "ios")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team, team2]
self.gh.org_get_teams.return_value = [team_update, team2]
attach = team_update.get_attachment()
status = f"1 teams changed, " \
f"0 added, " \
f"0 deleted. Wonderful."
with self.app.app_context():
resp, code = self.testcommand.handle("team refresh", user)
expect = json.loads(jsonify({'attachments': [attach],
'text': status}).data)
resp = json.loads(resp.data)
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.query.assert_called_once_with(Team)
def test_handle_remove_not_in_team(self):
"""Test team command remove parser when user is not in team."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
other_user = User("anotheruser")
other_user.github_id = "githubID"
other_user.github_username = "******"
self.db.retrieve.side_effect = [test_user, other_user]
self.db.query.return_value = [team]
self.gh.has_team_member.return_value = False
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team remove"
" brs ID", user),
("User not in team!", 200))
self.gh.has_team_member.assert_called_once_with("myuser", "githubid")
self.db.store.assert_not_called()
self.gh.remove_team_member.assert_not_called()
def test_handle_delete_not_admin(self):
"""Test team command delete parser with improper permission."""
team = Team("BRS", "brs", "web")
test_user = User("userid")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
self.assertTupleEqual(self.testcommand.handle("team delete brs", user),
(self.testcommand.permission_error, 200))
self.db.delete.assert_not_called()
self.gh.org_delete_team.assert_not_called()
def test_handle_edit_not_admin(self):
"""Test team command edit parser with insufficient permission."""
test_user = User("userid")
team = Team("BRS", "brs", "brS")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team "
"edit brs", user),
(self.testcommand.permission_error, 200))
self.db.store.assert_not_called()
def test_handle_remove_not_admin(self):
"""Test team command remove parser with insufficient permission."""
test_user = User("userid")
team = Team("BRS", "brs", "web")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team remove"
" brs ID", user),
(self.testcommand.permission_error, 200))
self.db.store.assert_not_called()
self.gh.remove_team_member.assert_not_called()
def test_handle_refresh_addition_and_deletion(self):
"""Test team command refresh parser if local differs from github."""
test_user = User(user)
test_user.permissions_level = Permissions.admin
team = Team("TeamID", "TeamName", "")
team2 = Team("OTEAM", "other team", "android")
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team2]
# In this case, github does not have team2!
self.gh.org_get_teams.return_value = [team]
attach = team.get_attachment()
attach2 = team2.get_attachment()
status = f"0 teams changed, " \
f"1 added, " \
f"1 deleted. Wonderful."
with self.app.app_context():
resp, code = self.testcommand.handle("team refresh", user)
expect = json.loads(jsonify({'attachments': [attach2, attach],
'text': status}).data)
resp = json.loads(resp.data)
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.query.assert_called_once_with(Team)
def test_handle_refresh_lookup_error(self):
"""Test team command refresh parser with lookup error."""
test_user = User(user)
team = Team("BRS", "brs", "brS")
test_user.permissions_level = Permissions.admin
self.db.retrieve.return_value = None
self.db.retrieve.side_effect = LookupError
self.gh.org_get_teams.return_value = [team]
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team refresh",
user),
(self.testcommand.lookup_error, 200))
self.db.store.assert_not_called()
def test_handle_lead_user_error(self):
"""Test team command lead remove parser with user not in team."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "web")
self.db.retrieve.side_effect = [test_user, test_user]
self.db.query.return_value = [team]
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team lead "
"--remove brs"
" ID", user),
("User not in team!", 200))
self.db.store.assert_not_called()
def test_handle_add(self):
"""Test team command add parser."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
test_user.github_username = "******"
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
add_user = User("anotheruser")
add_user.github_username = "******"
add_user.github_id = "otherID"
self.db.retrieve.side_effect = [test_user, add_user]
self.db.query.return_value = [team]
with self.app.app_context():
resp, code = self.testcommand.handle("team add brs ID", user)
team_attach = team.get_attachment()
expect = {'attachments': [team_attach],
'text': 'Added User to brs'}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.store.assert_called_with(team)
assert team.has_member("otherID")
self.gh.add_team_member.assert_called_once_with("myuser", "githubid")
def test_handle_lead_github_error(self):
"""Test team command lead parser with github error."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "web")
self.db.retrieve.side_effect = [test_user, test_user]
self.db.query.return_value = [team]
self.gh.add_team_member.side_effect = GithubAPIException("error")
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team lead brs ID",
user),
("Edit team lead was unsuccessful with the "
"following error: error", 200))
self.db.store.assert_not_called()
def test_handle_remove_github_error(self):
"""Test team command remove parser with github error."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "web")
other_user = User("anotheruser")
other_user.github_id = "githubID"
other_user.github_username = "******"
self.db.retrieve.side_effect = [test_user, other_user]
self.db.query.return_value = [team]
self.gh.has_team_member.side_effect = GithubAPIException("error")
with self.app.app_context():
self.assertTupleEqual(self.testcommand.handle("team remove"
" brs ID", user),
("User removed unsuccessfully with the "
"following error: error", 200))
self.db.store.assert_not_called()
self.gh.remove_team_member.assert_not_called()
def test_handle_list(self):
"""Test team command list parser."""
team = Team("BRS", "brs", "web")
team2 = Team("OTEAM", "other team", "android")
self.db.query.return_value = [team, team2]
attach = team.get_basic_attachment()
attach2 = team2.get_basic_attachment()
attachment = [attach, attach2]
with self.app.app_context():
resp, code = self.testcommand.handle("team list", user)
expect = {'attachments': attachment}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.query.assert_called_once_with(Team)
def test_handle_edit(self):
"""Test team command edit parser."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
team = Team("BRS", "brs", "brS")
team.platform = "web"
team_attach = [team.get_attachment()]
team.platform = "iOS"
team.display_name = "b-s"
self.db.retrieve.return_value = test_user
self.db.query.return_value = [team]
with self.app.app_context():
resp, code = self.testcommand.handle("team edit brs "
"--name brS "
"--platform web", user)
expect = {'attachments': team_attach,
'text': 'Team edited: brs, '
'name: brS, '
'platform: web'}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
self.db.store.assert_called_once_with(team)
def test_handle_lead(self):
"""Test team command lead parser with add and remove options."""
test_user = User("userid")
test_user.permissions_level = Permissions.admin
test_user.github_username = "******"
team = Team("BRS", "brs", "web")
team.github_team_id = "githubid"
other_user = User("anotheruser")
other_user.github_id = "githubID"
other_user.github_username = "******"
self.db.retrieve.side_effect = [test_user, other_user,
test_user, other_user]
self.db.query.return_value = [team]
team.add_member("githubID")
team_before_attach = [team.get_attachment()]
team.add_team_lead("githubID")
team_attach = [team.get_attachment()]
team.discard_member("githubID")
with self.app.app_context():
resp, code = self.testcommand.handle("team lead brs ID", user)
expect = {'attachments': team_attach,
'text': 'User added ' 'as team lead to brs'}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
assert team.has_member("githubID")
self.gh.add_team_member.assert_called_once_with("myuser",
"githubid")
self.db.store.assert_called()
resp, code = self.testcommand.handle("team lead "
"--remove brs ID", user)
expect = {'attachments': team_before_attach,
'text': 'User removed as team lead' ' from brs'}
self.assertDictEqual(resp, expect)
self.assertEqual(code, 200)
assert not team.has_team_lead("githubID")
self.db.store.assert_called()
