Index: chrome/browser/ui/intents/intents_model_unittest.cc |
=================================================================== |
--- chrome/browser/ui/intents/intents_model_unittest.cc (revision 97964) |
+++ chrome/browser/ui/intents/intents_model_unittest.cc (working copy) |
@@ -1,185 +0,0 @@ |
-// 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(®istry_); |
- 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(®istry_); |
- 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()); |
-} |