From 26ac7d82f19323c60318c605f76e6777633d7c9b Mon Sep 17 00:00:00 2001
From: Elkana Bardugo <ttv200@gmail.com>
Date: Fri, 13 Sep 2019 02:25:26 +0300
Subject: [PATCH] fix(example): sample exited before finishing write to disk
 (#1798)

---
 samples/drive/export.js | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/samples/drive/export.js b/samples/drive/export.js
index 99606d5149e..4a3f4552ec2 100644
--- a/samples/drive/export.js
+++ b/samples/drive/export.js
@@ -35,15 +35,10 @@ async function runSample() {
   );
   await new Promise((resolve, reject) => {
     res.data
-      .on('end', () => {
-        console.log(`Done downloading document: ${destPath}.`);
-        resolve();
-      })
-      .on('error', err => {
-        console.error('Error downloading document.');
-        reject(err);
-      })
-      .pipe(dest);
+      .on('error', reject)
+      .pipe(dest)
+      .on('error', reject)
+      .on('finish', resolve);
   });
   // [END main_body]
 }