Issue 6959 - Use async/await in test/stub-modules/io.js

test/stub-modules/io.js

 "use strict";
 
 let data = new Map();
 
 exports.IO = {
   // Non-public API, for tests only
   _getFileContents(fileName)
   {
     if (data.has(fileName))
       return data.get(fileName).contents;
@@ skipping 12 matching lines @@
       return data.get(fileName).lastModified;
     return 0;
   },
   _setModifiedTime(fileName, lastModified)
   {
     if (data.has(fileName))
       data.get(fileName).lastModified = lastModified;
   },
 
   // Public API
-  writeToFile(fileName, generator)
+  async writeToFile(fileName, generator)
   {
-    return Promise.resolve().then(() =>
-    {
-      data.set(fileName, {
-        lastModified: Date.now(),
-        contents: Array.from(generator)
-      });
+    data.set(fileName, {
+      lastModified: Date.now(),
+      contents: Array.from(generator)
     });
   },
-  readFromFile(fileName, listener)
+  async readFromFile(fileName, listener)
   {
-    return Promise.resolve().then(() =>
+    if (!data.has(fileName))
+      throw new Error("File doesn't exist");
+
+    let lines = data.get(fileName).contents;
+    for (let line of lines)
+      listener(line);
+  },
+  async copyFile(fromName, toName)
+  {
+    if (!data.has(fromName))
+      throw new Error("File doesn't exist");
+
+    if (fromName == toName)
+      throw new Error("Cannot copy file to itself");
+
+    data.set(toName, data.get(fromName));
+  },
+  async renameFile(fromName, toName)
+  {
+    await this.copyFile(fromName, toName);
+    await this.removeFile(fromName);

[Manish Jethani, 2019/01/17 09:29:55]

Note that to be exactly like the previous version the last line would be:

  return await this.removeFile(fromName);

But the promise returned by removeFile() resolves with undefined anyway so it's
the same in practice.

+  },
+  async removeFile(fileName)
+  {
+    if (!data.has(fileName))
+      throw new Error("File doesn't exist");
+
+    data.delete(fileName);
+  },
+  async statFile(fileName)
+  {
+    if (data.has(fileName))
     {
-      if (!data.has(fileName))
-        throw new Error("File doesn't exist");
+      return {
+        exists: true,
+        lastModified: data.get(fileName).lastModified
+      };
+    }
 
-      let lines = data.get(fileName).contents;
-      for (let line of lines)
-        listener(line);
-    });
-  },
-  copyFile(fromName, toName)
-  {
-    return Promise.resolve().then(() =>
-    {
-      if (!data.has(fromName))
-        throw new Error("File doesn't exist");
-      if (fromName == toName)
-        throw new Error("Cannot copy file to itself");
-
-      data.set(toName, data.get(fromName));
-    });
-  },
-  renameFile(fromName, toName)
-  {
-    return this.copyFile(fromName, toName).then(() => this.removeFile(fromName));
-  },
-  removeFile(fileName)
-  {
-    return Promise.resolve().then(() =>
-    {
-      if (!data.has(fileName))
-        throw new Error("File doesn't exist");
-
-      data.delete(fileName);
-    });
-  },
-  statFile(fileName)
-  {
-    return Promise.resolve().then(() =>
-    {
-      if (data.has(fileName))
-      {
-        return {
-          exists: true,
-          lastModified: data.get(fileName).lastModified
-        };
-      }
-      return {
-        exists: false,
-        lastModified: 0
-      };
-    });
+    return {
+      exists: false,
+      lastModified: 0
+    };
   }
 };