Fix: PromisePoolCluster.of returns PromisePoolCluster instead of Pool…

…Namespace (sidorares#3091)

promise.js

@@ -74,6 +74,34 @@ class PromisePoolNamespace {
       });
     });
   }
+
+  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 {

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

@@ -43,4 +43,22 @@ const { createPoolCluster } = require('../../../../promise.js');
 
     poolCluster.poolCluster.emit('remove');
   });
+
+  await test(async () => {
+    const poolCluster = createPoolCluster();
+    poolCluster.add('MASTER', common.config);
+
+    const poolNamespace = 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');
+  });
 })();