org.apache.usergrid.rest.applications.assets.aws.NoAWSCredsRule.java Source code

Introduction

Here is the source code for org.apache.usergrid.rest.applications.assets.aws.NoAWSCredsRule.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.usergrid.rest.applications.assets.aws;

import com.amazonaws.AmazonClientException;
import com.amazonaws.SDKGlobalConfiguration;
import org.apache.usergrid.rest.test.resource.AbstractRestIT;
import org.apache.usergrid.services.exceptions.AwsPropertiesNotFoundException;
import org.junit.Assume;
import org.junit.internal.runners.model.MultipleFailureException;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.Map;
import java.util.Properties;

/**
 * Created in an attempt to mark no aws cred tests as ignored.  Blocked by this issue
 * https://github.com/junit-team/junit/issues/116
 *
 * Until then, simply marks as passed, which is a bit dangerous
 */
public class NoAWSCredsRule extends AbstractRestIT implements TestRule {

    @Autowired
    private Properties properties;

    public Statement apply(final Statement base, final Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                String accessId;
                String secretKey;
                String bucketName;
                try {
                    Map<String, Object> properties = getRemoteTestProperties();
                    //TODO: GREY change this so that it checks for the properties, then if it doesn't have them, mark the tests as ignored.
                    accessId = (String) System.getProperty(SDKGlobalConfiguration.ACCESS_KEY_ENV_VAR);
                    secretKey = (String) System.getProperty(SDKGlobalConfiguration.SECRET_KEY_ENV_VAR);
                    bucketName = (String) properties.get("usergrid.binary.bucketname");

                    if (accessId == null || secretKey == null || bucketName == null) {
                        throw new AwsPropertiesNotFoundException("Access Keys");
                    }
                    base.evaluate();

                } catch (Throwable t) {

                    if (!isMissingCredsException(t)) {
                        throw t;
                    }

                    //do this so our test gets marked as ignored.  Not pretty, but it works
                    Assume.assumeTrue(false);

                }
            }
        };
    }

    private boolean isMissingCredsException(final Throwable t) {

        if (t instanceof AmazonClientException) {

            final AmazonClientException ace = (AmazonClientException) t;

            if (ace.getMessage().contains("could not get aws access key")
                    || ace.getMessage().contains("could not get aws secret key from system properties")) {
                //swallow
                return true;
            }
        }

        if (t instanceof AwsPropertiesNotFoundException) {
            return true;
        }

        /**
         * Handle the multiple failure junit trace
         */
        if (t instanceof MultipleFailureException) {
            for (final Throwable failure : ((MultipleFailureException) t).getFailures()) {
                final boolean isMissingCreds = isMissingCredsException(failure);

                if (isMissingCreds) {
                    return true;
                }
            }
        }
        final Throwable cause = t.getCause();

        if (cause == null) {
            return false;
        }

        return isMissingCredsException(cause);
    }
}