bugfix: S3C-2052 Add error functions

scality · Aug 20, 2019 · 9d02f86 · 9d02f86
1 parent 5de8571
commit 9d02f86
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/lib/storage/metadata/in_memory/metastore.js b/lib/storage/metadata/in_memory/metastore.js
@@ -26,7 +26,18 @@ function inc(str) {
 }
 
 const metastore = {
-    errors: {}, // Used for simulation of metadata errors.
+    // Used for simulation of metadata errors.
+    error: () => {},
+
+    setErrorFunc(func) {
+        this.error = func;
+        return this;
+    },
+
+    clearErrorFunc() {
+        this.error = () => {};
+        return this;
+    },
 
     createBucket: (bucketName, bucketMD, log, cb) => {
         process.nextTick(() => {
@@ -83,8 +94,9 @@ const metastore = {
     },
 
     putObject: (bucketName, objName, objVal, params, log, cb) => {
-        if (metastore.errors.putObject) {
-            return process.nextTick(() => cb(metastore.errors.putObject));
+        const err = metastore.error(metastore.putObject.name);
+        if (err) {
+            return process.nextTick(() => cb(err));
         }
         return process.nextTick(() => {
             metastore.getBucketAttributes(bucketName, log, err => {
@@ -157,7 +169,11 @@ const metastore = {
     },
 
     getObject: (bucketName, objName, params, log, cb) => {
-        process.nextTick(() => {
+        const err = metastore.error(metastore.getObject.name, objName);
+        if (err) {
+            return process.nextTick(() => cb(err));
+        }
+        return process.nextTick(() => {
             metastore.getBucketAttributes(bucketName, log, err => {
                 if (err) {
                     return cb(err);