m

Author: rishav-karanjit

DynamoDbEncryption/runtimes/java/src/test/sdkv2/com/amazonaws/services/dynamodbv2/datamodeling/sdkv2/encryption/providers/DirectKmsMaterialProviderTest.java

@@ -64,10 +64,7 @@ public void setUp() {
 
   @Test
   public void simple() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
 
     EncryptionMaterials eMat = prov.getEncryptionMaterials(ctx);
     SecretKey encryptionKey = eMat.getEncryptionKey();
@@ -96,10 +93,7 @@ public void simple() {
 
   @Test
   public void simpleWithKmsEc() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
 
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put("hk", AttributeValue.builder().s("HashKeyValue").build());
@@ -136,10 +130,7 @@ public void simpleWithKmsEc() {
 
   @Test
   public void simpleWithKmsEc2() throws GeneralSecurityException {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
 
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put("hk", AttributeValue.builder().n("10").build());
@@ -176,10 +167,7 @@ public void simpleWithKmsEc2() throws GeneralSecurityException {
 
   @Test
   public void simpleWithKmsEc3() throws GeneralSecurityException {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
 
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put(
@@ -243,10 +231,7 @@ public void simpleWithKmsEc3() throws GeneralSecurityException {
 
   @Test
   public void randomEnvelopeKeys() throws GeneralSecurityException {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
 
     EncryptionMaterials eMat = prov.getEncryptionMaterials(ctx);
     SecretKey encryptionKey = eMat.getEncryptionKey();
@@ -264,10 +249,7 @@ public void randomEnvelopeKeys() throws GeneralSecurityException {
   @Test
   public void testRefresh() {
     // This does nothing, make sure we don't throw and exception.
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
     prov.refresh();
   }
 
@@ -558,10 +540,7 @@ public void constructorNullKmsClient() {
 
   @Test(expectedExceptions = UnsupportedOperationException.class)
   public void unsupportedHashKeyType() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put("hk", AttributeValue.builder().bool(true).build());
     EncryptionContext boolCtx = new EncryptionContext.Builder()
@@ -573,10 +552,7 @@ public void unsupportedHashKeyType() {
 
   @Test(expectedExceptions = UnsupportedOperationException.class)
   public void unsupportedRangeKeyType() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put("hk", AttributeValue.builder().s("HashKeyValue").build());
     attrVals.put("rk", AttributeValue.builder().bool(true).build());
@@ -590,10 +566,7 @@ public void unsupportedRangeKeyType() {
 
   @Test(expectedExceptions = NullPointerException.class)
   public void hashKeyMissingFromAttributeValues() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
     EncryptionContext missingCtx = new EncryptionContext.Builder()
       .hashKeyName("hk")
       .attributeValues(Collections.emptyMap())
@@ -603,10 +576,7 @@ public void hashKeyMissingFromAttributeValues() {
 
   @Test(expectedExceptions = NullPointerException.class)
   public void rangeKeyMissingFromAttributeValues() {
-    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(
-      kms,
-      keyId
-    );
+    DirectKmsMaterialProvider prov = new DirectKmsMaterialProvider(kms, keyId);
     Map<String, AttributeValue> attrVals = new HashMap<>();
     attrVals.put("hk", AttributeValue.builder().s("HashKeyValue").build());
     EncryptionContext missingCtx = new EncryptionContext.Builder()

Examples/runtimes/java/DDBECwithSDKV2/src/main/java/AwsKmsMultiRegionKey.java

@@ -104,8 +104,10 @@ public static void encryptRecord(
     record.put(IGNORED_FIELD_NAME, AttributeValue.builder().s("alone").build());
 
     // Set up encryptor with first region's KMS key
-    final DirectKmsMaterialProvider cmpEncrypt =
-      new DirectKmsMaterialProvider(kmsEncrypt, cmkArnEncrypt);
+    final DirectKmsMaterialProvider cmpEncrypt = new DirectKmsMaterialProvider(
+      kmsEncrypt,
+      cmkArnEncrypt
+    );
     final DynamoDBEncryptor encryptor = DynamoDBEncryptor.getInstance(
       cmpEncrypt
     );
@@ -149,8 +151,10 @@ public static void encryptRecord(
     );
 
     // Set up decryptor using the replica key from second region
-    final DirectKmsMaterialProvider cmpDecrypt =
-      new DirectKmsMaterialProvider(kmsDecrypt, cmkArnDecrypt);
+    final DirectKmsMaterialProvider cmpDecrypt = new DirectKmsMaterialProvider(
+      kmsDecrypt,
+      cmkArnDecrypt
+    );
     final DynamoDBEncryptor decryptor = DynamoDBEncryptor.getInstance(
       cmpDecrypt
     );