fix: PromisePoolCluster.of returns PromisePoolCluster instead of PoolNamespace

promise.js

@@ -55,6 +55,55 @@
   return new PromisePool(corePool, thePromise);
 }
 
+class PromisePoolNamespace {
+
+  constructor(poolNamespace, thePromise) {
+    this.poolNamespace = poolNamespace;
+    this.Promise = thePromise || Promise;
+  }
+
+  getConnection() {
+    const corePoolNamespace = this.poolNamespace;
+    return new this.Promise((resolve, reject) => {
+      corePoolNamespace.getConnection((err, coreConnection) => {
+        if (err) {
+          reject(err);
+        } else {
+          resolve(new PromisePoolConnection(coreConnection, this.Promise));
+        }
+      });
+    });
+  }
+
+  query(sql, values) {
+    const corePoolNamespace = this.poolNamespace;
+    const localErr = new Error();
+    if (typeof values === 'function') {
+      throw new Error(
+        'Callback function is not available with promise clients.',
+      );
+    }
+    return new this.Promise((resolve, reject) => {
+      const done = makeDoneCb(resolve, reject, localErr);
+      corePoolNamespace.query(sql, values, done);
+    });
+  }
+
+  execute(sql, values) {
+    const corePoolNamespace = this.poolNamespace;
+    const localErr = new Error();
+    if (typeof values === 'function') {
+      throw new Error(
+        'Callback function is not available with promise clients.',
+      );
+    }
+    return new this.Promise((resolve, reject) => {
+      const done = makeDoneCb(resolve, reject, localErr);
+      corePoolNamespace.execute(sql, values, done);
+    });
+  }
+}
+
 class PromisePoolCluster extends EventEmitter {
   constructor(poolCluster, thePromise) {
     super();
@@ -109,7 +158,7 @@
   }
 
   of(pattern, selector) {
-    return new PromisePoolCluster(
+    return new PromisePoolNamespace(
       this.poolCluster.of(pattern, selector),
       this.Promise,
     );

test/esm/integration/pool-cluster/test-promise-wrapper.test.mjs

@@ -79,4 +79,35 @@ const { createPoolCluster } = require('../../../../promise.js');
 
     poolCluster.poolCluster.emit('online');
   });
+
+  await test(async () => {
+    const poolCluster = createPoolCluster();
+    poolCluster.add('MASTER', common.config);
+
+    const poolNamespace = poolCluster.of('MASTER');
+
+    assert.equal(
+      poolNamespace.poolNamespace,
+      poolCluster.poolCluster.of('MASTER'),
+    );
+
+    const connection = await poolNamespace.getConnection();
+
+    assert.ok(connection, 'should get connection');
+    connection.release();
+
+    const [result] = await poolNamespace.query(
+      'SELECT 1 as a from dual where 1 = ?',
+      [1],
+    );
+    assert.equal(result[0]['a'], 1, 'should query successfully');
+
+    const [result2] = await poolNamespace.execute(
+      'SELECT 1 as a from dual where 1 = ?',
+      [1],
+    );
+    assert.equal(result2[0]['a'], 1, 'should execute successfully');
+
+    poolCluster.end();
+  });
 })();