| OLD | NEW | 
 | (Empty) | 
|    1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |  | 
|    2 // Use of this source code is governed by a BSD-style license that can be |  | 
|    3 // found in the LICENSE file. |  | 
|    4  |  | 
|    5 #include "base/file_util.h" |  | 
|    6 #include "base/scoped_temp_dir.h" |  | 
|    7 #include "base/synchronization/waitable_event.h" |  | 
|    8 #include "base/values.h" |  | 
|    9 #include "base/utf_string_conversions.h" |  | 
|   10 #include "chrome/browser/intents/web_intents_registry.h" |  | 
|   11 #include "chrome/browser/webdata/web_data_service.h" |  | 
|   12 #include "chrome/test/base/testing_browser_process_test.h" |  | 
|   13 #include "chrome/browser/ui/intents/intents_model.h" |  | 
|   14 #include "testing/gtest/include/gtest/gtest.h" |  | 
|   15 #include "ui/base/models/tree_node_model.h" |  | 
|   16  |  | 
|   17 class IntentsModelTest : public TestingBrowserProcessTest { |  | 
|   18  public: |  | 
|   19   IntentsModelTest() |  | 
|   20       : ui_thread_(BrowserThread::UI, &message_loop_), |  | 
|   21         db_thread_(BrowserThread::DB) {} |  | 
|   22  |  | 
|   23  protected: |  | 
|   24   virtual void SetUp() { |  | 
|   25     db_thread_.Start(); |  | 
|   26     wds_ = new WebDataService(); |  | 
|   27     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |  | 
|   28     wds_->Init(temp_dir_.path()); |  | 
|   29  |  | 
|   30     registry_.Initialize(wds_); |  | 
|   31   } |  | 
|   32  |  | 
|   33   virtual void TearDown() { |  | 
|   34     if (wds_.get()) |  | 
|   35       wds_->Shutdown(); |  | 
|   36  |  | 
|   37     db_thread_.Stop(); |  | 
|   38     MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask); |  | 
|   39     MessageLoop::current()->Run(); |  | 
|   40   } |  | 
|   41  |  | 
|   42   void LoadRegistry() { |  | 
|   43     { |  | 
|   44       WebIntentData provider; |  | 
|   45       provider.service_url = GURL("http://www.google.com/share"); |  | 
|   46       provider.action = ASCIIToUTF16("SHARE"); |  | 
|   47       provider.type = ASCIIToUTF16("text/url"); |  | 
|   48       provider.title = ASCIIToUTF16("Google"); |  | 
|   49       registry_.RegisterIntentProvider(provider); |  | 
|   50     } |  | 
|   51     { |  | 
|   52       WebIntentData provider; |  | 
|   53       provider.service_url = GURL("http://picasaweb.google.com/share"); |  | 
|   54       provider.action = ASCIIToUTF16("EDIT"); |  | 
|   55       provider.type = ASCIIToUTF16("image/*"); |  | 
|   56       provider.title = ASCIIToUTF16("Picasa"); |  | 
|   57       registry_.RegisterIntentProvider(provider); |  | 
|   58     } |  | 
|   59     { |  | 
|   60       WebIntentData provider; |  | 
|   61       provider.service_url = GURL("http://www.digg.com/share"); |  | 
|   62       provider.action = ASCIIToUTF16("SHARE"); |  | 
|   63       provider.type = ASCIIToUTF16("text/url"); |  | 
|   64       provider.title = ASCIIToUTF16("Digg"); |  | 
|   65       registry_.RegisterIntentProvider(provider); |  | 
|   66     } |  | 
|   67   } |  | 
|   68  |  | 
|   69   MessageLoopForUI message_loop_; |  | 
|   70   BrowserThread ui_thread_; |  | 
|   71   BrowserThread db_thread_; |  | 
|   72   scoped_refptr<WebDataService> wds_; |  | 
|   73   WebIntentsRegistry registry_; |  | 
|   74   ScopedTempDir temp_dir_; |  | 
|   75 }; |  | 
|   76  |  | 
|   77 class WaitingIntentsObserver : public IntentsModel::Observer { |  | 
|   78  public: |  | 
|   79   WaitingIntentsObserver() : event_(true, false), added_(0) {} |  | 
|   80  |  | 
|   81   virtual void TreeModelBeginBatch(IntentsModel* model) {} |  | 
|   82  |  | 
|   83   virtual void TreeModelEndBatch(IntentsModel* model) { |  | 
|   84     event_.Signal(); |  | 
|   85      MessageLoop::current()->Quit(); |  | 
|   86   } |  | 
|   87  |  | 
|   88   virtual void TreeNodesAdded(ui::TreeModel* model, |  | 
|   89                               ui::TreeModelNode* parent, |  | 
|   90                               int start, |  | 
|   91                               int count) { |  | 
|   92     added_++; |  | 
|   93   } |  | 
|   94  |  | 
|   95   virtual void TreeNodesRemoved(ui::TreeModel* model, |  | 
|   96                                 ui::TreeModelNode* node, |  | 
|   97                                 int start, |  | 
|   98                                 int count) { |  | 
|   99   } |  | 
|  100  |  | 
|  101   virtual void TreeNodeChanged(ui::TreeModel* model, ui::TreeModelNode* node) { |  | 
|  102   } |  | 
|  103  |  | 
|  104   void Wait() { |  | 
|  105     MessageLoop::current()->Run(); |  | 
|  106     event_.Wait(); |  | 
|  107     LOG(INFO) << "DONE!"; |  | 
|  108   } |  | 
|  109  |  | 
|  110   base::WaitableEvent event_; |  | 
|  111   int added_; |  | 
|  112 }; |  | 
|  113  |  | 
|  114 TEST_F(IntentsModelTest, NodeIDs) { |  | 
|  115   LoadRegistry(); |  | 
|  116   WaitingIntentsObserver obs; |  | 
|  117   IntentsModel intents_model(®istry_); |  | 
|  118   intents_model.AddIntentsTreeObserver(&obs); |  | 
|  119   obs.Wait(); |  | 
|  120  |  | 
|  121   IntentsTreeNode* n1 = new IntentsTreeNode(ASCIIToUTF16("origin")); |  | 
|  122   intents_model.Add(intents_model.GetRoot(), n1, |  | 
|  123                     intents_model.GetRoot()->child_count()); |  | 
|  124   EXPECT_EQ(ASCIIToUTF16("origin"), intents_model.GetTreeNodeId(n1)); |  | 
|  125  |  | 
|  126   IntentsTreeNode* ncheck = intents_model.GetTreeNode("origin"); |  | 
|  127   EXPECT_EQ(ncheck, n1); |  | 
|  128  |  | 
|  129   base::ListValue nodes; |  | 
|  130   intents_model.GetChildNodeList( |  | 
|  131       intents_model.GetTreeNode("www.google.com"), 0, 1, &nodes); |  | 
|  132   EXPECT_EQ(static_cast<size_t>(1), nodes.GetSize()); |  | 
|  133   base::DictionaryValue* dict; |  | 
|  134   EXPECT_TRUE(nodes.GetDictionary(0, &dict)); |  | 
|  135  |  | 
|  136   std::string val; |  | 
|  137   EXPECT_TRUE(dict->GetString("site", &val)); |  | 
|  138   EXPECT_EQ("www.google.com", val); |  | 
|  139   EXPECT_TRUE(dict->GetString("name", &val)); |  | 
|  140   EXPECT_EQ("Google", val); |  | 
|  141   EXPECT_TRUE(dict->GetString("url", &val)); |  | 
|  142   EXPECT_EQ("http://www.google.com/share", val); |  | 
|  143   EXPECT_TRUE(dict->GetString("icon", &val)); |  | 
|  144   EXPECT_EQ("http://www.google.com/favicon.ico", val); |  | 
|  145   base::ListValue* types_list; |  | 
|  146   EXPECT_TRUE(dict->GetList("types", &types_list)); |  | 
|  147   EXPECT_EQ(static_cast<size_t>(1), types_list->GetSize()); |  | 
|  148   EXPECT_TRUE(types_list->GetString(0, &val)); |  | 
|  149   EXPECT_EQ("text/url", val); |  | 
|  150   bool bval; |  | 
|  151   EXPECT_TRUE(dict->GetBoolean("blocked", &bval)); |  | 
|  152   EXPECT_FALSE(bval); |  | 
|  153   EXPECT_TRUE(dict->GetBoolean("disabled", &bval)); |  | 
|  154   EXPECT_FALSE(bval); |  | 
|  155 } |  | 
|  156  |  | 
|  157 TEST_F(IntentsModelTest, LoadFromWebData) { |  | 
|  158   LoadRegistry(); |  | 
|  159   WaitingIntentsObserver obs; |  | 
|  160   IntentsModel intents_model(®istry_); |  | 
|  161   intents_model.AddIntentsTreeObserver(&obs); |  | 
|  162   obs.Wait(); |  | 
|  163   EXPECT_EQ(3, obs.added_); |  | 
|  164  |  | 
|  165   IntentsTreeNode* node = intents_model.GetTreeNode("www.google.com"); |  | 
|  166   ASSERT_NE(static_cast<IntentsTreeNode*>(NULL), node); |  | 
|  167   EXPECT_EQ(IntentsTreeNode::TYPE_ORIGIN, node->Type()); |  | 
|  168   EXPECT_EQ(ASCIIToUTF16("www.google.com"), node->GetTitle()); |  | 
|  169   EXPECT_EQ(1, node->child_count()); |  | 
|  170   node = node->GetChild(0); |  | 
|  171   ASSERT_EQ(IntentsTreeNode::TYPE_SERVICE, node->Type()); |  | 
|  172   ServiceTreeNode* snode = static_cast<ServiceTreeNode*>(node); |  | 
|  173   EXPECT_EQ(ASCIIToUTF16("Google"), snode->ServiceName()); |  | 
|  174   EXPECT_EQ(ASCIIToUTF16("SHARE"), snode->Action()); |  | 
|  175   EXPECT_EQ(ASCIIToUTF16("http://www.google.com/share"), snode->ServiceUrl()); |  | 
|  176   EXPECT_EQ(static_cast<size_t>(1), snode->Types().GetSize()); |  | 
|  177   string16 stype; |  | 
|  178   ASSERT_TRUE(snode->Types().GetString(0, &stype)); |  | 
|  179   EXPECT_EQ(ASCIIToUTF16("text/url"), stype); |  | 
|  180  |  | 
|  181   node = intents_model.GetTreeNode("www.digg.com"); |  | 
|  182   ASSERT_NE(static_cast<IntentsTreeNode*>(NULL), node); |  | 
|  183   EXPECT_EQ(IntentsTreeNode::TYPE_ORIGIN, node->Type()); |  | 
|  184   EXPECT_EQ(ASCIIToUTF16("www.digg.com"), node->GetTitle()); |  | 
|  185 } |  | 
| OLD | NEW |