From 85ee9f0f6520bea0e8cdac011d7bb5068cfd6821 Mon Sep 17 00:00:00 2001
From: Andrey Lushnikov <aslushnikov@gmail.com>
Date: Fri, 22 Feb 2019 23:59:32 -0800
Subject: [PATCH] feat(firefox): implement browserContext.overridePermissions
 (#4060)

---
 experimental/puppeteer-firefox/lib/Browser.js | 24 ++++++++++++++++++
 experimental/puppeteer-firefox/package.json   |  2 +-
 test/page.spec.js                             | 25 ++++++++++++++++---
 3 files changed, 47 insertions(+), 4 deletions(-)

diff --git a/experimental/puppeteer-firefox/lib/Browser.js b/experimental/puppeteer-firefox/lib/Browser.js
index 34dbae03c7cff..44a1768b35f08 100644
--- a/experimental/puppeteer-firefox/lib/Browser.js
+++ b/experimental/puppeteer-firefox/lib/Browser.js
@@ -283,6 +283,30 @@ class BrowserContext extends EventEmitter {
     this._browserContextId = browserContextId;
   }
 
+  /**
+   * @param {string} origin
+   * @param {!Array<string>} permissions
+   */
+  async overridePermissions(origin, permissions) {
+    const webPermissionToProtocol = new Map([
+      ['geolocation', 'geo'],
+      ['microphone', 'microphone'],
+      ['camera', 'camera'],
+      ['notifications', 'desktop-notifications'],
+    ]);
+    permissions = permissions.map(permission => {
+      const protocolPermission = webPermissionToProtocol.get(permission);
+      if (!protocolPermission)
+        throw new Error('Unknown permission: ' + permission);
+      return protocolPermission;
+    });
+    await this._connection.send('Browser.grantPermissions', {origin, browserContextId: this._browserContextId || undefined, permissions});
+  }
+
+  async clearPermissionOverrides() {
+    await this._connection.send('Browser.resetPermissions', {browserContextId: this._browserContextId || undefined});
+  }
+
   /**
    * @return {Array<Target>}
    */
diff --git a/experimental/puppeteer-firefox/package.json b/experimental/puppeteer-firefox/package.json
index d75438e567fdf..af11aa02168ab 100644
--- a/experimental/puppeteer-firefox/package.json
+++ b/experimental/puppeteer-firefox/package.json
@@ -9,7 +9,7 @@
     "node": ">=8.9.4"
   },
   "puppeteer": {
-    "firefox_revision": "e0433ecec0642eca12cba778b0587ab2ae06e833"
+    "firefox_revision": "f7b25713dd00f0deda7032dc25a72d4c7b42446e"
   },
   "scripts": {
     "install": "node install.js",
diff --git a/test/page.spec.js b/test/page.spec.js
index 57b55eed02b79..c5633f9e185c8 100644
--- a/test/page.spec.js
+++ b/test/page.spec.js
@@ -163,7 +163,7 @@ module.exports.addTests = function({testRunner, expect, headless, Errors, Device
     });
   });
 
-  describe_fails_ffox('BrowserContext.overridePermissions', function() {
+  describe('BrowserContext.overridePermissions', function() {
     function getPermission(page, name) {
       return page.evaluate(name => navigator.permissions.query({name}).then(result => result.state), name);
     }
@@ -199,7 +199,7 @@ module.exports.addTests = function({testRunner, expect, headless, Errors, Device
       await page.goto(server.EMPTY_PAGE);
       await page.evaluate(() => {
         window.events = [];
-        return navigator.permissions.query({name: 'clipboard-read'}).then(function(result) {
+        return navigator.permissions.query({name: 'geolocation'}).then(function(result) {
           window.events.push(result.state);
           result.onchange = function() {
             window.events.push(result.state);
@@ -209,11 +209,30 @@ module.exports.addTests = function({testRunner, expect, headless, Errors, Device
       expect(await page.evaluate(() => window.events)).toEqual(['prompt']);
       await context.overridePermissions(server.EMPTY_PAGE, []);
       expect(await page.evaluate(() => window.events)).toEqual(['prompt', 'denied']);
-      await context.overridePermissions(server.EMPTY_PAGE, ['clipboard-read']);
+      await context.overridePermissions(server.EMPTY_PAGE, ['geolocation']);
       expect(await page.evaluate(() => window.events)).toEqual(['prompt', 'denied', 'granted']);
       await context.clearPermissionOverrides();
       expect(await page.evaluate(() => window.events)).toEqual(['prompt', 'denied', 'granted', 'prompt']);
     });
+    it('should isolate permissions between browser contexs', async({page, server, context, browser}) => {
+      await page.goto(server.EMPTY_PAGE);
+      const otherContext = await browser.createIncognitoBrowserContext();
+      const otherPage = await otherContext.newPage();
+      await otherPage.goto(server.EMPTY_PAGE);
+      expect(await getPermission(page, 'geolocation')).toBe('prompt');
+      expect(await getPermission(otherPage, 'geolocation')).toBe('prompt');
+
+      await context.overridePermissions(server.EMPTY_PAGE, []);
+      await otherContext.overridePermissions(server.EMPTY_PAGE, ['geolocation']);
+      expect(await getPermission(page, 'geolocation')).toBe('denied');
+      expect(await getPermission(otherPage, 'geolocation')).toBe('granted');
+
+      await context.clearPermissionOverrides();
+      expect(await getPermission(page, 'geolocation')).toBe('prompt');
+      expect(await getPermission(otherPage, 'geolocation')).toBe('granted');
+
+      await otherContext.close();
+    });
   });
 
   describe_fails_ffox('Page.setGeolocation', function() {