Python verify_account_accessの例

コード例 #1

ファイル: test_aws_helper.py プロジェクト: Nanyte25/cloudigrade

 def test_verify_account_access_failure(self, mock_verify_policy_action):
     """Assert that account access fails when some actions are not OK."""
     mock_session = Mock()
     expected_calls = [
         call(mock_session, 'ec2:DescribeImages'),
         call(mock_session, 'ec2:DescribeInstances'),
         call(mock_session, 'ec2:ModifySnapshotAttribute'),
         call(mock_session, 'ec2:DescribeSnapshotAttribute'),
         call(mock_session, 'ec2:DescribeSnapshots'),
         call(mock_session, 'ec2:CopyImage'),
         call(mock_session, 'ec2:CreateTags'),
         call(mock_session, 'cloudtrail:CreateTrail'),
         call(mock_session, 'cloudtrail:UpdateTrail'),
         call(mock_session, 'cloudtrail:PutEventSelectors'),
         call(mock_session, 'cloudtrail:DescribeTrails'),
         call(mock_session, 'cloudtrail:StartLogging'),
     ]
     mock_verify_policy_action.side_effect = [
         True, True, True, False, True, True, True, True, True, True, True,
         True
     ]
     verified, failed_actions = helper.verify_account_access(mock_session)
     self.assertFalse(verified)
     self.assertEqual(len(failed_actions), 1)
     mock_verify_policy_action.assert_has_calls(expected_calls)

コード例 #2

ファイル: test_aws_helper.py プロジェクト: cdcabrera/cloudigrade

 def test_verify_account_access_success(self, mock_verify_policy_action):
     """Assert that account access is verified when all actions are OK."""
     mock_session = Mock()
     expected_calls = [
         call(mock_session, 'ec2:DescribeImages'),
         call(mock_session, 'ec2:DescribeInstances'),
         call(mock_session, 'ec2:ModifySnapshotAttribute'),
         call(mock_session, 'ec2:DescribeSnapshotAttribute'),
         call(mock_session, 'ec2:DescribeSnapshots'),
     ]
     mock_verify_policy_action.side_effect = [True, True, True, True, True]
     verified, failed_actions = helper.verify_account_access(mock_session)
     self.assertTrue(verified)
     self.assertEqual(len(failed_actions), 0)
     mock_verify_policy_action.assert_has_calls(expected_calls)

コード例 #3

 def test_verify_account_access_success(self, mock_verify_policy_action):
     """Assert that account access is verified when all actions are OK."""
     mock_session = Mock()
     expected_calls = [
         call(mock_session, "ec2:DescribeImages"),
         call(mock_session, "ec2:DescribeInstances"),
         call(mock_session, "ec2:ModifySnapshotAttribute"),
         call(mock_session, "ec2:DescribeSnapshotAttribute"),
         call(mock_session, "ec2:DescribeSnapshots"),
         call(mock_session, "ec2:CopyImage"),
         call(mock_session, "ec2:CreateTags"),
         call(mock_session, "ec2:DescribeRegions"),
         call(mock_session, "cloudtrail:CreateTrail"),
         call(mock_session, "cloudtrail:UpdateTrail"),
         call(mock_session, "cloudtrail:PutEventSelectors"),
         call(mock_session, "cloudtrail:DescribeTrails"),
         call(mock_session, "cloudtrail:StartLogging"),
         call(mock_session, "cloudtrail:DeleteTrail"),
     ]
     mock_verify_policy_action.side_effect = [
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
         True,
     ]
     verified, failed_actions = helper.verify_account_access(mock_session)
     self.assertTrue(verified)
     self.assertEqual(len(failed_actions), 0)
     mock_verify_policy_action.assert_has_calls(expected_calls)