Python user_role_prompt示例

示例#1

    def get_role(self):
        """
        Prompt the user to choose a Role ARN if more than one is available
        """

        if not self.all_roles:
            self.get_arns()

        if not self.all_roles:
            raise Exception("No roles found")

        # If I have more than one role, ask the user which one they want,
        # otherwise just proceed

        self.role_arn, self.principal_arn = user_role_prompt(self.all_roles)

示例#2

文件： test_userquery.py 项目： glnds/onelogin-aws-cli

    def test_user_role_prompt(self):
        mock_stdout = StringIO()

        with patch('builtins.input', side_effect=['2']):
            with contextlib.redirect_stdout(mock_stdout):
                selected_role = user_role_prompt([
                    ('mock_role1', 'mock_principal_1'),
                    ('mock_role2', 'mock_principal_2'),
                    ('mock_role3', 'mock_principal_3')
                ])

        self.assertEqual(('mock_role2', 'mock_principal_2'), selected_role)
        self.assertEqual("""[1] mock_role1
[2] mock_role2
[3] mock_role3
""", mock_stdout.getvalue())

示例#3

文件： test_userquery.py 项目： zaccafferty/onelogin-aws-cli

    def test_user_role_prompt_one_option(self):
        selected_role = user_role_prompt([
            ('mock_role1', 'mock_principal_1'),
        ])

        self.assertEqual(('mock_role1', 'mock_principal_1'), selected_role)