From 011766768a77e543bac0c23cde20ff211288c5da Mon Sep 17 00:00:00 2001
From: Antonio Scandurra <as-cii@github.com>
Date: Fri, 22 Dec 2017 12:05:03 +0100
Subject: [PATCH] Fix AtomEnvironment tests

---
 spec/atom-environment-spec.js | 2 +-
 src/atom-environment.js       | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/spec/atom-environment-spec.js b/spec/atom-environment-spec.js
index e3b7b83e7..70ca9c309 100644
--- a/spec/atom-environment-spec.js
+++ b/spec/atom-environment-spec.js
@@ -592,7 +592,7 @@ describe('AtomEnvironment', () => {
       const promise = new Promise((r) => { resolve = r })
       envLoaded = () => {
         resolve()
-        promise
+        return promise
       }
       atomEnvironment = new AtomEnvironment({
         applicationDelegate: atom.applicationDelegate,
diff --git a/src/atom-environment.js b/src/atom-environment.js
index 50a41fb53..ae0ba8276 100644
--- a/src/atom-environment.js
+++ b/src/atom-environment.js
@@ -1013,8 +1013,10 @@ class AtomEnvironment {
   }
 
   addProjectFolder () {
-    this.pickFolder((selectedPaths = []) => {
-      this.addToProject(selectedPaths)
+    return new Promise((resolve) => {
+      this.pickFolder((selectedPaths) => {
+        this.addToProject(selectedPaths || []).then(resolve)
+      })
     })
   }