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 "chrome/browser/autocomplete/autocomplete_popup_model.h" | |
6 | |
7 #include "base/utf_string_conversions.h" | |
8 #include "chrome/browser/autocomplete/autocomplete_match.h" | |
9 #include "chrome/browser/search_engines/template_url.h" | |
10 #include "chrome/browser/search_engines/template_url_service.h" | |
11 #include "chrome/browser/search_engines/template_url_service_factory.h" | |
12 #include "chrome/test/base/testing_profile.h" | |
13 #include "testing/gtest/include/gtest/gtest.h" | |
14 | |
15 class AutoCompletePopupModelTest : public testing::Test { | |
16 public: | |
17 AutoCompletePopupModelTest() { | |
18 } | |
19 | |
20 virtual void SetUp(); | |
21 virtual void TearDown(); | |
22 | |
23 protected: | |
24 AutocompleteMatch CreateMatch(const string16& keyword, | |
25 const string16& query_string, | |
26 AutocompleteMatch::Type type); | |
27 void RunTest(const char* input, AutocompleteMatch::Type type, | |
28 const char* keyword); | |
29 | |
30 scoped_ptr<TestingProfile> profile_; | |
31 scoped_ptr<AutocompletePopupModel> model_; | |
32 scoped_ptr<AutocompleteEditModel> edit_model_; | |
33 }; | |
34 | |
35 void AutoCompletePopupModelTest::SetUp() { | |
36 profile_.reset(new TestingProfile()); | |
37 profile_->CreateTemplateURLService(); | |
38 edit_model_.reset(new AutocompleteEditModel(NULL, NULL, profile_.get())); | |
39 model_.reset(new AutocompletePopupModel(NULL, edit_model_.get())); | |
40 | |
41 TemplateURLService* turl_model = | |
42 TemplateURLServiceFactory::GetForProfile(profile_.get()); | |
43 | |
44 turl_model->Load(); | |
45 | |
46 // Reset the default TemplateURL. | |
47 TemplateURL* default_t_url = new TemplateURL(); | |
48 default_t_url->set_keyword(ASCIIToUTF16("t")); | |
49 default_t_url->SetURL("http://defaultturl/{searchTerms}", 0, 0); | |
50 turl_model->Add(default_t_url); | |
51 turl_model->SetDefaultSearchProvider(default_t_url); | |
52 ASSERT_NE(0, default_t_url->id()); | |
53 | |
54 // Create another TemplateURL for KeywordProvider. | |
55 TemplateURL* keyword_t_url = new TemplateURL(); | |
56 keyword_t_url->set_short_name(ASCIIToUTF16("k")); | |
57 keyword_t_url->set_keyword(ASCIIToUTF16("k")); | |
58 keyword_t_url->SetURL("http://keyword/{searchTerms}", 0, 0); | |
59 turl_model->Add(keyword_t_url); | |
60 ASSERT_NE(0, keyword_t_url->id()); | |
61 } | |
62 | |
63 void AutoCompletePopupModelTest::TearDown() { | |
64 profile_.reset(); | |
65 model_.reset(); | |
66 edit_model_.reset(); | |
67 } | |
68 | |
69 AutocompleteMatch AutoCompletePopupModelTest::CreateMatch( | |
70 const string16& keyword, | |
71 const string16& query_string, | |
72 AutocompleteMatch::Type type) { | |
73 AutocompleteMatch match(NULL, 0, false, type); | |
74 match.contents = query_string; | |
75 TemplateURLService* template_url_service = | |
76 TemplateURLServiceFactory::GetForProfile(profile_.get()); | |
77 if (!keyword.empty()) { | |
78 const TemplateURL* template_url = | |
79 template_url_service->GetTemplateURLForKeyword(keyword); | |
80 match.template_url = | |
81 TemplateURL::SupportsReplacement(template_url) ? template_url : NULL; | |
82 } | |
83 if (match.template_url) | |
84 match.fill_into_edit = match.template_url->keyword() + char16(' '); | |
85 else | |
86 match.template_url = template_url_service->GetDefaultSearchProvider(); | |
87 match.fill_into_edit.append(query_string); | |
88 match.transition = keyword.empty() ? | |
89 content::PAGE_TRANSITION_GENERATED : content::PAGE_TRANSITION_KEYWORD; | |
90 return match; | |
91 } | |
92 | |
93 void AutoCompletePopupModelTest::RunTest(const char* input, | |
94 AutocompleteMatch::Type type, | |
95 const char* keyword) { | |
96 string16 keyword16(ASCIIToUTF16(keyword)); | |
97 string16 detected_keyword; | |
98 EXPECT_FALSE(model_->GetKeywordForMatch( | |
99 CreateMatch(keyword16, ASCIIToUTF16(input), type), &detected_keyword)); | |
100 EXPECT_EQ(keyword16, detected_keyword); | |
101 } | |
102 | |
103 TEST_F(AutoCompletePopupModelTest, GetKeywordForMatch) { | |
104 string16 keyword; | |
105 | |
106 // Possible matches when the input is "tfoo" | |
107 RunTest("tfoo", AutocompleteMatch::SEARCH_WHAT_YOU_TYPED, ""); | |
108 RunTest("tfoo", AutocompleteMatch::SEARCH_HISTORY, ""); | |
109 RunTest("tfoo", AutocompleteMatch::SEARCH_SUGGEST, ""); | |
110 | |
111 // Possible matches when the input is "t foo" | |
112 RunTest("foo", AutocompleteMatch::SEARCH_HISTORY, "t"); | |
113 RunTest("foo", AutocompleteMatch::SEARCH_OTHER_ENGINE, "t"); | |
114 | |
115 // Possible matches when the input is "k foo" | |
116 RunTest("foo", AutocompleteMatch::SEARCH_HISTORY, "k"); | |
117 RunTest("foo", AutocompleteMatch::SEARCH_OTHER_ENGINE, "k"); | |
118 } | |
OLD | NEW |