Add UI to the Clear Browsing Data dialog to remove Flash LSO data.

chrome/browser/ui/cocoa/clear_browsing_data_controller_unittest.mm

 // Copyright (c) 2010 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.
 
 #import <Cocoa/Cocoa.h>
 
 #include "base/scoped_nsobject.h"
 #include "chrome/browser/browsing_data_remover.h"
 #include "chrome/browser/prefs/pref_service.h"
+#include "chrome/browser/prefs/pref_service_mock_builder.h"
+#include "chrome/browser/prefs/testing_pref_store.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/cocoa/browser_test_helper.h"
 #import "chrome/browser/ui/cocoa/clear_browsing_data_controller.h"
 #import "chrome/browser/ui/cocoa/cocoa_test_helper.h"
 #include "chrome/common/pref_names.h"
+#include "chrome/test/testing_browser_process.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 #import "third_party/ocmock/OCMock/OCMock.h"
 
 namespace {
 
 class ClearBrowsingDataControllerTest : public CocoaTest {
  public:
   virtual void SetUp() {
     CocoaTest::SetUp();
+
     // Set up some interesting prefs:
     PrefService* prefs = helper_.profile()->GetPrefs();
     prefs->SetBoolean(prefs::kDeleteBrowsingHistory, true);
     prefs->SetBoolean(prefs::kDeleteDownloadHistory, false);
     prefs->SetBoolean(prefs::kDeleteCache, true);
     prefs->SetBoolean(prefs::kDeleteCookies, false);
     prefs->SetBoolean(prefs::kDeletePasswords, true);
     prefs->SetBoolean(prefs::kDeleteFormData, false);
+    prefs->SetBoolean(prefs::kDeleteLSOData, true);
     prefs->SetInteger(prefs::kDeleteTimePeriod,
                       BrowsingDataRemover::FOUR_WEEKS);
     controller_ =
         [ClearBrowsingDataController controllerForProfile:helper_.profile()];
   }
 
   virtual void TearDown() {
+    MessageLoop::current()->RunAllPending();
+
     [controller_ closeDialog];
     CocoaTest::TearDown();
   }
 
   BrowserTestHelper helper_;
   ClearBrowsingDataController* controller_;
 };
 
 TEST_F(ClearBrowsingDataControllerTest, InitialState) {
   // Check properties match the prefs set above:
   EXPECT_TRUE([controller_ clearBrowsingHistory]);
   EXPECT_FALSE([controller_ clearDownloadHistory]);
   EXPECT_TRUE([controller_ emptyCache]);
   EXPECT_FALSE([controller_ deleteCookies]);
   EXPECT_TRUE([controller_ clearSavedPasswords]);
   EXPECT_FALSE([controller_ clearFormData]);
+  EXPECT_TRUE([controller_ clearLSOData]);
   EXPECT_EQ(BrowsingDataRemover::FOUR_WEEKS,
             [controller_ timePeriod]);
 }
 
+TEST_F(ClearBrowsingDataControllerTest, ClearLSOData) {
+  PrefService* prefs = helper_.profile()->GetPrefs();
+
+  // Disable clearing LSO data.
+  prefs->SetBoolean(prefs::kClearPluginLSODataEnabled, false);
+  EXPECT_TRUE([controller_ clearLSOData]);
+  EXPECT_FALSE([controller_ clearLSODataEnabled]);
+  EXPECT_FALSE([controller_ clearLSODataChecked]);
+
+  // Enabling it again should restore the old value of the checkbox.
+  prefs->SetBoolean(prefs::kClearPluginLSODataEnabled, true);
+  EXPECT_TRUE([controller_ clearLSODataEnabled]);
+  EXPECT_TRUE([controller_ clearLSODataChecked]);
+
+  // Unchecking the checkbox should set the preference to false.
+  [controller_ setClearLSODataChecked:NO];
+  [controller_ persistToPrefs];
+  EXPECT_FALSE(prefs->GetBoolean(prefs::kDeleteLSOData));
+}
+
 TEST_F(ClearBrowsingDataControllerTest, InitialRemoveMask) {
   // Check that the remove-mask matches the initial properties:
   EXPECT_EQ(BrowsingDataRemover::REMOVE_HISTORY |
                 BrowsingDataRemover::REMOVE_CACHE |
-                BrowsingDataRemover::REMOVE_PASSWORDS,
+                BrowsingDataRemover::REMOVE_PASSWORDS |
+                BrowsingDataRemover::REMOVE_LSO_DATA,
             [controller_ removeMask]);
 }
 
 TEST_F(ClearBrowsingDataControllerTest, ModifiedRemoveMask) {
   // Invert all properties and check that the remove-mask is still correct:
   [controller_ setClearBrowsingHistory:false];
   [controller_ setClearDownloadHistory:true];
   [controller_ setEmptyCache:false];
   [controller_ setDeleteCookies:true];
   [controller_ setClearSavedPasswords:false];
   [controller_ setClearFormData:true];
+  [controller_ setClearLSOData:false];
 
   EXPECT_EQ(BrowsingDataRemover::REMOVE_DOWNLOADS |
                 BrowsingDataRemover::REMOVE_COOKIES |
                 BrowsingDataRemover::REMOVE_FORM_DATA,
             [controller_ removeMask]);
 }
 
 TEST_F(ClearBrowsingDataControllerTest, EmptyRemoveMask) {
   // Clear all properties and check that the remove-mask is zero:
   [controller_ setClearBrowsingHistory:false];
   [controller_ setClearDownloadHistory:false];
   [controller_ setEmptyCache:false];
   [controller_ setDeleteCookies:false];
   [controller_ setClearSavedPasswords:false];
   [controller_ setClearFormData:false];
+  [controller_ setClearLSOData:false];
 
-  EXPECT_EQ(0,
-            [controller_ removeMask]);
+  EXPECT_EQ(0, [controller_ removeMask]);
 }
 
 TEST_F(ClearBrowsingDataControllerTest, PersistToPrefs) {
   // Change some settings and store to prefs:
   [controller_ setClearBrowsingHistory:false];
   [controller_ setClearDownloadHistory:true];
   [controller_ persistToPrefs];
 
   // Test that the modified settings were stored to prefs:
   PrefService* prefs = helper_.profile()->GetPrefs();
@@ skipping 36 matching lines @@
                     object:controller
                   userInfo:expectedInfo];
 
   // This calls |-closeDialog| and cleans the controller up.
   [controller dataRemoverDidFinish];
 
   [observer verify];
 }
 
 }  // namespace