Fix MessageCollector.next edge case (#765)

This commit is contained in:
Programmix
2016-10-01 21:23:35 -07:00
committed by Schuyler Cebulskie
parent 1fcc618532
commit d35372d3e9

View File

@@ -100,6 +100,11 @@ class MessageCollector extends EventEmitter {
*/
get next() {
return new Promise((resolve, reject) => {
if (this.ended) {
reject(this.collected);
return;
}
const cleanup = () => {
this.removeListener('message', onMessage);
this.removeListener('end', onEnd);