def test_enterprise_team(self):
params = get_connected_params()
api.query_enterprise(params)
cmd = enterprise.EnterpriseTeamCommand()
with mock.patch('builtins.print'):
cmd.execute(params, team=[ent_env.team1_uid])
TestEnterprise.expected_commands = ['team_add']
cmd.execute(params,
add=True,
restrict_edit='on',
node=str(ent_env.node1_id),
team=['Team 3'])
self.assertEqual(len(TestEnterprise.expected_commands), 0)
with mock.patch('keepercommander.commands.enterprise.user_choice'
) as mock_choice:
TestEnterprise.expected_commands = ['team_delete']
mock_choice.return_value = 'y'
cmd.execute(params, delete=True, team=['Team 1'])
self.assertEqual(len(TestEnterprise.expected_commands), 0)
with mock.patch('builtins.print'):
mock_choice.return_value = 'n'
cmd.execute(params, delete=True, team=[ent_env.team1_uid])
self.assertEqual(len(TestEnterprise.expected_commands), 0)
with self.assertLogs(level=logging.WARNING):
cmd.execute(params, delete=True, team=['Unknown Team'])
self.assertEqual(len(TestEnterprise.expected_commands), 0)
def test_enterprise_team_user(self):
params = get_connected_params()
api.query_enterprise(params)
cmd = enterprise.EnterpriseTeamCommand()
TestEnterprise.expected_commands = ['team_enterprise_user_add']
cmd.execute(params, add_user=[ent_env.user2_email], team=ent_env.team1_uid)
self.assertEqual(len(TestEnterprise.expected_commands), 0)
TestEnterprise.expected_commands = ['team_enterprise_user_remove']
cmd.execute(params, remove_user=[ent_env.user2_email], team=ent_env.team1_uid)
self.assertEqual(len(TestEnterprise.expected_commands), 0)