40
40
import software .amazon .awssdk .enhanced .dynamodb .AttributeConverterProvider ;
41
41
import software .amazon .awssdk .enhanced .dynamodb .DefaultAttributeConverterProvider ;
42
42
import software .amazon .awssdk .enhanced .dynamodb .EnhancedType ;
43
- import software .amazon .awssdk .enhanced .dynamodb .internal .converter .ChainConverterProvider ;
44
- import software .amazon .awssdk .enhanced .dynamodb .internal .converter .attribute .BigDecimalAttributeConverter ;
45
- import software .amazon .awssdk .enhanced .dynamodb .internal .converter .attribute .BigIntegerAttributeConverter ;
46
- import software .amazon .awssdk .enhanced .dynamodb .internal .converter .attribute .LocalDateAttributeConverter ;
47
- import software .amazon .awssdk .enhanced .dynamodb .internal .converter .attribute .StringAttributeConverter ;
48
43
import software .amazon .awssdk .enhanced .dynamodb .internal .document .DefaultEnhancedDocument ;
49
44
import software .amazon .awssdk .services .dynamodb .model .AttributeValue ;
50
45
import software .amazon .awssdk .utils .Pair ;
@@ -271,21 +266,18 @@ private static Set<String> getStringSet(String... strings) {
271
266
return Stream .of (strings ).collect (Collectors .toCollection (LinkedHashSet ::new ));
272
267
}
273
268
274
-
275
269
private static DefaultEnhancedDocument .DefaultBuilder documentBuilder () {
276
270
DefaultEnhancedDocument .DefaultBuilder defaultBuilder = DefaultEnhancedDocument .builder ();
277
271
defaultBuilder .addAttributeConverterProvider (AttributeConverterProvider .defaultProvider ());
278
272
return defaultBuilder ;
279
273
}
280
274
281
275
private static AttributeStringValueMap map () {
282
- return new AttributeStringValueMap ();
283
-
276
+ return new AttributeStringValueMap ();\
284
277
}
285
278
286
279
public static void validateAttributeValueMapAndDocument (AttributeStringValueMap attributeStringValueMap ,
287
280
DefaultEnhancedDocument enhancedDocument ) {
288
-
289
281
// assert for keys in Document
290
282
assertThat (attributeStringValueMap .getAttributeValueMap ().keySet ()).isEqualTo (enhancedDocument .asMap ().keySet ());
291
283
@@ -298,7 +290,6 @@ public static void validateAttributeValueMapAndDocument(AttributeStringValueMap
298
290
);
299
291
}
300
292
301
-
302
293
@ ParameterizedTest
303
294
@ MethodSource ("attributeValueMapsCorrespondingDocuments" )
304
295
void validate_BuilderMethodsOfDefaultDocument (AttributeStringValueMap expectedMap ,
@@ -308,9 +299,6 @@ void validate_BuilderMethodsOfDefaultDocument(AttributeStringValueMap expectedMa
308
299
* The builder method internally creates a AttributeValueMap which is saved to the ddb, if this matches then
309
300
* the document is as expected
310
301
*/
311
-
312
- System .out .println ("enhancedDocument " +enhancedDocument .toJson ());
313
-
314
302
assertThat (enhancedDocument .getAttributeValueMap ()).isEqualTo (expectedMap .getAttributeValueMap ());
315
303
System .out .println ("enhancedDocument amp here " +enhancedDocument .getAttributeValueMap ());
316
304
}
@@ -332,10 +320,9 @@ void validate_GetterMethodsOfDefaultDocument(AttributeStringValueMap expectedMap
332
320
333
321
@ Test
334
322
void copyCreatedFromToBuilder (){
335
-
336
-
337
- DefaultEnhancedDocument originalDoc = (DefaultEnhancedDocument ) documentBuilder ().add (SIMPLE_STRING_KEY , SIMPLE_STRING )
338
- .build ();
323
+ DefaultEnhancedDocument originalDoc = (DefaultEnhancedDocument ) documentBuilder ()
324
+ .add (SIMPLE_STRING_KEY , SIMPLE_STRING )
325
+ .build ();
339
326
DefaultEnhancedDocument copiedDoc = (DefaultEnhancedDocument ) originalDoc .toBuilder ().build ();
340
327
DefaultEnhancedDocument copyAndAlter =
341
328
(DefaultEnhancedDocument ) originalDoc .toBuilder ().addString ("keyOne" , "valueOne" ).build ();
@@ -345,8 +332,6 @@ void copyCreatedFromToBuilder(){
345
332
assertThat (copyAndAlter .getString (SIMPLE_STRING_KEY )).isEqualTo (SIMPLE_STRING );
346
333
assertThat (copyAndAlter .getString ("keyOne" )).isEqualTo ("valueOne" );
347
334
assertThat (originalDoc ).isEqualTo (copiedDoc );
348
-
349
-
350
335
}
351
336
352
337
@ Test
@@ -369,7 +354,6 @@ void nullDocumentGet(){
369
354
assertThat (document .isNull ("nullAttribute" )).isTrue ();
370
355
}
371
356
372
-
373
357
@ Test
374
358
void multipleGetterForDocument (){
375
359
@@ -391,7 +375,6 @@ void multipleGetterForDocument(){
391
375
getStringSet (STRINGS_ARRAY )).build ())
392
376
.build ();
393
377
394
-
395
378
assertThat (document .getString (SIMPLE_STRING_KEY )).isEqualTo (SIMPLE_STRING );
396
379
assertThat (document .getSdkNumber (SIMPLE_NUMBER_KEY ).intValue ()).isEqualTo (1 );
397
380
@@ -419,7 +402,6 @@ void multipleGetterForDocument(){
419
402
assertThat (document .getTypeOf ("nullKey" )).isNull ();
420
403
}
421
404
422
-
423
405
static class AttributeStringValueMap {
424
406
Map <String , AttributeValue > attributeValueMap = new LinkedHashMap <>();
425
407
@@ -432,5 +414,4 @@ AttributeStringValueMap withKeyValue(String key, AttributeValue value) {
432
414
return this ;
433
415
}
434
416
}
435
-
436
417
}
0 commit comments