KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use get_bucket_encryption method in localstack

Best Python code snippet using localstack_python

test_get_bucket_encryption.py

Source:test_get_bucket_encryption.py

...28        # create mock data return when called function get_bucket_encryption29        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:30            mock_method.return_value = expected_bucket_encryption31            # call function test32            actual_bucket_encryption = S3Utils.get_bucket_encryption(33                trace_id, client_s3, bucket_name, aws_account, region_name)34        # check result35        self.assertEqual(36            expected_bucket_encryption['ServerSideEncryptionConfiguration']37            ['Rules'], actual_bucket_encryption)38    def test_get_bucket_encryption_success_response_not_exists_server_side_encryption_configuration_and_rules(self):39        # create mock data return when called function get_bucket_encryption40        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:41            mock_method.return_value = {}42            # call function test43            actual_bucket_encryption = S3Utils.get_bucket_encryption(44                trace_id, client_s3, bucket_name, aws_account, region_name)45        # check result46        expected_bucket_encryption = []47        self.assertEqual(expected_bucket_encryption, actual_bucket_encryption)48    def test_get_bucket_encryption_error_server_side_encryption_configuration_not_found_error(self):49        # create mock throw error when called function get_bucket_encryption50        expected_error_response = copy.deepcopy(DataCommon.ERROR_RESPONSE)51        expected_operation_name = copy.deepcopy(DataCommon.OPERATION_NAME)52        expected_error_response['Error'][53            'Code'] = 'ServerSideEncryptionConfigurationNotFoundError'54        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:55            mock_method.side_effect = ClientError(56                error_response=expected_error_response,57                operation_name=expected_operation_name)58            with patch.object(PmLogAdapter, 'info',59                              return_value=None) as mock_method_info:60                with self.assertRaises(PmError) as exception:61                    # call function test62                    S3Utils.get_bucket_encryption(trace_id, client_s3,63                                                  bucket_name, aws_account,64                                                  region_name)65        # check error66        actual_cause_error = exception.exception.cause_error67        self.assertEqual(expected_error_response['Error'],68                         actual_cause_error.response['Error'])69        self.assertEqual(expected_operation_name,70                         actual_cause_error.operation_name)71        # check write log info72        mock_method_info.assert_any_call('[%s]S3ãã±ããæå·åæå ±ãããã¾ãããï¼%s/%sï¼',73                                         aws_account, region_name, bucket_name)74    def test_get_bucket_encryption_error_access_denied(self):75        # create mock throw error when called function get_bucket_encryption76        expected_error_response = copy.deepcopy(DataCommon.ERROR_RESPONSE)77        expected_operation_name = copy.deepcopy(DataCommon.OPERATION_NAME)78        expected_error_response['Error']['Code'] = 'AccessDenied'79        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:80            mock_method.side_effect = ClientError(81                error_response=expected_error_response,82                operation_name=expected_operation_name)83            with patch.object(PmLogAdapter, 'warning',84                              return_value=None) as mock_method_warning:85                with self.assertRaises(PmError) as exception:86                    # call function test87                    S3Utils.get_bucket_encryption(trace_id, client_s3,88                                                  bucket_name, aws_account,89                                                  region_name)90        # check error91        actual_cause_error = exception.exception.cause_error92        self.assertEqual(expected_error_response['Error'],93                         actual_cause_error.response['Error'])94        self.assertEqual(expected_operation_name,95                         actual_cause_error.operation_name)96        # check write log warning97        mock_method_warning.assert_any_call(98            '[%s] æ¨©éã¨ã©ã¼ã«ããS3ãã±ãããªã¼ã¸ã§ã³æå ±ã®åå¾ã«å¤±æãã¾ãããï¼%s/%sï¼', aws_account,99            region_name, bucket_name)100    def test_get_bucket_encryption_error_method_not_allowed(self):101        # create mock throw error when called function get_bucket_encryption102        expected_error_response = copy.deepcopy(DataCommon.ERROR_RESPONSE)103        expected_operation_name = copy.deepcopy(DataCommon.OPERATION_NAME)104        expected_error_response['Error']['Code'] = 'MethodNotAllowed'105        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:106            mock_method.side_effect = ClientError(107                error_response=expected_error_response,108                operation_name=expected_operation_name)109            with patch.object(PmLogAdapter, 'warning',110                              return_value=None) as mock_method_warning:111                with self.assertRaises(PmError) as exception:112                    # call function test113                    S3Utils.get_bucket_encryption(trace_id, client_s3,114                                                  bucket_name, aws_account,115                                                  region_name)116        # check error117        actual_cause_error = exception.exception.cause_error118        self.assertEqual(expected_error_response['Error'],119                         actual_cause_error.response['Error'])120        self.assertEqual(expected_operation_name,121                         actual_cause_error.operation_name)122        # check write log warning123        mock_method_warning.assert_any_call(124            '[%s] æ¨©éã¨ã©ã¼ã«ããS3ãã±ãããªã¼ã¸ã§ã³æå ±ã®åå¾ã«å¤±æãã¾ãããï¼%s/%sï¼', aws_account,125            region_name, bucket_name)126    def test_get_bucket_encryption_error_other(self):127        # create mock throw error when called function get_bucket_encryption128        expected_error_response = copy.deepcopy(DataCommon.ERROR_RESPONSE)129        expected_operation_name = copy.deepcopy(DataCommon.OPERATION_NAME)130        with patch.object(client_s3, 'get_bucket_encryption') as mock_method:131            mock_method.side_effect = ClientError(132                error_response=expected_error_response,133                operation_name=expected_operation_name)134            with patch.object(PmLogAdapter, 'error',135                              return_value=None) as mock_method_error:136                with self.assertRaises(PmError) as exception:137                    # call function test138                    S3Utils.get_bucket_encryption(trace_id, client_s3,139                                                  bucket_name, aws_account,140                                                  region_name)141        # check error142        actual_cause_error = exception.exception.cause_error143        self.assertEqual(expected_error_response['Error'],144                         actual_cause_error.response['Error'])145        self.assertEqual(expected_operation_name,146                         actual_cause_error.operation_name)147        # check write log error148        mock_method_error.assert_any_call("[%s]S3ãã±ããæå·åæå ±ã®åå¾ã«å¤±æãã¾ãããï¼%s/%sï¼",149                                          aws_account, region_name,...

test_main.py

Source:test_main.py

...19    except ClientError:20        return False, None21def get_encryption_information(bucket_name):22    try:23        s3_client.get_bucket_encryption(Bucket=bucket_name)24        return True25    except ClientError:26        return False27class TestingMain(unittest.TestCase):28    def test_format_two_decimal_points(self):29        self.assertEqual('123.00', format_two_decimal_points(123))30        self.assertEqual('12345.68', format_two_decimal_points(12345.6789))31    def test_get_public_access_information(self):32        stubber.add_response('get_public_access_block', {33            'PublicAccessBlockConfiguration': {34                'RestrictPublicBuckets': True35            }36        })37        with stubber:...

enforce_encryption.py

Source:enforce_encryption.py

...4def lambda_handler(event, context):5  loads = json.loads(event['invokingEvent'])6  bucket_name = loads["configurationItem"]["resourceName"]7  try:8    get_bucket_encryption = s3.get_bucket_encryption(Bucket=bucket_name)9  except Exception:10    put_encryption = s3.put_bucket_encryption(11      Bucket = bucket_name,12      ServerSideEncryptionConfiguration={13        'Rules': [ 14          {15            'ApplyServerSideEncryptionByDefault': { 'SSEAlgorithm': 'AES256' }16          }17        ]18      })...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.