fix(example): sample exited before finishing write to disk (#1798)

googleapis · Sep 12, 2019 · 26ac7d8 · 26ac7d8
1 parent bc32427
commit 26ac7d8
Showing 1 changed file with 4 additions and 9 deletions.
diff --git 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]
 }