Adds local storage nodes to cookie tree model and cookies view.

chrome/browser/cocoa/cookies_window_controller_unittest.mm

 // Copyright (c) 2009-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 "app/tree_model.h"
 #import "base/scoped_nsobject.h"
 #include "base/scoped_ptr.h"
 #include "chrome/browser/cocoa/browser_test_helper.h"
 #include "chrome/browser/browsing_data_remover.h"
 #include "chrome/browser/cocoa/clear_browsing_data_controller.h"
 #import "chrome/browser/cocoa/cookies_window_controller.h"
 #include "chrome/browser/cocoa/cocoa_test_helper.h"
+#include "chrome/browser/mock_browsing_data_local_storage_helper.h"
 #include "chrome/browser/net/url_request_context_getter.h"
 #include "chrome/browser/cookies_tree_model.h"
 #include "chrome/test/testing_profile.h"
 #include "googleurl/src/gurl.h"
 #include "net/url_request/url_request_context.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 #import "third_party/ocmock/OCMock/OCMock.h"
 
 // Used to test FindCocoaNode. This only sets the title and node, without
@@ skipping 33 matching lines @@
 @end
 
 namespace {
 
 class CookiesWindowControllerTest : public CocoaTest {
  public:
   virtual void SetUp() {
     CocoaTest::SetUp();
     TestingProfile* profile = browser_helper_.profile();
     profile->CreateRequestContext();
+    local_storage_helper_ = new MockBrowsingDataLocalStorageHelper(profile);
     controller_.reset(
-        [[CookiesWindowController alloc] initWithProfile:profile]);
+        [[CookiesWindowController alloc] initWithProfile:profile
+                                           storageHelper:local_storage_helper_]
+    );
   }
 
   virtual void TearDown() {
     CocoaTest::TearDown();
   }
 
   CocoaCookieTreeNode* CocoaNodeFromTreeNode(TreeModelNode* node,
                                              bool recurse) {
     return [controller_ modelObserver]->CocoaNodeFromTreeNode(node, recurse);
   }
 
   CocoaCookieTreeNode* FindCocoaNode(TreeModelNode* node,
                                      CocoaCookieTreeNode* start) {
     return [controller_ modelObserver]->FindCocoaNode(node, start);
   }
 
  protected:
   BrowserTestHelper browser_helper_;
   scoped_nsobject<CookiesWindowController> controller_;
+  BrowsingDataLocalStorageHelper* local_storage_helper_;
 };
 
 TEST_F(CookiesWindowControllerTest, Construction) {
   std::vector<SkBitmap> skia_icons;
   [controller_ treeModel]->GetIcons(&skia_icons);
 
   EXPECT_EQ([[controller_ icons] count], skia_icons.size() + 1U);
 }
 
 TEST_F(CookiesWindowControllerTest, FindCocoaNodeRoot) {
@@ skipping 31 matching lines @@
       [[FakeCocoaCookieTreeNode alloc] initWithTreeNode:child2.get()]);
   [[cocoaParent childs] addObject:cocoaChild1.get()];
   [[cocoaChild1 childs] addObject:cocoaChild2.get()];
 
   EXPECT_EQ(cocoaChild2.get(), FindCocoaNode(child2.get(), cocoaParent.get()));
 }
 
 TEST_F(CookiesWindowControllerTest, CocoaNodeFromTreeNodeCookie) {
   net::CookieMonster* cm = browser_helper_.profile()->GetCookieMonster();
   cm->SetCookie(GURL("http://foo.com"), "A=B");
-  CookiesTreeModel model(browser_helper_.profile());
+  CookiesTreeModel model(browser_helper_.profile(), local_storage_helper_);
 
   // Root --> foo.com --> Cookies --> A. Create node for 'A'.
   TreeModelNode* node = model.GetRoot()->GetChild(0)->GetChild(0)->GetChild(0);
   CocoaCookieTreeNode* cookie = CocoaNodeFromTreeNode(node, false);
 
   EXPECT_TRUE([@"B" isEqualToString:[cookie content]]);
   EXPECT_TRUE([@"End of the session" isEqualToString:[cookie expires]]);
   EXPECT_TRUE([@"Any kind of connection" isEqualToString:[cookie sendFor]]);
   EXPECT_TRUE([@"A" isEqualToString:[cookie title]]);
   EXPECT_TRUE([@"A" isEqualToString:[cookie name]]);
   EXPECT_TRUE([@"/" isEqualToString:[cookie path]]);
   EXPECT_EQ(0U, [[cookie childs] count]);
   EXPECT_TRUE([cookie created]);
   EXPECT_TRUE([cookie isLeaf]);
   EXPECT_EQ(node, [cookie treeNode]);
 }
 
 TEST_F(CookiesWindowControllerTest, CocoaNodeFromTreeNodeRecursive) {
   net::CookieMonster* cm = browser_helper_.profile()->GetCookieMonster();
   cm->SetCookie(GURL("http://foo.com"), "A=B");
-  CookiesTreeModel model(browser_helper_.profile());
+  CookiesTreeModel model(browser_helper_.profile(), local_storage_helper_);
 
   // Root --> foo.com --> Cookies --> A. Create node for 'foo.com'.
   CookieTreeNode* node = model.GetRoot()->GetChild(0);
   CocoaCookieTreeNode* domain = CocoaNodeFromTreeNode(node, true);
   CocoaCookieTreeNode* cookies = [[domain childs] objectAtIndex:0];
   CocoaCookieTreeNode* cookie = [[cookies childs] objectAtIndex:0];
 
   // Test domain-level node.
   EXPECT_TRUE([@"foo.com" isEqualToString:[domain title]]);
 
@@ skipping 21 matching lines @@
   EXPECT_EQ(node->GetChild(0)->GetChild(0), [cookie treeNode]);
 }
 
 TEST_F(CookiesWindowControllerTest, TreeNodesAdded) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
 
   controller_.reset(
-      [[CookiesWindowController alloc] initWithProfile:profile]);
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_]);
 
   // Root --> foo.com --> Cookies.
   NSMutableArray* cocoa_children =
       [[[[[[controller_ cocoaTreeModel] childs] objectAtIndex:0]
           childs] objectAtIndex:0] childs];
   EXPECT_EQ(1U, [cocoa_children count]);
 
   // Create some cookies.
   cm->SetCookie(url, "C=D");
   cm->SetCookie(url, "E=F");
@@ skipping 20 matching lines @@
 
 TEST_F(CookiesWindowControllerTest, TreeNodesRemoved) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
   cm->SetCookie(url, "C=D");
   cm->SetCookie(url, "E=F");
 
   controller_.reset(
-      [[CookiesWindowController alloc] initWithProfile:profile]);
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_]);
 
   // Root --> foo.com --> Cookies.
   NSMutableArray* cocoa_children =
       [[[[[[controller_ cocoaTreeModel] childs] objectAtIndex:0]
           childs] objectAtIndex:0] childs];
   EXPECT_EQ(3U, [cocoa_children count]);
 
   CookiesTreeModel* model = [controller_ treeModel];
   // Root --> foo.com --> Cookies.
   CookieTreeNode* parent = model->GetRoot()->GetChild(0)->GetChild(0);
 
   // Pretend to remove the nodes.
   [controller_ modelObserver]->TreeNodesRemoved(model, parent, 1, 2);
 
   EXPECT_EQ(1U, [cocoa_children count]);
 
   NSString* title = [[cocoa_children objectAtIndex:0] name];
   EXPECT_TRUE([@"A" isEqualToString:title]);
 }
 
 TEST_F(CookiesWindowControllerTest, TreeNodeChildrenReordered) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
   cm->SetCookie(url, "C=D");
   cm->SetCookie(url, "E=F");
 
   controller_.reset(
-      [[CookiesWindowController alloc] initWithProfile:profile]);
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_]);
 
   // Root --> foo.com --> Cookies.
   NSMutableArray* cocoa_children =
       [[[[[[controller_ cocoaTreeModel] childs] objectAtIndex:0]
           childs] objectAtIndex:0] childs];
   EXPECT_EQ(3U, [cocoa_children count]);
 
   // Check default ordering.
   CocoaCookieTreeNode* node = [cocoa_children objectAtIndex:0];
   EXPECT_TRUE([@"A" isEqualToString:[node name]]);
@@ skipping 24 matching lines @@
   EXPECT_TRUE([@"C" isEqualToString:[node name]]);
 }
 
 TEST_F(CookiesWindowControllerTest, TreeNodeChanged) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
 
   controller_.reset(
-      [[CookiesWindowController alloc] initWithProfile:profile]);
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_]);
 
   CookiesTreeModel* model = [controller_ treeModel];
   // Root --> foo.com --> Cookies.
   CookieTreeNode* node = model->GetRoot()->GetChild(0)->GetChild(0);
 
   // Root --> foo.com --> Cookies.
   CocoaCookieTreeNode* cocoa_node =
       [[[[[controller_ cocoaTreeModel] childs] objectAtIndex:0]
           childs] objectAtIndex:0];
 
@@ skipping 11 matching lines @@
 TEST_F(CookiesWindowControllerTest, TestDeleteCookie) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
   cm->SetCookie(url, "C=D");
 
   // This will clean itself up when we call |-closeSheet:|. If we reset the
   // scoper, we'd get a double-free.
   CookiesWindowController* controller =
-      [[CookiesWindowController alloc] initWithProfile:profile];
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_];
   [controller attachSheetTo:test_window()];
   NSTreeController* treeController = [controller treeController];
 
   // Pretend to select cookie A.
   NSUInteger path[3] = {0, 0, 0};
   NSIndexPath* indexPath = [NSIndexPath indexPathWithIndexes:path length:3];
   [treeController setSelectionIndexPath:indexPath];
 
   // Press the "Delete" button.
   [controller deleteCookie:nil];
 
   // Root --> foo.com --> Cookies.
   NSArray* cookies = [[[[[[controller cocoaTreeModel] childs] objectAtIndex:0]
       childs] objectAtIndex:0] childs];
   EXPECT_EQ(1U, [cookies count]);
   EXPECT_TRUE([@"C" isEqualToString:[[cookies lastObject] title]]);
   EXPECT_TRUE([indexPath isEqual:[treeController selectionIndexPath]]);
 
   [controller closeSheet:nil];
 }
 
 TEST_F(CookiesWindowControllerTest, TestDidExpandItem) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
   cm->SetCookie(url, "C=D");
 
   controller_.reset(
-      [[CookiesWindowController alloc] initWithProfile:profile]);
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_]);
 
   // Root --> foo.com.
   CocoaCookieTreeNode* foo =
       [[[controller_ cocoaTreeModel] childs] objectAtIndex:0];
 
   // Create the objects we are going to be testing with.
   id outlineView = [OCMockObject mockForClass:[NSOutlineView class]];
   id treeNode = [OCMockObject mockForClass:[NSTreeNode class]];
   NSTreeNode* childTreeNode =
       [NSTreeNode treeNodeWithRepresentedObject:[[foo childs] lastObject]];
@@ skipping 45 matching lines @@
 TEST_F(CookiesWindowControllerTest, RemoveButtonEnabled) {
   const GURL url = GURL("http://foo.com");
   TestingProfile* profile = browser_helper_.profile();
   net::CookieMonster* cm = profile->GetCookieMonster();
   cm->SetCookie(url, "A=B");
   cm->SetCookie(url, "C=D");
 
   // This will clean itself up when we call |-closeSheet:|. If we reset the
   // scoper, we'd get a double-free.
   CookiesWindowController* controller =
-      [[CookiesWindowController alloc] initWithProfile:profile];
+      [[CookiesWindowController alloc] initWithProfile:profile
+                                         storageHelper:local_storage_helper_];
   [controller attachSheetTo:test_window()];
 
   // Nothing should be selected right now.
   EXPECT_FALSE([controller removeButtonEnabled]);
 
   {
     // Pretend to select cookie A.
     NSUInteger path[3] = {0, 0, 0};
     NSIndexPath* indexPath = [NSIndexPath indexPathWithIndexes:path length:3];
     [[controller treeController] setSelectionIndexPath:indexPath];
@@ skipping 25 matching lines @@
     NSIndexPath* indexPath = [NSIndexPath indexPathWithIndexes:path length:3];
     [[controller treeController] setSelectionIndexPath:indexPath];
     [controller outlineViewSelectionDidChange:nil];
     EXPECT_FALSE([controller removeButtonEnabled]);
   }
 
   [controller closeSheet:nil];
 }
 
 }  // namespace