Site Settings Desktop: Support SessionOnly for cookies on Site Details page.

chrome/test/data/webui/settings/site_details_permission_tests.js

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 /** @fileoverview Suite of tests for site-details. */
 cr.define('site_details_permission', function() {
   function registerTests() {
     suite('SiteDetailsPermission', function() {
       /**
        * A site list element created before each test.
        * @type {SiteDetailsPermission}
        */
       var testElement;
 
       /**
        * An example pref with only camera allowed.
        */
       var prefs = {
         exceptions: {
           camera: [
             {
-              embeddingOrigin: 'https://foo-allow.com:443',
-              origin: 'https://foo-allow.com:443',
+              embeddingOrigin: '',
+              origin: 'https://www.example.com',
               setting: 'allow',
               source: 'preference',
             },
+          ]
+        }
+      };
+
+      /**
+       * An example pref with only one entry allowed.
+       */
+      var prefsCookies = {
+        exceptions: {
+          cookies: [
+            {
+              embeddingOrigin: '',
+              origin: 'https://www.example.com',
+              setting: 'allow',
+              source: 'preference',
+            },
           ]
         }
       };
 
       // Import necessary html before running suite.
       suiteSetup(function() {
         return PolymerTest.importHtml(
             'chrome://md-settings/site_settings/site_details_permission.html'
             );
       });
@@ skipping 15 matching lines @@
             return exceptionList[i].setting == 'allow';
         }
         return false;
       };
 
       function validatePermissionFlipWorks(origin, allow) {
         MockInteractions.tap(allow ? testElement.$.allow : testElement.$.block);
         return browserProxy.whenCalled('setCategoryPermissionForOrigin').then(
             function(arguments) {
               assertEquals(origin, arguments[0]);
-              assertEquals(origin, arguments[1]);
+              assertEquals('', arguments[1]);
               assertEquals(testElement.category, arguments[2]);
               assertEquals(allow ?
                   settings.PermissionValues.ALLOW :
                   settings.PermissionValues.BLOCK, arguments[3]);
             });
       };
 
       test('empty state', function() {
         browserProxy.setPrefs(prefsEmpty);
         testElement.category = settings.ContentSettingsTypes.CAMERA;
         testElement.site = {
           origin: 'http://www.google.com',
           embeddingOrigin: '',
         };
 
         return browserProxy.whenCalled('getExceptionList').then(function() {
           assertTrue(testElement.$.details.hidden);
         });
       });
 
       test('camera category', function() {
-        var origin = "https://foo-allow.com:443";
+        var origin = "https://www.example.com";
         browserProxy.setPrefs(prefs);
         testElement.category = settings.ContentSettingsTypes.CAMERA;
         testElement.site = {
           origin: origin,
-          embeddingOrigin: origin,
+          embeddingOrigin: '',
         };
 
         return browserProxy.whenCalled('getExceptionList').then(function() {
           assertFalse(testElement.$.details.hidden);
 
           var header = testElement.$.details.querySelector(
               '#permissionHeader');
           assertEquals('Camera', header.innerText.trim(),
               'Widget should be labelled correctly');
 
           // Flip the permission and validate that prefs stay in sync.
           return validatePermissionFlipWorks(origin, true);
         }).then(function() {
           browserProxy.resetResolver('setCategoryPermissionForOrigin');
           return validatePermissionFlipWorks(origin, false);
         }).then(function() {
           browserProxy.resetResolver('setCategoryPermissionForOrigin');
           return validatePermissionFlipWorks(origin, true);
         });
       });
 
+      test('cookies category', function() {
+        var origin = "https://www.example.com";
+        browserProxy.setPrefs(prefsCookies);
+        testElement.category = settings.ContentSettingsTypes.COOKIES;
+        testElement.site = {
+          origin: origin,
+          embeddingOrigin: '',
+        };
+
+        return browserProxy.whenCalled('getExceptionList').then(function() {
+          assertFalse(testElement.$.details.hidden);
+
+          var header = testElement.$.details.querySelector(
+              '#permissionHeader');
+          assertEquals('Cookies', header.innerText.trim(),
+              'Widget should be labelled correctly');
+
+          MockInteractions.tap(testElement.$.sessionOnly);
+          return browserProxy.whenCalled('setCategoryPermissionForOrigin');
+        }).then(function(arguments) {
+          assertEquals(origin, arguments[0]);
+          assertEquals('', arguments[1]);
+          assertEquals(testElement.category, arguments[2]);
+          assertEquals(settings.PermissionValues.SESSION_ONLY, arguments[3]);
+          // Flip the permission and validate that prefs stay in sync.
+          browserProxy.resetResolver('setCategoryPermissionForOrigin');
+          return validatePermissionFlipWorks(origin, true);
+        }).then(function() {
+          browserProxy.resetResolver('setCategoryPermissionForOrigin');
+          return validatePermissionFlipWorks(origin, false);
+        });
+      });
+
       test('disappear on empty', function() {
-        var origin = "https://foo-allow.com:443";
+        var origin = "https://www.example.com";
         browserProxy.setPrefs(prefs);
         testElement.category = settings.ContentSettingsTypes.CAMERA;
         testElement.site = {
           origin: origin,
-          embeddingOrigin: origin,
+          embeddingOrigin: '',
         };
 
         return browserProxy.whenCalled('getExceptionList').then(function() {
           assertFalse(testElement.$.details.hidden);
 
           browserProxy.setPrefs(prefsEmpty);
           return browserProxy.whenCalled('getExceptionList');
         }).then(function() {
           assertTrue(testElement.$.details.hidden);
         });
       });
     });
   }
   return {
     registerTests: registerTests,
   };
 });