Merge branches 'w/8.1/feature/S3C-3112_ObjectLockEnabledSetterForBuck…

…et' and 'q/1174/7.8/feature/S3C-3112_ObjectLockEnabledSetterForBucket' into tmp/octopus/q/8.1
scality · Jun 30, 2020 · 668d90b · 668d90b
3 parents c1cfc59 + 6383d14 + 9f90e1e
commit 668d90b
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);
+                });
+            });
         });
     })
 );