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)