f

ali-sdk · Mar 27, 2018 · 519e164 · 519e164
1 parent 1de2b1a
commit 519e164
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/lib/mq_client.js b/lib/mq_client.js
@@ -686,6 +686,8 @@ module.exports = MQClient;
 
 // Helper
 // -----------------
-function sleep(interval) {
-  return callback => setTimeout(callback, interval);
+function sleep(n) {
+  return new Promise(resolve => {
+    setTimeout(resolve, n);
+  });
 }
diff --git a/test/index.test.js b/test/index.test.js
@@ -33,7 +33,7 @@ describe('test/index.test.js', () => {
       }, config));
       await consumer.ready();
 
-      consumer.subscribe('TEST_TOPIC', function* (msg) {
+      consumer.subscribe('TEST_TOPIC', async msg => {
         console.log('message receive ------------> ', msg.body.toString());
       });
     });
@@ -64,7 +64,7 @@ describe('test/index.test.js', () => {
       let sendResult = await producer.send(msg);
       assert(sendResult && sendResult.msgId);
 
-      mm(producer, 'sendKernelImpl', function* () {
+      mm(producer, 'sendKernelImpl', async () => {
         mm.restore();
         const err = new Error('mock err');
         err.name = 'MQClientException';
@@ -186,7 +186,7 @@ describe('test/index.test.js', () => {
       });
       assert(offset === -1);
 
-      mm(consumer._offsetStore, 'readOffset', function* () {
+      mm(consumer._offsetStore, 'readOffset', async () => {
         return 100;
       });
       offset = await consumer.computePullFromWhere({
@@ -254,7 +254,7 @@ describe('test/index.test.js', () => {
 
         it('should subscribe message ok', async () => {
           let msgId;
-          consumer.subscribe('TEST_TOPIC', 'TagA', function* (msg) {
+          consumer.subscribe('TEST_TOPIC', 'TagA', async msg => {
             console.log('message receive ------------> ', msg.body.toString());
             assert(msg.tags !== 'TagB');
             if (msg.msgId === msgId) {