1 /*
2 * Copyright (c) 2012-2023, jcabi.com
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met: 1) Redistributions of source code must retain the above
8 * copyright notice, this list of conditions and the following
9 * disclaimer. 2) Redistributions in binary form must reproduce the above
10 * copyright notice, this list of conditions and the following
11 * disclaimer in the documentation and/or other materials provided
12 * with the distribution. 3) Neither the name of the jcabi.com nor
13 * the names of its contributors may be used to endorse or promote
14 * products derived from this software without specific prior written
15 * permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT
19 * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
20 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
21 * THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
22 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
24 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
26 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
28 * OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30 package com.jcabi.dynamo;
31
32 import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
33 import com.amazonaws.services.dynamodbv2.model.AttributeValue;
34 import com.amazonaws.services.dynamodbv2.model.ConsumedCapacity;
35 import com.amazonaws.services.dynamodbv2.model.QueryRequest;
36 import com.amazonaws.services.dynamodbv2.model.QueryResult;
37 import com.google.common.collect.ImmutableMap;
38 import java.util.ArrayList;
39 import java.util.Collections;
40 import org.hamcrest.MatcherAssert;
41 import org.hamcrest.Matchers;
42 import org.junit.jupiter.api.Test;
43 import org.mockito.Mockito;
44
45 /**
46 * Test case for {@link QueryValve}.
47 * @since 0.1
48 */
49 final class QueryValveTest {
50
51 @Test
52 @SuppressWarnings("unchecked")
53 void fetchesData() throws Exception {
54 final Valve valve = new QueryValve();
55 final Credentials credentials = Mockito.mock(Credentials.class);
56 final ImmutableMap<String, AttributeValue> item =
57 new ImmutableMap.Builder<String, AttributeValue>()
58 .build();
59 final AmazonDynamoDB aws = Mockito.mock(AmazonDynamoDB.class);
60 Mockito.doReturn(aws).when(credentials).aws();
61 Mockito.doReturn(
62 new QueryResult()
63 .withItems(
64 Collections.singletonList(item)
65 )
66 .withConsumedCapacity(
67 new ConsumedCapacity().withCapacityUnits(1.0d)
68 )
69 ).when(aws).query(Mockito.any(QueryRequest.class));
70 final Dosage dosage = valve.fetch(
71 credentials, "table",
72 new Conditions(), new ArrayList<>(0)
73 );
74 MatcherAssert.assertThat(dosage.hasNext(), Matchers.is(false));
75 MatcherAssert.assertThat(dosage.items(), Matchers.hasItem(item));
76 }
77
78 }