/**
* 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;
}
}