Start/Stop Cards Polling Depending on Notification Center State

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 449 matching lines @@
     'recordEvent',
     'setBackgroundEnable',
     'startPollingCards'
   ]);
   fixture.makeAndRegisterMockApis([
     'authenticationManager.isSignedIn',
     'chrome.location.clearWatch',
     'chrome.storage.local.remove',
     'instrumented.metricsPrivate.getVariationParams',
     'instrumented.notifications.getAll',
+    'instrumented.notifications.getPermissionLevel',
     'instrumented.preferencesPrivate.googleGeolocationAccessEnabled.get',
     'instrumented.storage.local.get',
     'tasks.add',
     'updateCardsAttempts.isRunning',
     'updateCardsAttempts.stop'
   ]);
 }
 
 /**
  * Sets up the test to expect the state machine calls and send
  * the specified state machine state. Currently used to test initialize().
  * 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
+ *     metricsPrivate.getVariationParams.
+ * @param {string} testExperimentVariationParams Response of
+ *     notifications.getPermissionLevel.
  */
 function expectStateMachineCalls(
     fixture,
     testIdentityToken,
     testGeolocationPref,
-    testExperimentVariationParams) {
+    testExperimentVariationParams,
+    testNotificationPermissionLevel) {
   var authenticationManagerIsSignedInSavedArgs = new SaveMockArguments();
   fixture.mockApis.expects(once()).
       authenticationManager_isSignedIn(
           authenticationManagerIsSignedInSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           authenticationManagerIsSignedInSavedArgs,
           0,
           testIdentityToken));
 
   var getVariationParamsSavedArgs = new SaveMockArguments();
   fixture.mockApis.expects(once()).
       instrumented_metricsPrivate_getVariationParams(
           getVariationParamsSavedArgs.match(ANYTHING),
           getVariationParamsSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           getVariationParamsSavedArgs, 1, testExperimentVariationParams));
 
   var googleGeolocationPrefGetSavedArgs = new SaveMockArguments();
   fixture.mockApis.expects(once()).
       instrumented_preferencesPrivate_googleGeolocationAccessEnabled_get(
           googleGeolocationPrefGetSavedArgs.match(eqJSON({})),
           googleGeolocationPrefGetSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           googleGeolocationPrefGetSavedArgs, 1, {value: testGeolocationPref}));
 
+  var notificationGetPermissionLevelSavedArgs = new SaveMockArguments();
+  fixture.mockApis.expects(once()).
+      instrumented_notifications_getPermissionLevel(
+          notificationGetPermissionLevelSavedArgs.match(ANYTHING)).
+      will(invokeCallback(
+          notificationGetPermissionLevelSavedArgs,
+          0,
+          testNotificationPermissionLevel))
+
   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 19 matching lines @@
     'Initialize_ToastStateEmpty',
     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';
 
       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);
+          testExperimentVariationParams,
+          testNotificationPermissionLevel);
 
       // 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';
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
           this,
           testIdentityToken,
           testGeolocationPref,
-          testExperimentVariationParams);
+          testExperimentVariationParams,
+          testNotificationPermissionLevel);
 
       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';
 
   mockInitializeDependencies(this);
 
   this.mockGlobals.expects(once()).recordEvent(
         GoogleNowEvent.EXTENSION_START);
 
   expectInitialization(this.mockApis);
 
   expectStateMachineCalls(
       this,
       testIdentityToken,
       testGeolocationPref,
-      testExperimentVariationParams);
+      testExperimentVariationParams,
+      testNotificationPermissionLevel);
 
   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';
 
   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);
+      testExperimentVariationParams,
+      testNotificationPermissionLevel,
+      testNotificationPermissionLevel);
 
   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);
     });