Add more post-test reset code for addon button test.

mozilla-services · Mar 2, 2018 · c83cccc · c83cccc
1 parent 8907447
commit c83cccc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/test.js b/test/test.js
@@ -202,7 +202,10 @@ describe("Test Screenshots", function() {
         done();
       }),
       () => {
-        driver.setContext(firefox.Context.CONTENT);
+        return driver.setContext(firefox.Context.CONTENT)
+          .then(() => driver.getAllWindowHandles())
+          .then((tabs) => driver.switchTo().window(tabs[tabs.length - 1]))
+          .then(() => driver.switchTo().defaultContent());
       }).catch(done);
   });