format

Author: kessplas

src/test/java/com/amazonaws/encryptionsdk/internal/BlockEncryptionHandlerTest.java

@@ -23,6 +23,12 @@
 import com.amazonaws.encryptionsdk.CryptoResult;
 import com.amazonaws.encryptionsdk.TestUtils;
 import com.amazonaws.encryptionsdk.model.CipherBlockHeaders;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.nio.ByteBuffer;
+import java.nio.charset.StandardCharsets;
+import java.security.SecureRandom;
+import java.util.Collections;
 import javax.crypto.SecretKey;
 import javax.crypto.spec.SecretKeySpec;
 import org.junit.Before;
@@ -34,13 +40,6 @@
 import software.amazon.cryptography.materialproviders.model.CreateRawAesKeyringInput;
 import software.amazon.cryptography.materialproviders.model.MaterialProvidersConfig;
 
-import java.io.ByteArrayInputStream;
-import java.io.ByteArrayOutputStream;
-import java.nio.ByteBuffer;
-import java.nio.charset.StandardCharsets;
-import java.security.SecureRandom;
-import java.util.Collections;
-
 public class BlockEncryptionHandlerTest {
   private final CryptoAlgorithm cryptoAlgorithm_ = TestUtils.DEFAULT_TEST_CRYPTO_ALG;
   private final byte[] messageId_ =
@@ -86,9 +85,9 @@ public void correctIVGenerated() throws Exception {
   }
 
   /**
-   * This isn't a unit test, but it reproduces a bug found in the FrameEncryptionHandler where the stream
-   * would be truncated when the offset is >0.
-   * For robustness' sake, the same test is included against the BlockEncryptionHandler.
+   * This isn't a unit test, but it reproduces a bug found in the FrameEncryptionHandler where the
+   * stream would be truncated when the offset is >0. For the sake of robustness, the same test is
+   * included against the BlockEncryptionHandler.
    *
    * @throws Exception
    */
@@ -100,21 +99,19 @@ public void testStreamTruncation() throws Exception {
     rnd.nextBytes(rawKey);
     SecretKeySpec cryptoKey = new SecretKeySpec(rawKey, "AES");
     MaterialProviders materialProviders =
-      MaterialProviders.builder()
-        .MaterialProvidersConfig(MaterialProvidersConfig.builder().build())
-        .build();
+        MaterialProviders.builder()
+            .MaterialProvidersConfig(MaterialProvidersConfig.builder().build())
+            .build();
     CreateRawAesKeyringInput keyringInput =
-      CreateRawAesKeyringInput.builder()
-        .wrappingKey(ByteBuffer.wrap(cryptoKey.getEncoded()))
-        .keyNamespace("Example")
-        .keyName("RandomKey")
-        .wrappingAlg(AesWrappingAlg.ALG_AES128_GCM_IV12_TAG16)
-        .build();
+        CreateRawAesKeyringInput.builder()
+            .wrappingKey(ByteBuffer.wrap(cryptoKey.getEncoded()))
+            .keyNamespace("Example")
+            .keyName("RandomKey")
+            .wrappingAlg(AesWrappingAlg.ALG_AES128_GCM_IV12_TAG16)
+            .build();
     IKeyring keyring = materialProviders.CreateRawAesKeyring(keyringInput);
     // Use unframed (block) encryption
-    AwsCrypto crypto = AwsCrypto.builder()
-      .withEncryptionFrameSize(0)
-      .build();
+    AwsCrypto crypto = AwsCrypto.builder().withEncryptionFrameSize(0).build();
 
     String testDataString = StringUtils.repeat("Hello, World! ", 5_000);
 
@@ -125,22 +122,22 @@ public void testStreamTruncation() throws Exception {
     // copy some data, starting at the startOffset
     // so the first |startOffset| bytes are 0s
     System.arraycopy(
-      testDataString.getBytes(StandardCharsets.UTF_8),
-      0,
-      inputDataWithOffset,
-      startOffset,
-      dataLength);
+        testDataString.getBytes(StandardCharsets.UTF_8),
+        0,
+        inputDataWithOffset,
+        startOffset,
+        dataLength);
     // decryptData (non-streaming) doesn't know about the offset
     // it will strip out the original 0s
     byte[] expectedOutput = new byte[10_000 - startOffset];
     System.arraycopy(
-      testDataString.getBytes(StandardCharsets.UTF_8), 0, expectedOutput, 0, dataLength);
+        testDataString.getBytes(StandardCharsets.UTF_8), 0, expectedOutput, 0, dataLength);
 
     // Encrypt the data
     byte[] encryptedData;
     try (ByteArrayOutputStream os = new ByteArrayOutputStream()) {
       try (CryptoOutputStream cryptoOutput =
-             crypto.createEncryptingStream(keyring, os, Collections.emptyMap())) {
+          crypto.createEncryptingStream(keyring, os, Collections.emptyMap())) {
         cryptoOutput.write(inputDataWithOffset, startOffset, dataLength);
       }
       encryptedData = os.toByteArray();
@@ -155,7 +152,7 @@ public void testStreamTruncation() throws Exception {
     int decryptedLength = 0;
     byte[] decryptedData = new byte[inputDataWithOffset.length];
     try (ByteArrayInputStream is = new ByteArrayInputStream(encryptedData);
-         CryptoInputStream cryptoInput = crypto.createDecryptingStream(keyring, is)) {
+        CryptoInputStream cryptoInput = crypto.createDecryptingStream(keyring, is)) {
       int offset = startOffset;
       do {
         int bytesRead = cryptoInput.read(decryptedData, offset, decryptedData.length - offset);