First pass on intents options UI.

chrome/browser/ui/intents/intents_model_unittest.cc

+// Copyright (c) 2011 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.
+
+#include "base/file_util.h"
+#include "base/scoped_temp_dir.h"
+#include "base/synchronization/waitable_event.h"
+#include "base/values.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/intents/web_intents_registry.h"
+#include "chrome/browser/webdata/web_data_service.h"
+#include "chrome/test/base/testing_browser_process_test.h"
+#include "chrome/browser/ui/intents/intents_model.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/models/tree_node_model.h"
+
+class IntentsModelTest : public TestingBrowserProcessTest {
+ public:
+  IntentsModelTest()
+      : ui_thread_(BrowserThread::UI, &message_loop_),
+        db_thread_(BrowserThread::DB) {}
+
+ protected:
+  virtual void SetUp() {
+    db_thread_.Start();
+    wds_ = new WebDataService();
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    wds_->Init(temp_dir_.path());
+
+    registry_.Initialize(wds_);
+  }
+
+  virtual void TearDown() {
+    if (wds_.get())
+      wds_->Shutdown();
+
+    db_thread_.Stop();
+    MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask);
+    MessageLoop::current()->Run();
+  }
+
+  void LoadRegistry() {
+    {
+      WebIntentData provider;
+      provider.service_url = GURL("http://www.google.com/share");
+      provider.action = ASCIIToUTF16("SHARE");
+      provider.type = ASCIIToUTF16("text/url");
+      provider.title = ASCIIToUTF16("Google");
+      registry_.RegisterIntentProvider(provider);
+    }
+    {
+      WebIntentData provider;
+      provider.service_url = GURL("http://picasaweb.google.com/share");
+      provider.action = ASCIIToUTF16("EDIT");
+      provider.type = ASCIIToUTF16("image/*");
+      provider.title = ASCIIToUTF16("Picasa");
+      registry_.RegisterIntentProvider(provider);
+    }
+    {
+      WebIntentData provider;
+      provider.service_url = GURL("http://www.digg.com/share");
+      provider.action = ASCIIToUTF16("SHARE");
+      provider.type = ASCIIToUTF16("text/url");
+      provider.title = ASCIIToUTF16("Digg");
+      registry_.RegisterIntentProvider(provider);
+    }
+  }
+
+  MessageLoopForUI message_loop_;
+  BrowserThread ui_thread_;
+  BrowserThread db_thread_;
+  scoped_refptr<WebDataService> wds_;
+  WebIntentsRegistry registry_;
+  ScopedTempDir temp_dir_;
+};
+
+class WaitingIntentsObserver : public IntentsModel::Observer {
+ public:
+  WaitingIntentsObserver() : event_(true, false), added_(0) {}
+
+  virtual void TreeModelBeginBatch(IntentsModel* model) {}
+
+  virtual void TreeModelEndBatch(IntentsModel* model) {
+    event_.Signal();
+     MessageLoop::current()->Quit();
+  }
+
+  virtual void TreeNodesAdded(ui::TreeModel* model,
+                              ui::TreeModelNode* parent,
+                              int start,
+                              int count) {
+    added_++;
+  }
+
+  virtual void TreeNodesRemoved(ui::TreeModel* model,
+                                ui::TreeModelNode* node,
+                                int start,
+                                int count) {
+  }
+
+  virtual void TreeNodeChanged(ui::TreeModel* model, ui::TreeModelNode* node) {
+  }
+
+  void Wait() {
+    MessageLoop::current()->Run();
+    event_.Wait();
+    LOG(INFO) << "DONE!";
+  }
+
+  base::WaitableEvent event_;
+  int added_;
+};
+
+TEST_F(IntentsModelTest, NodeIDs) {
+  LoadRegistry();
+  WaitingIntentsObserver obs;
+  IntentsModel intents_model(&registry_);
+  intents_model.AddIntentsTreeObserver(&obs);
+  obs.Wait();
+
+  IntentsTreeNode* n1 = new IntentsTreeNode(ASCIIToUTF16("origin"));
+  intents_model.Add(intents_model.GetRoot(), n1,
+                    intents_model.GetRoot()->child_count());
+  EXPECT_EQ(ASCIIToUTF16("origin"), intents_model.GetTreeNodeId(n1));
+
+  IntentsTreeNode* ncheck = intents_model.GetTreeNode("origin");
+  EXPECT_EQ(ncheck, n1);
+
+  base::ListValue nodes;
+  intents_model.GetChildNodeList(
+      intents_model.GetTreeNode("www.google.com"), 0, 1, &nodes);
+  EXPECT_EQ(static_cast<size_t>(1), nodes.GetSize());
+  base::DictionaryValue* dict;
+  EXPECT_TRUE(nodes.GetDictionary(0, &dict));
+
+  std::string val;
+  EXPECT_TRUE(dict->GetString("site", &val));
+  EXPECT_EQ("www.google.com", val);
+  EXPECT_TRUE(dict->GetString("name", &val));
+  EXPECT_EQ("Google", val);
+  EXPECT_TRUE(dict->GetString("url", &val));
+  EXPECT_EQ("http://www.google.com/share", val);
+  EXPECT_TRUE(dict->GetString("icon", &val));
+  EXPECT_EQ("http://www.google.com/favicon.ico", val);
+  base::ListValue* types_list;
+  EXPECT_TRUE(dict->GetList("types", &types_list));
+  EXPECT_EQ(static_cast<size_t>(1), types_list->GetSize());
+  EXPECT_TRUE(types_list->GetString(0, &val));
+  EXPECT_EQ("text/url", val);
+  bool bval;
+  EXPECT_TRUE(dict->GetBoolean("blocked", &bval));
+  EXPECT_FALSE(bval);
+  EXPECT_TRUE(dict->GetBoolean("disabled", &bval));
+  EXPECT_FALSE(bval);
+}
+
+TEST_F(IntentsModelTest, LoadFromWebData) {
+  LoadRegistry();
+  WaitingIntentsObserver obs;
+  IntentsModel intents_model(&registry_);
+  intents_model.AddIntentsTreeObserver(&obs);
+  obs.Wait();
+  EXPECT_EQ(3, obs.added_);
+
+  IntentsTreeNode* node = intents_model.GetTreeNode("www.google.com");
+  ASSERT_NE(static_cast<IntentsTreeNode*>(NULL), node);
+  EXPECT_EQ(IntentsTreeNode::TYPE_ORIGIN, node->Type());
+  EXPECT_EQ(ASCIIToUTF16("www.google.com"), node->GetTitle());
+  EXPECT_EQ(1, node->child_count());
+  node = node->GetChild(0);
+  ASSERT_EQ(IntentsTreeNode::TYPE_SERVICE, node->Type());
+  ServiceTreeNode* snode = static_cast<ServiceTreeNode*>(node);
+  EXPECT_EQ(ASCIIToUTF16("Google"), snode->ServiceName());
+  EXPECT_EQ(ASCIIToUTF16("SHARE"), snode->Action());
+  EXPECT_EQ(ASCIIToUTF16("http://www.google.com/share"), snode->ServiceUrl());
+  EXPECT_EQ(static_cast<size_t>(1), snode->Types().GetSize());
+  string16 stype;
+  ASSERT_TRUE(snode->Types().GetString(0, &stype));
+  EXPECT_EQ(ASCIIToUTF16("text/url"), stype);
+
+  node = intents_model.GetTreeNode("www.digg.com");
+  ASSERT_NE(static_cast<IntentsTreeNode*>(NULL), node);
+  EXPECT_EQ(IntentsTreeNode::TYPE_ORIGIN, node->Type());
+  EXPECT_EQ(ASCIIToUTF16("www.digg.com"), node->GetTitle());
+}