原创

DynamoDB クエリの単体テスト

温馨提示:
本文最后更新于 2024年04月12日,已超过 47 天没有更新。若文章内的图片失效(无法正常加载),请留言反馈或直接联系我

この方法があります

    public List<T> queryGSI(final String indexName, final String gsiHashKeyName, final String gsiHashKeyValue,
        final String gsiRangeKeyName, final String gsiRangeKeyValue) {
        Map<String, AttributeValue> eav = new HashMap<>();
        eav.put(":gsiHashKey", new AttributeValue().withS(gsiHashKeyValue));
        eav.put(":gsiRangeKey", new AttributeValue().withS(gsiRangeKeyValue));
        DynamoDBQueryExpression<T> queryExpression = new DynamoDBQueryExpression<T>()
            .withIndexName(indexName)
            .withKeyConditionExpression(gsiHashKeyName + " = :gsiHashKey and " + gsiRangeKeyName + " = :gsiRangeKey")
            .withExpressionAttributeValues(eav);
        try {
            return mapper.query(tableClass, queryExpression);
        } catch (final AmazonClientException e) {
            throw new RuntimeException("Failed to query GSI with hash key and range key", e);
        }
    }

DynamoDBMapper と TableClass が挿入されています。

すべてのアノテーションを使用する DDB レコードクラスがあります。 @DynamoDBIndexHashKey @DynamoDBIndexRangeKey など

私のテストケースではこれを行いました

@Mock
    private DynamoDBMapper mockMapper;

    @Captor
    ArgumentCaptor<DynamoDBQueryExpression<DDBItem>> expressionArgumentCaptor;

        DDBItem testItem = DDBItem.builder()
            .att1("val1")
            .att2("val2")
            .att3("val3")
            .build();

        when(mockMapper.query(eq(DDBItem.class), any(DynamoDBQueryExpression.class))).thenReturn(
            mock(PaginatedQueryList.class, Mockito.withSettings().defaultAnswer(new ForwardsInvocations(singletonList(testItem)))));

        BaseOperations<DDBItem> baseOperations = new BaseOperations<>(mockMapper, DDBItem.class);
        List<DDBItem> result = baseDynamoDBOperations.queryGSI(
           "IndexName",
            "att1",
            "val1",
            att2,
            "val2"
        );
        
        verify(mockMapper, times(1)).query(eq(DDBItem.class), expressionArgumentCaptor.capture());

        DynamoDBQueryExpression<DDBItem> expression = expressionArgumentCaptor.getValue();

        assertEquals(testItem, result.get(0));

これはヌルポインタ例外をスローしています

     when(mockMapper.query(eq(DDBItem.class), any(DynamoDBQueryExpression.class))).thenReturn(
            mock(PaginatedQueryList.class, Mockito.withSettings().defaultAnswer(new ForwardsInvocations(singletonList(testItem)))));

誰かがこの問題の解決策を考えてくれませんか?

正文到此结束
热门推荐
本文目录