def test_object_not_exists(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.head_object = MagicMock(side_effect=Exception) self.assertFalse( gw.object_exists(self.TEST_BUCKET, self.TEST_ACCESS_KEY_ID)) gw.client.head_object.assert_called()
def test_copy(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.copy = MagicMock(return_value=None) gw.copy(self.TEST_BUCKET, self.TEST_FILE, self.TEST_BUCKET, f"{self.TEST_FILE}_COPY") gw.client.copy.assert_called()
def test_object_exists(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.head_object = MagicMock(return_value=None) self.assertTrue( gw.object_exists(self.TEST_BUCKET, self.TEST_ACCESS_KEY_ID)) gw.client.head_object.assert_called()
def test_put_object(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.put_object = MagicMock(return_value=None) gw.put_object(self.TEST_BUCKET, self.TEST_ACCESS_KEY_ID, self.TEST_ACCESS_KEY_DATA) gw.client.put_object.assert_called()
def test_download_file(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.head_object.return_value = {"ContentLength": 100} gw.client.download_file = MagicMock(return_value=None) gw.download_file(self.TEST_BUCKET, self.TEST_FILE, self.TEST_LOCAL_FILE) gw.client.download_file.assert_called()
def __init__( self, region: str = "us-west-1", access_key_id: Optional[str] = None, access_key_data: Optional[str] = None, config: Optional[Dict[str, Any]] = None, ) -> None: self.s3_gateway = S3Gateway(region, access_key_id, access_key_data, config)
def test_list_object2(self, BotoClient): test_page_content_key1 = "test-page-content-key1" test_page_content_key2 = "test-page-content-key2" client_return_response = [{ "Contents": [ { "Key": test_page_content_key1 }, { "Key": test_page_content_key2 }, ], }] gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.get_paginator("list_objects_v2").paginate = MagicMock( return_value=client_return_response) key_list = gw.list_object2(self.TEST_BUCKET, self.TEST_ACCESS_KEY_ID) expected_key_list = [ test_page_content_key1, test_page_content_key2, ] self.assertEqual(key_list, expected_key_list) gw.client.get_paginator("list_object_v2").paginate.assert_called()
def test_delete_object(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.delete_object = MagicMock(return_value=None) gw.delete_object(self.TEST_BUCKET, self.TEST_FILE) gw.client.delete_object.assert_called()
def test_upload_file(self, BotoClient, mock_getsize): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.upload_file = MagicMock(return_value=None) gw.upload_file(self.TEST_LOCAL_FILE, self.TEST_BUCKET, self.TEST_FILE) gw.client.upload_file.assert_called()
def test_create_bucket(self, BotoClient): gw = S3Gateway(self.REGION) gw.client = BotoClient() gw.client.create_bucket = MagicMock(return_value=None) gw.create_bucket(self.TEST_BUCKET) gw.client.create_bucket.assert_called()