S3BucketTest.java

/**
 * Copyright 2010 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package datameer.awstasks.aws.s3;

import static org.fest.assertions.Assertions.*;

import static org.junit.Assert.*;

import java.io.File;
import java.util.List;

import org.junit.Test;

import awstasks.com.amazonaws.services.s3.AmazonS3;
import awstasks.com.amazonaws.services.s3.model.S3ObjectSummary;

import datameer.awstasks.aws.AbstractAwsIntegrationTest;
import datameer.awstasks.util.Retry;
import datameer.awstasks.util.S3Util;

public class S3BucketTest extends AbstractAwsIntegrationTest {

    public static final String AWS_TEST_BUCKET = "aws.test.bucket";

    @SuppressWarnings("unchecked")
    // retry necessary because of 'eventual consistency' (see
    // http://shlomoswidler.com/2009/12/read-after-write-consistency-in-amazon.html)
    private Retry _retry = Retry.onExceptions(AssertionError.class).withMaxRetries(5).withWaitTime(500);

    private void assertBucketExist(final AmazonS3 s3Service, final String bucketName, final boolean expectedValue) {
        _retry.execute(new Runnable() {
            @Override
            public void run() {
                assertThat(s3Service.doesBucketExist(bucketName)).isEqualTo(expectedValue);
            }
        });
    }

    @Test
    public void testCreateDeleteBucket() throws Exception {
        final AmazonS3 s3Service = _ec2Conf.createS3Service();
        final String bucketName = AWS_TEST_BUCKET;
        removeBucket(s3Service, bucketName);
        assertBucketExist(s3Service, bucketName, false);

        s3Service.createBucket(bucketName);
        assertBucketExist(s3Service, bucketName, true);

        s3Service.deleteBucket(bucketName);
        assertBucketExist(s3Service, bucketName, false);
    }

    private void removeBucket(AmazonS3 s3Service, String bucketName) {
        boolean bucketExists = s3Service.doesBucketExist(bucketName);
        if (bucketExists) {
            List<S3ObjectSummary> objectSummaries = s3Service.listObjects(bucketName).getObjectSummaries();
            for (S3ObjectSummary objectSummary : objectSummaries) {
                s3Service.deleteObject(bucketName, objectSummary.getKey());
            }
            s3Service.deleteBucket(bucketName);
        }
    }

    @Test
    public void testUploadFile_ExistsFile() throws Exception {
        AmazonS3 s3Service = _ec2Conf.createS3Service();
        s3Service.createBucket(AWS_TEST_BUCKET);
        assertBucketExist(s3Service, AWS_TEST_BUCKET, true);
        String remotePath = "/tmp/build.xml";
        S3Util.uploadFile(s3Service, AWS_TEST_BUCKET, new File("build.xml"), remotePath);
        assertTrue(S3Util.existsFile(s3Service, AWS_TEST_BUCKET, remotePath));
        assertFalse(S3Util.existsFile(s3Service, AWS_TEST_BUCKET, remotePath + "/dwefwfe"));
    }
}