Add Finch Checks to the State Machine

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.
 
 /**
  * Test fixture for background.js.
  * @constructor
  * @extends {testing.Test}
  */
 function GoogleNowBackgroundUnitTest () {
@@ skipping 37 matching lines @@
   testTaskPair(STATE_CHANGED_TASK_NAME, STATE_CHANGED_TASK_NAME, false);
 });
 
 /**
  * Mocks global functions and APIs that initialize() depends upon.
  * @param {Test} fixture Test fixture.
  */
 function mockInitializeDependencies(fixture) {
   fixture.makeAndRegisterMockGlobals([
     'recordEvent',
+    'setBackgroundEnable',
     'showWelcomeToast',
     'startPollingCards'
   ]);
   fixture.makeAndRegisterMockApis([
     'authenticationManager.isSignedIn',
     'chrome.location.clearWatch',
     'chrome.storage.local.set',
+    'instrumented.metricsPrivate.getFieldTrial',
     'instrumented.notifications.getAll',
     '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} mockApisObj Mock APIs Object.
+ * @param {object} fixture Test fixture.
  * @param {string} testIdentityToken getAuthToken callback token.
  * @param {boolean} testGeolocationPref Geolocation Preference callback value.
  * @param {boolean} testUserRespondedToToast User Response to toast
  &     callback value.
  */
 function expectStateMachineCalls(
-    mockApisObj,
+    fixture,
     testIdentityToken,
     testGeolocationPref,
-    testUserRespondedToToast) {
+    testUserRespondedToToast,
+    testExperimentResponse) {
   var authenticationManagerIsSignedInSavedArgs = new SaveMockArguments();
-  mockApisObj.expects(once()).
+  fixture.mockApis.expects(once()).
       authenticationManager_isSignedIn(
           authenticationManagerIsSignedInSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           authenticationManagerIsSignedInSavedArgs,
           0,
           testIdentityToken));
 
+  var getFieldTrialSavedArgs = new SaveMockArguments();
+  fixture.mockApis.expects(once()).
+      instrumented_metricsPrivate_getFieldTrial(
+          getFieldTrialSavedArgs.match(ANYTHING),
+          getFieldTrialSavedArgs.match(ANYTHING)).
+      will(invokeCallback(
+          getFieldTrialSavedArgs, 1, testExperimentResponse));
+
   var googleGeolocationPrefGetSavedArgs = new SaveMockArguments();
-  mockApisObj.expects(once()).
+  fixture.mockApis.expects(once()).
       instrumented_preferencesPrivate_googleGeolocationAccessEnabled_get(
           googleGeolocationPrefGetSavedArgs.match(eqJSON({})),
           googleGeolocationPrefGetSavedArgs.match(ANYTHING)).
       will(invokeCallback(
           googleGeolocationPrefGetSavedArgs, 1, {value: testGeolocationPref}));
 
   var storageGetSavedArgs = new SaveMockArguments();
-  mockApisObj.expects(once()).
+  fixture.mockApis.expects(once()).
       instrumented_storage_local_get(
           storageGetSavedArgs.match(eq('userRespondedToToast')),
           storageGetSavedArgs.match(ANYTHING)).
       will(invokeCallback(storageGetSavedArgs, 1, testUserRespondedToToast));
+
+  var setBackgroundEnableSavedArgs = new SaveMockArguments();
+  fixture.mockGlobals.expects(once()).
+      setBackgroundEnable(
+          setBackgroundEnableSavedArgs.match(ANYTHING),
+          setBackgroundEnableSavedArgs.match(ANYTHING)).
+      will(invokeCallback(
+          setBackgroundEnableSavedArgs, 1, {}));
 }
 
 /**
  * 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.
  */
 function expectInitialization(mockApisObj) {
@@ skipping 24 matching lines @@
     function() {
       // Tests the case when the user isn't signed in and NOTIFICATION_CARDS_URL
       // is not set. Since NOTIFICATION_CARDS_URL is empty,
       // nothing should start.
 
       // Setup and expectations.
       NOTIFICATION_CARDS_URL = undefined;
       var testIdentityToken = undefined;
       var testGeolocationPref = false;
       var testUserRespondedToToast = {};
+      var testExperimentResponse = '';
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.STOPPED);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
-          this.mockApis,
+          this,
           testIdentityToken,
           testGeolocationPref,
-          testUserRespondedToToast);
+          testUserRespondedToToast,
+          testExperimentResponse);
 
       var chromeNotificationGetAllSavedArgs = new SaveMockArguments();
       this.mockApis.expects(exactly(2)).
           instrumented_notifications_getAll(
               chromeNotificationGetAllSavedArgs.match(ANYTHING)).
           will(
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}),
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}));
 
       // TODO(robliao,vadimt): Determine the granularity of testing to perform.
 
       // Invoking the tested function.
       initialize();
     });
 
 TEST_F(
     'GoogleNowBackgroundUnitTest',
     'Initialize_ToastStateEmpty2',
     function() {
       // Tests the case when NOTIFICATION_CARDS_URL is but 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.
       NOTIFICATION_CARDS_URL = 'https://some.server.url.com';
       var testIdentityToken = undefined;
       var testGeolocationPref = false;
       var testUserRespondedToToast = {};
+      var testExperimentResponse = '';
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.STOPPED);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
-          this.mockApis,
+          this,
           testIdentityToken,
           testGeolocationPref,
-          testUserRespondedToToast);
+          testUserRespondedToToast,
+          testExperimentResponse);
 
       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.
       initialize();
     });
 
 TEST_F(
     'GoogleNowBackgroundUnitTest',
     'Initialize_ToastStateEmpty3',
     function() {
       // Tests the case when NOTIFICATION_CARDS_URL is set, getAuthToken
       // succeeds, and the user has never responded to the toast.
       // In this case, the function should invoke showWelcomeToast().
 
       // Setup and expectations.
       NOTIFICATION_CARDS_URL = 'https://some.server.url.com';
       var testIdentityToken = 'some identity token';
       var testGeolocationPref = false;
       var testUserRespondedToToast = {};
+      var testExperimentResponse = '';
 
       mockInitializeDependencies(this);
 
       this.mockGlobals.expects(once()).recordEvent(
           GoogleNowEvent.EXTENSION_START);
 
       expectInitialization(this.mockApis);
 
       expectStateMachineCalls(
-          this.mockApis,
+          this,
           testIdentityToken,
           testGeolocationPref,
-          testUserRespondedToToast);
+          testUserRespondedToToast,
+          testExperimentResponse);
 
       var chromeNotificationGetAllSavedArgs = new SaveMockArguments();
       this.mockApis.expects(exactly(2)).
           instrumented_notifications_getAll(
               chromeNotificationGetAllSavedArgs.match(ANYTHING)).
           will(
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}),
               invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}));
 
       this.mockGlobals.expects(once()).showWelcomeToast();
 
       // 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.
   NOTIFICATION_CARDS_URL = 'https://some.server.url.com';
   var testIdentityToken = 'some identity token';
   var testGeolocationPref = true;
   var testUserRespondedToToast = {userRespondedToToast: true};
+  var testExperimentResponse = '';
 
   mockInitializeDependencies(this);
 
   this.mockGlobals.expects(once()).recordEvent(
         GoogleNowEvent.EXTENSION_START);
 
   expectInitialization(this.mockApis);
 
   expectStateMachineCalls(
-      this.mockApis,
+      this,
       testIdentityToken,
       testGeolocationPref,
-      testUserRespondedToToast);
+      testUserRespondedToToast,
+      testExperimentResponse);
 
   var chromeNotificationGetAllSavedArgs = new SaveMockArguments();
   this.mockApis.expects(exactly(2)).
       instrumented_notifications_getAll(
           chromeNotificationGetAllSavedArgs.match(ANYTHING)).
       will(
           invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}),
           invokeCallback(chromeNotificationGetAllSavedArgs, 0, {}));
 
   this.mockGlobals.expects(once()).startPollingCards();
 
   // Invoking the tested function.
   initialize();
 });
 
 TEST_F('GoogleNowBackgroundUnitTest', '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.
   NOTIFICATION_CARDS_URL = 'https://some.server.url.com';
   var testIdentityToken = 'some identity token';
   var testGeolocationPref = false;
   var testUserRespondedToToast = {userRespondedToToast: true};
+  var testExperimentResponse = '';
 
   mockInitializeDependencies(this);
 
   this.mockGlobals.expects(once()).recordEvent(
         GoogleNowEvent.EXTENSION_START);
 
   this.mockGlobals.expects(once()).recordEvent(
       GoogleNowEvent.USER_SUPPRESSED);
 
   expectInitialization(this.mockApis);
 
   expectStateMachineCalls(
-      this.mockApis,
+      this,
       testIdentityToken,
       testGeolocationPref,
-      testUserRespondedToToast);
+      testUserRespondedToToast,
+      testExperimentResponse);
 
   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 172 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}));
 
       // Invoking the tested function.
       onNotificationClicked(
           testNotificationId, this.mockLocalFunctions.functions().selector);
     });