/*
 * Copyright 2010-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * 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://aws.amazon.com/apache2.0
 *
 * This file 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 com.amazonaws.services.s3;

import androidx.test.core.app.ApplicationProvider;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import com.amazonaws.AmazonServiceException;

import org.apache.commons.io.IOUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.InputStream;

/**
 * Integration tests for S3 bucket policy operations.
 */
public class BucketPolicyIntegrationTest extends S3IntegrationTestBase {

    /** Name of the bucket created by this test */
    private static final String BUCKET_NAME = "android-sdk-bucket-policy-integ-test-" + System.currentTimeMillis();

    /** Path to the sample policy for this test */
    private static final String POLICY_FILE = "samplePolicy.json";

    /** Create bucket and wait for creation */
    @Before
    public void setUpBucket() throws Exception {
        s3.createBucket(BUCKET_NAME);
        S3IntegrationTestBase.waitForBucketCreation(BUCKET_NAME);
    }

    /** Release all allocated resources */
    @After
    public void tearDown() {
        try {
            super.deleteBucketAndAllContents(BUCKET_NAME);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    /** Tests that we can get/put/delete bucket policies. */
    @Test
    public void testBucketPolicies() throws Exception {
        InputStream policyInputStream = ApplicationProvider.getApplicationContext().getResources().getAssets().open(POLICY_FILE);
        String policyText = IOUtils.toString(policyInputStream);
        policyText = replace(policyText, "@BUCKET_NAME@", BUCKET_NAME);

        // Verify that no policy exists yet
        assertNull(s3.getBucketPolicy(BUCKET_NAME).getPolicyText());

        // Upload a new bucket policy
        s3.setBucketPolicy(BUCKET_NAME, policyText);

        // Try to retrieve it - then set what we get back to make sure
        // we correctly parsed the policy text
        String retrievedPolicyText = s3.getBucketPolicy(BUCKET_NAME).getPolicyText();
        assertTrue(retrievedPolicyText.indexOf(BUCKET_NAME) != -1);
        s3.setBucketPolicy(BUCKET_NAME, retrievedPolicyText);

        // Delete it - and verify it's gone
        s3.deleteBucketPolicy(BUCKET_NAME);
        assertNull(s3.getBucketPolicy(BUCKET_NAME).getPolicyText());

        // Try to get the policy for a bucket we don't own to test error
        // handling
        try {
            s3.getBucketPolicy("bucket");
            fail("Expected AmazonServiceException");
        } catch (AmazonServiceException ase) {
            assertNotNull(ase.getRequestId());
            assertNotNull(ase.getMessage());
        }
    }

    protected String replace(String source, String token, String value) {
        int startIndexOfToken = source.indexOf(token);
        int endIndexOfToken = startIndexOfToken + token.length();

        return source.substring(0, startIndexOfToken) + value + source.substring(endIndexOfToken);
    }
}