1 files changed, 13 insertions, 13 deletions
diff --git a/node_modules/selenium-webdriver/test/upload_test.js b/node_modules/selenium-webdriver/test/upload_test.js
index 3329f7ca7..c677550fc 100644
--- a/node_modules/selenium-webdriver/test/upload_test.js
+++ b/node_modules/selenium-webdriver/test/upload_test.js
@@ -41,13 +41,13 @@ test.suite(function(env) {
   })
 
   var driver;
-  test.before(function() {
-    driver = env.builder().build();
+  test.before(function*() {
+    driver = yield env.builder().build();
   });
 
   test.after(function() {
     if (driver) {
-      driver.quit();
+      return driver.quit();
     }
   });
 
@@ -58,29 +58,29 @@ test.suite(function(env) {
       // See https://github.com/ariya/phantomjs/issues/12506
       Browser.PHANTOM_JS,
       Browser.SAFARI)).
-  it('can upload files', function() {
+  it('can upload files', function*() {
     driver.setFileDetector(new remote.FileDetector);
 
-    driver.get(Pages.uploadPage);
+    yield driver.get(Pages.uploadPage);
 
-    var fp = driver.call(function() {
+    var fp = yield driver.call(function() {
       return io.tmpFile().then(function(fp) {
         fs.writeFileSync(fp, FILE_HTML);
         return fp;
       });
     });
 
-    driver.findElement(By.id('upload')).sendKeys(fp);
-    driver.findElement(By.id('go')).click();
+    yield driver.findElement(By.id('upload')).sendKeys(fp);
+    yield driver.findElement(By.id('go')).click();
 
     // Uploading files across a network may take a while, even if they're small.
-    var label = driver.findElement(By.id('upload_label'));
-    driver.wait(until.elementIsNotVisible(label),
+    var label = yield driver.findElement(By.id('upload_label'));
+    yield driver.wait(until.elementIsNotVisible(label),
         10 * 1000, 'File took longer than 10 seconds to upload!');
 
-    var frame = driver.findElement(By.id('upload_target'));
-    driver.switchTo().frame(frame);
-    assert(driver.findElement(By.css('body')).getText())
+    var frame = yield driver.findElement(By.id('upload_target'));
+    yield driver.switchTo().frame(frame);
+    yield assert(driver.findElement(By.css('body')).getText())
         .equalTo(LOREM_IPSUM_TEXT);
   });
 });