Change LocalNode stop() function to return a Promise (#465)

change LocalNode stop() function to return a promise

CHANGELOG.md

@@ -4,6 +4,8 @@ All notable changes to the Aptos TypeScript SDK will be captured in this file. T
 
 # Unreleased
 
+- Change the `stop()` function on `LocalNode` to return a `Promise` so we can wait for the processes to be killed
+
 # 1.24.0 (2024-07-12)
 
 - Make `fundAccount` to wait for the `fungible_asset_processor` indexer processor

examples/typescript/local_node.ts

@@ -8,9 +8,11 @@
 
 const cli = require("@aptos-labs/ts-sdk/dist/common/cli/index.js");
 
+let localNode: any;
+
 // Run local node
 async function runLocalNode() {
-  const localNode = new cli.LocalNode();
+  localNode = new cli.LocalNode();
   await localNode.run();
 }
 
@@ -99,15 +101,31 @@ async function runScript() {
   });
 }
 
+// Stop local node
+async function stopLocalNode() {
+  await localNode.stop();
+  try {
+    // Query localnet endpoint
+    await fetch("http://localhost:8080");
+  } catch (err: any) {
+    console.log("localnet stopped");
+  }
+}
+
 async function run() {
+  // start the localnet
   await runLocalNode();
+
   await init();
   await compile();
   await tests();
   await publish();
   await createObjectAndPublishPackage();
   await upgradeObjectPackage();
   await runScript();
+
+  // stop the localnet
+  await stopLocalNode();
 }
 
 run();

src/cli/localNode.ts

@@ -15,9 +15,17 @@ export class LocalNode {
    * kills all the descendent processes
    * of the node process, including the node process itself
    */
-  stop() {
-    if (!this.process?.pid) return;
-    kill(this.process.pid);
+  async stop() {
+    await new Promise((resolve, reject) => {
+      if (!this.process?.pid) return;
+      kill(this.process.pid, (err) => {
+        if (err) {
+          reject(err);
+        } else {
+          resolve(true);
+        }
+      });
+    });
   }
 
   /**
@@ -40,7 +48,7 @@ export class LocalNode {
    */
   start() {
     const cliCommand = "npx";
-    const cliArgs = ["aptos", "node", "run-local-testnet", "--force-restart", "--assume-yes", "--with-indexer-api"];
+    const cliArgs = ["aptos", "node", "run-localnet", "--force-restart", "--assume-yes", "--with-indexer-api"];
 
     const currentPlatform = platform();
     let childProcess;