Requesting background permission only when user is opted in to GN

chrome/browser/resources/google_now/background_unittest.gtestjs

 // Copyright 2013 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.
 
 // TODO(robliao,vadimt): Determine the granularity of testing to perform.
 
 /**
  * Test fixture for background.js.
  * @constructor
  * @extends {testing.Test}
@@ skipping 468 matching lines @@
  * Note that this CAN NOT be used if any of the methods below are called
  * outside of this context with the same argument matchers.
  * expects() calls cannot be chained with the same argument matchers.
  * @param {object} fixture Test fixture.
  * @param {string} testIdentityToken getAuthToken callback token.
  * @param {boolean} testGeolocationPref Geolocation Preference callback value.
  * @param {object} testExperimentVariationParams Response of
  *     metricsPrivate.getVariationParams.
  * @param {string} testExperimentVariationParams Response of
  *     notifications.getPermissionLevel.
+ * @param {boolean} testGoogleNowEnabled True if the user is opted in to Google
+ *     Now.
  */
 function expectStateMachineCalls(
     fixture,
     testIdentityToken,
     testGeolocationPref,
     testExperimentVariationParams,
-    testNotificationPermissionLevel) {
+    testNotificationPermissionLevel,
+    testGoogleNowEnabled) {
   var authenticationManagerIsSignedInSavedArgs = new SaveMockArguments();
   fixture.mockApis.expects(once()).
       authenticationManager_isSignedIn(
           authenticationManagerIsSignedInSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           authenticationManagerIsSignedInSavedArgs,
           0,
           testIdentityToken));
 
   var getVariationParamsSavedArgs = new SaveMockArguments();
@@ skipping 14 matching lines @@
 
   var notificationGetPermissionLevelSavedArgs = new SaveMockArguments();
   fixture.mockApis.expects(once()).
       instrumented_notifications_getPermissionLevel(
           notificationGetPermissionLevelSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           notificationGetPermissionLevelSavedArgs,
           0,
           testNotificationPermissionLevel))
 
+   var storageGetSavedArgs = new SaveMockArguments();    
+   fixture.mockApis.expects(once()).     
+       instrumented_storage_local_get(   
+           storageGetSavedArgs.match(eq('googleNowEnabled')),    
+           storageGetSavedArgs.match(ANYTHING)).         
+       will(invokeCallback(
+          storageGetSavedArgs, 1, {googleNowEnabled: testGoogleNowEnabled}));    
+
   fixture.mockGlobals.expects(once()).
       setBackgroundEnable(ANYTHING);
 }
 
 /**
  * Sets up the test to expect the initialization calls that
  * initialize() invokes.
  * Note that this CAN NOT be used if any of the methods below are called
  * outside of this context with the same argument matchers.
  * expects() calls cannot be chained with the same argument matchers.
@@ skipping 20 matching lines @@
     function() {
       // Tests the case when getAuthToken fails most likely because the user is
       // not signed in. In this case, the function should quietly exit after
       // finding out that getAuthToken fails.
 
       // Setup and expectations.
       var testIdentityToken = undefined;
       var testGeolocationPref = false;
       var testExperimentVariationParams = {};
       var testNotificationPermissionLevel = 'denied';
+      var testGoogleNowEnabled = undefined;
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.STOPPED);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
           this,
           testIdentityToken,
           testGeolocationPref,
           testExperimentVariationParams,
-          testNotificationPermissionLevel);
+          testNotificationPermissionLevel,
+          testGoogleNowEnabled);
 
       // Invoking the tested function.
       initialize();
     });
 
 TEST_F(
     'GoogleNowBackgroundUnitTest',
     'DISABLED_Initialize_ToastStateEmpty2',
     function() {
       // Tests the case when getAuthToken succeeds, and the user has never
       // responded to the toast.
       // In this case, the function should invoke showWelcomeToast().
 
       // Setup and expectations.
       var testIdentityToken = 'some identity token';
       var testGeolocationPref = false;
       var testExperimentVariationParams = {};
       var testNotificationPermissionLevel = 'denied';
+      var testGoogleNowEnabled = undefined;
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
           this,
           testIdentityToken,
           testGeolocationPref,
           testExperimentVariationParams,
-          testNotificationPermissionLevel);
+          testNotificationPermissionLevel,
+          testGoogleNowEnabled);
 
       var chromeNotificationGetAllSavedArgs = new SaveMockArguments();
       this.mockApis.expects(exactly(1)).
           instrumented_notifications_getAll(
               chromeNotificationGetAllSavedArgs.match(ANYTHING)).
           will(
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}),
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}));
 
       // Invoking the tested function.
       initialize();
     });
 
 TEST_F('GoogleNowBackgroundUnitTest', 'Initialize_RunGoogleNow', function() {
   // Tests if Google Now will invoke startPollingCards when all
   // of the required state is fulfilled.
 
   // Setup and expectations.
   var testIdentityToken = 'some identity token';
   var testGeolocationPref = true;
   var testExperimentVariationParams = {};
   var testNotificationPermissionLevel = 'granted';
+  var testGoogleNowEnabled = true;
 
   mockInitializeDependencies(this);
 
   this.mockGlobals.expects(once()).recordEvent(
         GoogleNowEvent.EXTENSION_START);
 
   expectInitialization(this.mockApis);
 
   expectStateMachineCalls(
       this,
       testIdentityToken,
       testGeolocationPref,
       testExperimentVariationParams,
-      testNotificationPermissionLevel);
+      testNotificationPermissionLevel,
+      testGoogleNowEnabled);
 
   this.mockGlobals.expects(once()).startPollingCards();
 
   // Invoking the tested function.
   initialize();
 });
 
 TEST_F(
     'GoogleNowBackgroundUnitTest',
     'DISABLED_Initialize_NoGeolocation',
     function() {
   // Tests the case where everything is in place except for the
   // Geolocation Preference after the user responded to the toast.
 
   // Setup and expectations.
   var testIdentityToken = 'some identity token';
   var testGeolocationPref = false;
   var testExperimentVariationParams = {};
   var testNotificationPermissionLevel = 'denied';
+  var testGoogleNowEnabled = undefined;
+
 
   mockInitializeDependencies(this);
 
   this.mockGlobals.expects(once()).recordEvent(
         GoogleNowEvent.EXTENSION_START);
 
   this.mockGlobals.expects(once()).recordEvent(
       GoogleNowEvent.USER_SUPPRESSED);
 
   expectInitialization(this.mockApis);
 
   expectStateMachineCalls(
       this,
       testIdentityToken,
       testGeolocationPref,
       testExperimentVariationParams,
       testNotificationPermissionLevel,
-      testNotificationPermissionLevel);
+      testNotificationPermissionLevel,
+      testGoogleNowEnabled);
 
   var chromeNotificationGetAllSavedArgs = new SaveMockArguments();
   this.mockApis.expects(exactly(2)).
       instrumented_notifications_getAll(
           chromeNotificationGetAllSavedArgs.match(ANYTHING)).
       will(
           invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}),
           invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}));
 
   // Invoking the tested function.
@@ skipping 145 matching lines @@
               chromeTabsCreateSavedArgs.match(eqJSON({url: testActionUrl})),
               chromeTabsCreateSavedArgs.match(ANYTHING)).
           will(invokeCallback(chromeTabsCreateSavedArgs, 1, testCreatedTab));
       this.mockApis.expects(once()).chrome_windows_create(
           eqJSON({url: testActionUrl, focused: true}));
 
       // Invoking the tested function.
       onNotificationClicked(
           testNotificationId, this.mockLocalFunctions.functions().selector);
     });