Package com.s3auth.hosts

Source Code of com.s3auth.hosts.DefaultDynamoTest

/**
* Copyright (c) 2012-2014, s3auth.com
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met: 1) Redistributions of source code must retain the above
* copyright notice, this list of conditions and the following
* disclaimer. 2) Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials provided
* with the distribution. 3) Neither the name of the s3auth.com nor
* the names of its contributors may be used to endorse or promote
* products derived from this software without specific prior written
* permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT
* NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
* FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
* THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.s3auth.hosts;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.amazonaws.services.dynamodbv2.model.PutItemRequest;
import com.amazonaws.services.dynamodbv2.model.PutItemResult;
import com.amazonaws.services.dynamodbv2.model.ScanRequest;
import com.amazonaws.services.dynamodbv2.model.ScanResult;
import com.jcabi.aspects.Tv;
import com.jcabi.urn.URN;
import com.jcabi.urn.URNMocker;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import org.hamcrest.MatcherAssert;
import org.hamcrest.Matchers;
import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

/**
* Test case for {@link DefaultDynamo}.
* @author Yegor Bugayenko (yegor@tpc2.com)
* @version $Id$
* @checkstyle ClassDataAbstractionCoupling (500 lines)
*/
public final class DefaultDynamoTest {

    /**
     * DefaultDynamo can load configuration.
     * @throws Exception If there is some problem inside
     */
    @Test
    public void loadsDynamoConfiguration() throws Exception {
        final AmazonDynamoDB amazon = this.amazon();
        final Dynamo dynamo = new DefaultDynamo(
            new Dynamo.Client() {
                @Override
                public AmazonDynamoDB get() {
                    return amazon;
                }
            },
            "table"
        );
        final int size = dynamo.load().size();
        MatcherAssert.assertThat(
            dynamo.load().size(),
            Matchers.is(size)
        );
        dynamo.add(new URN("urn:test:alpha"), new DomainMocker().mock());
        MatcherAssert.assertThat(
            dynamo.load().size(),
            Matchers.is(size + 1)
        );
        dynamo.close();
    }

    /**
     * Create and return an amazon client with 20 random items.
     * @return The client
     */
    private AmazonDynamoDB amazon() {
        final List<Map<String, AttributeValue>> items =
            new LinkedList<Map<String, AttributeValue>>();
        for (int num = 0; num < Tv.TWENTY; ++num) {
            items.add(this.item());
        }
        final AmazonDynamoDB aws =
            Mockito.mock(AmazonDynamoDB.class);
        Mockito.doAnswer(
            new Answer<ScanResult>() {
                @Override
                public ScanResult answer(final InvocationOnMock invocation) {
                    return new ScanResult().withItems(items);
                }
            }
        ).when(aws).scan(Mockito.any(ScanRequest.class));
        Mockito.doAnswer(
            new Answer<PutItemResult>() {
                @Override
                public PutItemResult answer(final InvocationOnMock invocation) {
                    items.add(
                        ((PutItemRequest) invocation.getArguments()[0])
                            .getItem()
                    );
                    return Mockito.mock(PutItemResult.class);
                }
            }
        ).when(aws).putItem(Mockito.any(PutItemRequest.class));
        return aws;
    }

    /**
     * Create and return a random amazon item.
     * @return The client
     */
    private Map<String, AttributeValue> item() {
        final ConcurrentMap<String, AttributeValue> item =
            new ConcurrentHashMap<String, AttributeValue>(0);
        item.put(
            DefaultDynamo.USER,
            new AttributeValue(new URNMocker().mock().toString())
        );
        item.put(
            DefaultDynamo.NAME,
            new AttributeValue(
                String.format(
                    "google-%d.com",
                    Math.abs(new Random().nextInt())
                )
            )
        );
        item.put(DefaultDynamo.KEY, new AttributeValue("aaaaaaaaaaaaaaaa"));
        item.put(
            DefaultDynamo.SECRET,
            new AttributeValue("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
        );
        item.put(DefaultDynamo.REGION, new AttributeValue("s3"));
        return item;
    }

}
TOP

Related Classes of com.s3auth.hosts.DefaultDynamoTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.