This is an automated email from the ASF dual-hosted git repository.

adoroszlai pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/ozone.git


The following commit(s) were added to refs/heads/master by this push:
     new 6a9e8b148f HDDS-12348. Reuse `TestDataUtil.createKey` method (#7971)
6a9e8b148f is described below

commit 6a9e8b148fc1c2749187a5236db71c485f68ec51
Author: Chia-Chuan Yu <[email protected]>
AuthorDate: Fri Feb 28 23:05:07 2025 +0800

    HDDS-12348. Reuse `TestDataUtil.createKey` method (#7971)
---
 .../fs/ozone/TestOzoneFSWithObjectStoreCreate.java | 23 +++++++-------
 .../java/org/apache/hadoop/ozone/TestDataUtil.java | 35 ++++++++++++++++++---
 .../apache/hadoop/ozone/om/TestBucketOwner.java    | 19 +++---------
 .../org/apache/hadoop/ozone/om/TestListKeys.java   | 25 ++++++---------
 .../hadoop/ozone/om/TestListKeysWithFSO.java       | 31 ++++++++-----------
 .../hadoop/ozone/om/TestObjectStoreWithFSO.java    | 36 ++++++++--------------
 .../hadoop/ozone/om/TestRecursiveAclWithFSO.java   | 15 ++-------
 .../om/snapshot/TestOmSnapshotFileSystem.java      | 14 ++++-----
 8 files changed, 92 insertions(+), 106 deletions(-)

diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/fs/ozone/TestOzoneFSWithObjectStoreCreate.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/fs/ozone/TestOzoneFSWithObjectStoreCreate.java
index 8f1516c2e2..f72b4c9f4d 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/fs/ozone/TestOzoneFSWithObjectStoreCreate.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/fs/ozone/TestOzoneFSWithObjectStoreCreate.java
@@ -21,6 +21,7 @@
 import static org.apache.hadoop.ozone.OzoneConsts.ETAG;
 import static org.apache.hadoop.ozone.OzoneConsts.MD5_HASH;
 import static org.apache.hadoop.ozone.OzoneConsts.OZONE_URI_SCHEME;
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static 
org.apache.hadoop.ozone.om.exceptions.OMException.ResultCodes.NOT_A_FILE;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -355,9 +356,9 @@ public void testListKeysWithNotNormalizedPath() throws 
Exception {
     byte[] input = new byte[length];
     Arrays.fill(input, (byte)96);
 
-    createKey(ozoneBucket, key1, 10, input);
-    createKey(ozoneBucket, key2, 10, input);
-    createKey(ozoneBucket, key3, 10, input);
+    createAndAssertKey(ozoneBucket, key1, 10, input);
+    createAndAssertKey(ozoneBucket, key2, 10, input);
+    createAndAssertKey(ozoneBucket, key3, 10, input);
 
     // Iterator with key name as prefix.
 
@@ -402,18 +403,18 @@ private void checkKeyList(Iterator<? extends OzoneKey > 
ozoneKeyIterator,
     assertEquals(keys, outputKeys);
   }
 
-  private void createKey(OzoneBucket ozoneBucket, String key, int length,
-      byte[] input)
+  private void createAndAssertKey(OzoneBucket ozoneBucket, String key, int 
length, byte[] input)
       throws Exception {
+    
+    createKey(ozoneBucket, key, input);
+    // Read the key with given key name.
+    readKey(ozoneBucket, key, length, input);
 
-    OzoneOutputStream ozoneOutputStream =
-        ozoneBucket.createKey(key, length);
+  }
 
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
+  private void readKey(OzoneBucket ozoneBucket, String key, int length, byte[] 
input)
+      throws Exception {
 
-    // Read the key with given key name.
     OzoneInputStream ozoneInputStream = ozoneBucket.readKey(key);
     byte[] read = new byte[length];
     ozoneInputStream.read(read, 0, length);
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/TestDataUtil.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/TestDataUtil.java
index 40447d054d..643e996087 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/TestDataUtil.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/TestDataUtil.java
@@ -107,28 +107,53 @@ public static OzoneVolume createVolume(OzoneClient client,
 
   public static void createKey(OzoneBucket bucket, String keyName,
                                String content) throws IOException {
+    createKey(bucket, keyName, ReplicationFactor.ONE,
+        ReplicationType.RATIS, content.getBytes(UTF_8));
+  }
+
+  public static void createKey(OzoneBucket bucket, String keyName,
+                               byte[] content) throws IOException {
     createKey(bucket, keyName, ReplicationFactor.ONE,
         ReplicationType.RATIS, content);
+
   }
 
   public static void createKey(OzoneBucket bucket, String keyName,
-      ReplicationFactor repFactor, ReplicationType repType, String content)
+                               ReplicationFactor repFactor, ReplicationType 
repType, byte[] content)
       throws IOException {
     ReplicationConfig repConfig = ReplicationConfig
         .fromTypeAndFactor(repType, repFactor);
-    createKey(bucket, keyName, repConfig, content);
+    try (OutputStream stream = bucket
+        .createKey(keyName, content.length, repConfig,
+            new HashMap<>())) {
+      stream.write(content);
+    }
   }
 
   public static void createKey(OzoneBucket bucket, String keyName,
-      ReplicationConfig repConfig, String content)
+                               ReplicationConfig repConfig, byte[] content)
       throws IOException {
     try (OutputStream stream = bucket
-        .createKey(keyName, content.length(), repConfig,
+        .createKey(keyName, content.length, repConfig,
             new HashMap<>())) {
-      stream.write(content.getBytes(UTF_8));
+      stream.write(content);
     }
   }
 
+  public static void createKey(OzoneBucket bucket, String keyName,
+      ReplicationFactor repFactor, ReplicationType repType, String content)
+      throws IOException {
+    ReplicationConfig repConfig = ReplicationConfig
+        .fromTypeAndFactor(repType, repFactor);
+    createKey(bucket, keyName, repConfig, content.getBytes(UTF_8));
+  }
+
+  public static void createKey(OzoneBucket bucket, String keyName,
+      ReplicationConfig repConfig, String content)
+      throws IOException {
+    createKey(bucket, keyName, repConfig, content.getBytes(UTF_8));
+  }
+
   public static void createKey(OzoneBucket bucket, String keyName,
       ReplicationFactor repFactor, ReplicationType repType,
       ByteBuffer data) throws IOException {
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestBucketOwner.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestBucketOwner.java
index 9a950aa452..8f682a31e4 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestBucketOwner.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestBucketOwner.java
@@ -18,6 +18,7 @@
 package org.apache.hadoop.ozone.om;
 
 import static org.apache.hadoop.ozone.OzoneAcl.AclScope.DEFAULT;
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static 
org.apache.hadoop.ozone.security.acl.IAccessAuthorizer.ACLIdentityType.USER;
 import static org.apache.hadoop.ozone.security.acl.OzoneObj.StoreType.OZONE;
 import static org.junit.jupiter.api.Assertions.assertThrows;
@@ -32,7 +33,6 @@
 import org.apache.hadoop.ozone.client.OzoneBucket;
 import org.apache.hadoop.ozone.client.OzoneClient;
 import org.apache.hadoop.ozone.client.OzoneVolume;
-import org.apache.hadoop.ozone.client.io.OzoneOutputStream;
 import org.apache.hadoop.ozone.client.protocol.ClientProtocol;
 import org.apache.hadoop.ozone.security.acl.IAccessAuthorizer;
 import org.apache.hadoop.ozone.security.acl.OzoneObj;
@@ -92,8 +92,8 @@ public void testBucketOwner() throws Exception {
           .getVolume(VOLUME_NAME);
       OzoneBucket ozoneBucket = volume.getBucket("bucket1");
       //Key Create
-      createKey(ozoneBucket, "key1", 10, new byte[10]);
-      createKey(ozoneBucket, "key2", 10, new byte[10]);
+      createKey(ozoneBucket, "key1", new byte[10]);
+      createKey(ozoneBucket, "key2", new byte[10]);
       //Key Delete
       ozoneBucket.deleteKey("key1");
       //Bucket Delete
@@ -118,7 +118,7 @@ public void testNonBucketNonVolumeOwner() throws Exception {
       assertThrows(Exception.class, () -> {
         OzoneVolume volume = client.getObjectStore().getVolume(VOLUME_NAME);
         OzoneBucket ozoneBucket = volume.getBucket("bucket1");
-        createKey(ozoneBucket, "key3", 10, new byte[10]);
+        createKey(ozoneBucket, "key3", new byte[10]);
       }, "Create key as non-volume and non-bucket owner should fail");
     }
     //Key Delete - should fail
@@ -173,8 +173,7 @@ public void testVolumeOwner() throws Exception {
     try (OzoneClient client = cluster().newClient()) {
       OzoneVolume volume = client.getObjectStore().getVolume(VOLUME_NAME);
       OzoneBucket ozoneBucket = volume.getBucket("bucket1");
-      //Key Create
-      createKey(ozoneBucket, "key2", 10, new byte[10]);
+      createKey(ozoneBucket, "key2", new byte[10]);
       //Key Delete
       ozoneBucket.deleteKey("key2");
       //List Keys
@@ -208,12 +207,4 @@ private static void setVolumeAcl(ObjectStore store, String 
volumeName,
         .setResType(OzoneObj.ResourceType.VOLUME).setStoreType(OZONE).build();
     assertTrue(store.setAcl(obj, OzoneAcl.parseAcls(aclString)));
   }
-
-  private void createKey(OzoneBucket ozoneBucket, String key, int length,
-       byte[] input) throws Exception {
-    OzoneOutputStream ozoneOutputStream = ozoneBucket.createKey(key, length);
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
-  }
 }
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeys.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeys.java
index 69cc07a8b2..7b2404e51f 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeys.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeys.java
@@ -20,6 +20,7 @@
 import static com.google.common.collect.Lists.newLinkedList;
 import static 
org.apache.hadoop.ozone.OzoneConfigKeys.OZONE_CLIENT_LIST_CACHE_SIZE;
 import static 
org.apache.hadoop.ozone.OzoneConfigKeys.OZONE_FS_ITERATE_BATCH_SIZE;
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.params.provider.Arguments.of;
 
@@ -33,6 +34,8 @@
 import java.util.Optional;
 import java.util.stream.Stream;
 import org.apache.hadoop.hdds.client.ReplicationConfig;
+import org.apache.hadoop.hdds.client.ReplicationFactor;
+import org.apache.hadoop.hdds.client.ReplicationType;
 import org.apache.hadoop.hdds.conf.OzoneConfiguration;
 import org.apache.hadoop.hdds.utils.IOUtils;
 import org.apache.hadoop.ozone.TestDataUtil;
@@ -41,7 +44,6 @@
 import org.apache.hadoop.ozone.client.OzoneClientFactory;
 import org.apache.hadoop.ozone.client.OzoneKey;
 import org.apache.hadoop.ozone.client.io.OzoneInputStream;
-import org.apache.hadoop.ozone.client.io.OzoneOutputStream;
 import org.apache.hadoop.ozone.om.helpers.BucketLayout;
 import org.apache.ozone.test.NonHATests;
 import org.junit.jupiter.api.AfterAll;
@@ -147,7 +149,7 @@ private static void buildNameSpaceTree(OzoneBucket 
ozoneBucket)
     keys.add("a1/b3/e2/e21.tx");
     keys.add("a1/b3/e3/e31.tx");
 
-    createKeys(ozoneBucket, keys);
+    createAndAssertKeys(ozoneBucket, keys);
 
     ozoneBucket.createDirectory("a1/b4/");
   }
@@ -369,27 +371,20 @@ private void checkKeyShallowList(String keyPrefix, String 
startKey,
     assertEquals(keys, outputKeysList);
   }
 
-  private static void createKeys(OzoneBucket ozoneBucket, List<String> keys)
+  private static void createAndAssertKeys(OzoneBucket ozoneBucket, 
List<String> keys)
       throws Exception {
     int length = 10;
     byte[] input = new byte[length];
     Arrays.fill(input, (byte) 96);
     for (String key : keys) {
-      createKey(ozoneBucket, key, 10, input);
+      createKey(ozoneBucket, key, ReplicationFactor.THREE, 
ReplicationType.RATIS, input);
+      // Read the key with given key name.
+      readkey(ozoneBucket, key, length, input);
     }
   }
 
-  private static void createKey(OzoneBucket ozoneBucket, String key, int 
length,
-      byte[] input) throws Exception {
-
-    OzoneOutputStream ozoneOutputStream =
-        ozoneBucket.createKey(key, length);
-
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
-
-    // Read the key with given key name.
+  private static void readkey(OzoneBucket ozoneBucket, String key, int length, 
byte[] input)
+      throws Exception {
     OzoneInputStream ozoneInputStream = ozoneBucket.readKey(key);
     byte[] read = new byte[length];
     ozoneInputStream.read(read, 0, length);
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeysWithFSO.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeysWithFSO.java
index 56d4cceb1a..6aa60365a7 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeysWithFSO.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestListKeysWithFSO.java
@@ -19,6 +19,7 @@
 
 import static 
org.apache.hadoop.ozone.OzoneConfigKeys.OZONE_CLIENT_LIST_CACHE_SIZE;
 import static 
org.apache.hadoop.ozone.OzoneConfigKeys.OZONE_FS_ITERATE_BATCH_SIZE;
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 import java.nio.charset.StandardCharsets;
@@ -30,6 +31,8 @@
 import java.util.Optional;
 import org.apache.commons.lang3.RandomStringUtils;
 import org.apache.hadoop.hdds.client.ReplicationConfig;
+import org.apache.hadoop.hdds.client.ReplicationFactor;
+import org.apache.hadoop.hdds.client.ReplicationType;
 import org.apache.hadoop.hdds.conf.OzoneConfiguration;
 import org.apache.hadoop.hdds.protocol.StorageType;
 import org.apache.hadoop.hdds.utils.IOUtils;
@@ -41,7 +44,6 @@
 import org.apache.hadoop.ozone.client.OzoneKey;
 import org.apache.hadoop.ozone.client.OzoneVolume;
 import org.apache.hadoop.ozone.client.io.OzoneInputStream;
-import org.apache.hadoop.ozone.client.io.OzoneOutputStream;
 import org.apache.hadoop.ozone.om.helpers.BucketLayout;
 import org.apache.ozone.test.NonHATests;
 import org.junit.jupiter.api.AfterAll;
@@ -548,7 +550,7 @@ private static void buildNameSpaceTree(OzoneBucket 
ozoneBucket)
     keys.add("/a1/b3/e2/e21.tx");
     keys.add("/a1/b3/e3/e31.tx");
 
-    createKeys(ozoneBucket, keys);
+    createAndAssertKeys(ozoneBucket, keys);
   }
 
   private static void buildNameSpaceTree2(OzoneBucket ozoneBucket)
@@ -575,7 +577,7 @@ private static void buildNameSpaceTree2(OzoneBucket 
ozoneBucket)
 
     keys.add("/dir1/dir2/dir3/d11.tx");
 
-    createKeys(ozoneBucket, keys);
+    createAndAssertKeys(ozoneBucket, keys);
   }
 
 
@@ -641,33 +643,26 @@ private void checkKeyShallowList(String keyPrefix, String 
startKey,
     checkKeyList(keyPrefix, startKey, keys, fsoBucket, true);
   }
 
-  private static void createKeys(OzoneBucket ozoneBucket, List<String> keys)
+  private static void createAndAssertKeys(OzoneBucket ozoneBucket, 
List<String> keys)
       throws Exception {
     int length = 10;
     byte[] input = new byte[length];
     Arrays.fill(input, (byte) 96);
     for (String key : keys) {
-      createKey(ozoneBucket, key, 10, input);
+      createKey(ozoneBucket, key, ReplicationFactor.THREE,
+          ReplicationType.RATIS, input);
+      // Read the key with given key name.
+      readkey(ozoneBucket, key, length, input);
     }
   }
 
-  private static void createKey(OzoneBucket ozoneBucket, String key, int 
length,
-      byte[] input) throws Exception {
-
-    OzoneOutputStream ozoneOutputStream =
-        ozoneBucket.createKey(key, length);
-
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
-
-    // Read the key with given key name.
+  private static void readkey(OzoneBucket ozoneBucket, String key, int length, 
byte[] input)
+      throws Exception {
     OzoneInputStream ozoneInputStream = ozoneBucket.readKey(key);
     byte[] read = new byte[length];
     ozoneInputStream.read(read, 0, length);
     ozoneInputStream.close();
 
-    assertEquals(new String(input, StandardCharsets.UTF_8),
-        new String(read, StandardCharsets.UTF_8));
+    assertEquals(new String(input, StandardCharsets.UTF_8), new String(read, 
StandardCharsets.UTF_8));
   }
 }
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestObjectStoreWithFSO.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestObjectStoreWithFSO.java
index b4d5a230a9..18309fb9b6 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestObjectStoreWithFSO.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestObjectStoreWithFSO.java
@@ -22,6 +22,7 @@
 import static 
org.apache.hadoop.ozone.OzoneConfigKeys.OZONE_FS_ITERATE_BATCH_SIZE;
 import static org.apache.hadoop.ozone.OzoneConsts.OZONE_URI_DELIMITER;
 import static org.apache.hadoop.ozone.OzoneConsts.OZONE_URI_SCHEME;
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static 
org.apache.hadoop.ozone.om.exceptions.OMException.ResultCodes.KEY_ALREADY_EXISTS;
 import static 
org.apache.hadoop.ozone.om.exceptions.OMException.ResultCodes.KEY_NOT_FOUND;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -397,7 +398,7 @@ public void testListKeysAtDifferentLevels() throws 
Exception {
     byte[] input = new byte[length];
     Arrays.fill(input, (byte)96);
 
-    createKeys(ozoneBucket, keys);
+    createAndAssertKeys(ozoneBucket, keys);
 
     // Root level listing keys
     Iterator<? extends OzoneKey> ozoneKeyIterator =
@@ -511,16 +512,9 @@ public void testListKeysWithNotNormalizedPath() throws 
Exception {
     keys.add(OmUtils.normalizeKey(key2, false));
     keys.add(OmUtils.normalizeKey(key3, false));
 
-    int length = 10;
-    byte[] input = new byte[length];
-    Arrays.fill(input, (byte)96);
-
-    createKey(ozoneBucket, key1, 10, input);
-    createKey(ozoneBucket, key2, 10, input);
-    createKey(ozoneBucket, key3, 10, input);
+    createAndAssertKeys(ozoneBucket, Arrays.asList(key1, key2, key3));
 
     // Iterator with key name as prefix.
-
     Iterator<? extends OzoneKey> ozoneKeyIterator =
             ozoneBucket.listKeys("/dir1//", null);
 
@@ -562,27 +556,23 @@ private void checkKeyList(Iterator<? extends OzoneKey > 
ozoneKeyIterator,
     assertEquals(keys, outputKeys);
   }
 
-  private void createKeys(OzoneBucket ozoneBucket, List<String> keys)
+  private void createAndAssertKeys(OzoneBucket ozoneBucket, List<String> keys)
       throws Exception {
+
     int length = 10;
     byte[] input = new byte[length];
     Arrays.fill(input, (byte) 96);
+
     for (String key : keys) {
-      createKey(ozoneBucket, key, 10, input);
+      createKey(ozoneBucket, key, input);
+      // Read the key with given key name.
+      readKey(ozoneBucket, key, length, input);
     }
   }
 
-  private void createKey(OzoneBucket ozoneBucket, String key, int length,
-      byte[] input) throws Exception {
-
-    OzoneOutputStream ozoneOutputStream =
-            ozoneBucket.createKey(key, length);
-
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
+  private void readKey(OzoneBucket ozoneBucket, String key, int length, byte[] 
input)
+      throws Exception {
 
-    // Read the key with given key name.
     OzoneInputStream ozoneInputStream = ozoneBucket.readKey(key);
     byte[] read = new byte[length];
     ozoneInputStream.read(read, 0, length);
@@ -593,9 +583,9 @@ private void createKey(OzoneBucket ozoneBucket, String key, 
int length,
 
     // Read using filesystem.
     String rootPath = String.format("%s://%s.%s/", OZONE_URI_SCHEME,
-            bucketName, volumeName, StandardCharsets.UTF_8);
+        bucketName, volumeName, StandardCharsets.UTF_8);
     OzoneFileSystem o3fs = (OzoneFileSystem) FileSystem.get(new URI(rootPath),
-            conf);
+        conf);
     FSDataInputStream fsDataInputStream = o3fs.open(new Path(key));
     read = new byte[length];
     fsDataInputStream.read(read, 0, length);
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestRecursiveAclWithFSO.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestRecursiveAclWithFSO.java
index 09b5ea5534..556f526403 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestRecursiveAclWithFSO.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/TestRecursiveAclWithFSO.java
@@ -17,6 +17,7 @@
 
 package org.apache.hadoop.ozone.om;
 
+import static org.apache.hadoop.ozone.TestDataUtil.createKey;
 import static org.apache.hadoop.ozone.security.acl.OzoneObj.StoreType.OZONE;
 import static org.junit.jupiter.api.Assertions.assertArrayEquals;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -37,7 +38,6 @@
 import org.apache.hadoop.ozone.client.OzoneBucket;
 import org.apache.hadoop.ozone.client.OzoneClient;
 import org.apache.hadoop.ozone.client.OzoneVolume;
-import org.apache.hadoop.ozone.client.io.OzoneOutputStream;
 import org.apache.hadoop.ozone.client.protocol.ClientProtocol;
 import org.apache.hadoop.ozone.om.exceptions.OMException;
 import org.apache.hadoop.ozone.security.acl.OzoneAclConfig;
@@ -250,7 +250,7 @@ public void testKeyDefaultACL() throws Exception {
       byte[] input = new byte[length];
       Arrays.fill(input, (byte) 96);
       String keyName = UUID.randomUUID().toString();
-      createKey(ozoneBucket, keyName, length, input);
+      createKey(ozoneBucket, keyName, input);
       obj = OzoneObjInfo.Builder.newBuilder().setVolumeName(volume.getName())
           .setBucketName(ozoneBucket.getName()).setKeyName(keyName)
           .setResType(OzoneObj.ResourceType.KEY).setStoreType(OZONE).build();
@@ -334,19 +334,10 @@ private void createKeys(ObjectStore objectStore, 
OzoneBucket ozoneBucket,
     byte[] input = new byte[length];
     Arrays.fill(input, (byte) 96);
     for (String key : keys) {
-      createKey(ozoneBucket, key, 10, input);
+      createKey(ozoneBucket, key, input);
       setKeyAcl(objectStore, ozoneBucket.getVolumeName(), 
ozoneBucket.getName(),
           key, aclWorldAll);
     }
   }
-
-  private void createKey(OzoneBucket ozoneBucket, String key, int length,
-      byte[] input) throws Exception {
-    OzoneOutputStream ozoneOutputStream = ozoneBucket.createKey(key, length);
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
-  }
-
 }
 
diff --git 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestOmSnapshotFileSystem.java
 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestOmSnapshotFileSystem.java
index 35d6bac0ed..a25f552185 100644
--- 
a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestOmSnapshotFileSystem.java
+++ 
b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestOmSnapshotFileSystem.java
@@ -76,7 +76,6 @@
 import org.apache.hadoop.ozone.client.OzoneSnapshot;
 import org.apache.hadoop.ozone.client.OzoneVolume;
 import org.apache.hadoop.ozone.client.io.OzoneInputStream;
-import org.apache.hadoop.ozone.client.io.OzoneOutputStream;
 import org.apache.hadoop.ozone.om.KeyManagerImpl;
 import org.apache.hadoop.ozone.om.OMConfigKeys;
 import org.apache.hadoop.ozone.om.OmSnapshotManager;
@@ -361,14 +360,13 @@ private void createKeys(OzoneBucket ozoneBucket, 
List<String> keys)
   private void createKey(OzoneBucket ozoneBucket, String key, int length,
                          byte[] input) throws Exception {
 
-    OzoneOutputStream ozoneOutputStream =
-        ozoneBucket.createKey(key, length);
-
-    ozoneOutputStream.write(input);
-    ozoneOutputStream.write(input, 0, 10);
-    ozoneOutputStream.close();
-
+    TestDataUtil.createKey(ozoneBucket, key, input);
     // Read the key with given key name.
+    readkey(ozoneBucket, key, length, input);
+  }
+
+  private void readkey(OzoneBucket ozoneBucket, String key, int length, byte[] 
input)
+      throws Exception {
     OzoneInputStream ozoneInputStream = ozoneBucket.readKey(key);
     byte[] read = new byte[length];
     ozoneInputStream.read(read, 0, length);


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to