test fix

sadanand48 · Sep 5, 2024 · 15170e8 · 15170e8
1 parent efb3516
commit 15170e8
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/.../ozonefs-common/src/main/java/org/apache/hadoop/fs/ozone/BasicOzoneClientAdapterImpl.java b/.../ozonefs-common/src/main/java/org/apache/hadoop/fs/ozone/BasicOzoneClientAdapterImpl.java
@@ -820,7 +820,15 @@ public boolean setSafeMode(SafeModeAction action, boolean isChecked)
 
   @Override
   public void setReplication(String key, short replication) throws IOException {
-    OzoneKeyDetails keyDetails = bucket.getKey(key);
+    OzoneKeyDetails keyDetails = null;
+    try {
+      keyDetails = bucket.getKey(key);
+    } catch (OMException ome) {
+      if (ome.getResult() == KEY_NOT_FOUND) {
+        return;
+      }
+      throw ome;
+    }
     ReplicationConfig newReplication = OzoneClientUtils
         .resolveClientSideReplicationConfig(replication, null,
             null, config);

diff --git a/...fs-common/src/main/java/org/apache/hadoop/fs/ozone/BasicRootedOzoneClientAdapterImpl.java b/...fs-common/src/main/java/org/apache/hadoop/fs/ozone/BasicRootedOzoneClientAdapterImpl.java
@@ -401,7 +401,16 @@ protected void incrementCounter(Statistic objectsRead, long count) {
   public void setReplication(String key, short replication) throws IOException {
     OFSPath ofsPath = new OFSPath(key, config);
     OzoneBucket bucket = getBucket(ofsPath, false);
-    OzoneKeyDetails keyDetails = bucket.getKey(ofsPath.getKeyName());
+    OzoneKeyDetails keyDetails = null;
+    try {
+      keyDetails = bucket.getKey(key);
+    } catch (OMException ome) {
+      // if key does not exist, do nothing
+      if (ome.getResult() == KEY_NOT_FOUND) {
+        return;
+      }
+      throw ome;
+    }
     ReplicationConfig newReplication = OzoneClientUtils
         .resolveClientSideReplicationConfig(replication, null,
             null, config);