diff --git a/jnosql-dynamodb/src/test/java/org/eclipse/jnosql/databases/dynamodb/communication/DynamoDBTestUtils.java b/jnosql-dynamodb/src/test/java/org/eclipse/jnosql/databases/dynamodb/communication/DynamoDBTestUtils.java index 1d0033746..97220dc23 100644 --- a/jnosql-dynamodb/src/test/java/org/eclipse/jnosql/databases/dynamodb/communication/DynamoDBTestUtils.java +++ b/jnosql-dynamodb/src/test/java/org/eclipse/jnosql/databases/dynamodb/communication/DynamoDBTestUtils.java @@ -15,6 +15,7 @@ package org.eclipse.jnosql.databases.dynamodb.communication; import org.eclipse.jnosql.communication.Settings; +import org.eclipse.jnosql.communication.document.DocumentManagerFactory; import org.eclipse.jnosql.communication.keyvalue.BucketManagerFactory; import org.testcontainers.containers.GenericContainer; import org.testcontainers.containers.wait.strategy.Wait; @@ -34,6 +35,12 @@ BucketManagerFactory getBucketManagerFactory() { return configuration.apply(settings); } + DocumentManagerFactory getDocumentManagerFactory() { + Settings settings = getSettings(); + var configuration = new DynamoDBDocumentConfiguration(); + return configuration.apply(settings); + } + Settings getSettings() { dynamodb.start(); return Settings.builder()