Merge remote-tracking branch 'origin/w/7.8/feature/S3C-3112_ObjectLoc…

…kEnabledSetterForBucket' into w/8.1/feature/S3C-3112_ObjectLockEnabledSetterForBucket
scality · Jun 29, 2020 · 6383d14 · 6383d14
2 parents dc4e182 + 1e47b00
commit 6383d14
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/lib/models/BucketInfo.js b/lib/models/BucketInfo.js
@@ -712,12 +712,21 @@ class BucketInfo {
         return this;
     }
     /**
-    * Check is object lock enabled.
+    * Check if object lock is enabled.
     * @return {boolean} - depending on whether object lock is enabled
     */
     isObjectLockEnabled() {
         return !!this._objectLockEnabled;
     }
+    /**
+    * Set the value of objectLockEnabled field.
+    * @param {boolean} enabled - true if object lock enabled else false.
+    * @return {BucketInfo} - bucket info instance
+    */
+    setObjectLockEnabled(enabled) {
+        this._objectLockEnabled = enabled;
+        return this;
+    }
 }
 
 module.exports = BucketInfo;
diff --git a/tests/unit/models/BucketInfo.js b/tests/unit/models/BucketInfo.js
@@ -564,6 +564,13 @@ Object.keys(acl).forEach(
                 assert.deepStrictEqual(dummyBucket.getObjectLockConfiguration(),
                     newObjectLockConfig);
             });
+            [true, false].forEach(bool => {
+                it('setObjectLockEnabled should set object lock status', () => {
+                    dummyBucket.setObjectLockEnabled(bool);
+                    assert.deepStrictEqual(dummyBucket.isObjectLockEnabled(),
+                        bool);
+                });
+            });
         });
     })
 );