| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/omnibox/browser/autocomplete_provider.h" | 5 #include "components/omnibox/browser/autocomplete_provider.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include <memory> | 9 #include <memory> |
| 10 | 10 |
| (...skipping 284 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 295 EXPECT_TRUE(client_owned_); | 295 EXPECT_TRUE(client_owned_); |
| 296 } | 296 } |
| 297 | 297 |
| 298 void AutocompleteProviderTest::RegisterTemplateURL( | 298 void AutocompleteProviderTest::RegisterTemplateURL( |
| 299 const base::string16 keyword, | 299 const base::string16 keyword, |
| 300 const std::string& template_url) { | 300 const std::string& template_url) { |
| 301 TemplateURLData data; | 301 TemplateURLData data; |
| 302 data.SetURL(template_url); | 302 data.SetURL(template_url); |
| 303 data.SetShortName(keyword); | 303 data.SetShortName(keyword); |
| 304 data.SetKeyword(keyword); | 304 data.SetKeyword(keyword); |
| 305 TemplateURL* default_t_url = new TemplateURL(data); | |
| 306 TemplateURLService* turl_model = client_->GetTemplateURLService(); | 305 TemplateURLService* turl_model = client_->GetTemplateURLService(); |
| 307 turl_model->Add(default_t_url); | 306 TemplateURL* default_turl = |
| 308 turl_model->SetUserSelectedDefaultSearchProvider(default_t_url); | 307 turl_model->Add(base::MakeUnique<TemplateURL>(data)); |
| 308 turl_model->SetUserSelectedDefaultSearchProvider(default_turl); |
| 309 turl_model->Load(); | 309 turl_model->Load(); |
| 310 TemplateURLID default_provider_id = default_t_url->id(); | 310 TemplateURLID default_provider_id = default_turl->id(); |
| 311 ASSERT_NE(0, default_provider_id); | 311 ASSERT_NE(0, default_provider_id); |
| 312 } | 312 } |
| 313 | 313 |
| 314 void AutocompleteProviderTest::ResetControllerWithTestProviders( | 314 void AutocompleteProviderTest::ResetControllerWithTestProviders( |
| 315 bool same_destinations, | 315 bool same_destinations, |
| 316 TestProvider** provider1_ptr, | 316 TestProvider** provider1_ptr, |
| 317 TestProvider** provider2_ptr) { | 317 TestProvider** provider2_ptr) { |
| 318 // TODO: Move it outside this method, after refactoring the existing | 318 // TODO: Move it outside this method, after refactoring the existing |
| 319 // unit tests. Specifically: | 319 // unit tests. Specifically: |
| 320 // (1) Make sure that AutocompleteMatch.keyword is set iff there is | 320 // (1) Make sure that AutocompleteMatch.keyword is set iff there is |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 360 if (provider2_ptr) | 360 if (provider2_ptr) |
| 361 *provider2_ptr = provider2; | 361 *provider2_ptr = provider2; |
| 362 } | 362 } |
| 363 | 363 |
| 364 void AutocompleteProviderTest::ResetControllerWithKeywordAndSearchProviders() { | 364 void AutocompleteProviderTest::ResetControllerWithKeywordAndSearchProviders() { |
| 365 // Reset the default TemplateURL. | 365 // Reset the default TemplateURL. |
| 366 TemplateURLData data; | 366 TemplateURLData data; |
| 367 data.SetShortName(base::ASCIIToUTF16("default")); | 367 data.SetShortName(base::ASCIIToUTF16("default")); |
| 368 data.SetKeyword(base::ASCIIToUTF16("default")); | 368 data.SetKeyword(base::ASCIIToUTF16("default")); |
| 369 data.SetURL("http://defaultturl/{searchTerms}"); | 369 data.SetURL("http://defaultturl/{searchTerms}"); |
| 370 TemplateURL* default_t_url = new TemplateURL(data); | |
| 371 TemplateURLService* turl_model = client_->GetTemplateURLService(); | 370 TemplateURLService* turl_model = client_->GetTemplateURLService(); |
| 372 turl_model->Add(default_t_url); | 371 TemplateURL* default_turl = |
| 373 turl_model->SetUserSelectedDefaultSearchProvider(default_t_url); | 372 turl_model->Add(base::MakeUnique<TemplateURL>(data)); |
| 374 TemplateURLID default_provider_id = default_t_url->id(); | 373 turl_model->SetUserSelectedDefaultSearchProvider(default_turl); |
| 374 TemplateURLID default_provider_id = default_turl->id(); |
| 375 ASSERT_NE(0, default_provider_id); | 375 ASSERT_NE(0, default_provider_id); |
| 376 | 376 |
| 377 // Create another TemplateURL for KeywordProvider. | 377 // Create another TemplateURL for KeywordProvider. |
| 378 TemplateURLData data2; | 378 TemplateURLData data2; |
| 379 data2.SetShortName(base::ASCIIToUTF16("k")); | 379 data2.SetShortName(base::ASCIIToUTF16("k")); |
| 380 data2.SetKeyword(base::ASCIIToUTF16("k")); | 380 data2.SetKeyword(base::ASCIIToUTF16("k")); |
| 381 data2.SetURL("http://keyword/{searchTerms}"); | 381 data2.SetURL("http://keyword/{searchTerms}"); |
| 382 TemplateURL* keyword_t_url = new TemplateURL(data2); | 382 TemplateURL* keyword_turl = |
| 383 turl_model->Add(keyword_t_url); | 383 turl_model->Add(base::MakeUnique<TemplateURL>(data2)); |
| 384 ASSERT_NE(0, keyword_t_url->id()); | 384 ASSERT_NE(0, keyword_turl->id()); |
| 385 | 385 |
| 386 ResetControllerWithType(AutocompleteProvider::TYPE_KEYWORD | | 386 ResetControllerWithType(AutocompleteProvider::TYPE_KEYWORD | |
| 387 AutocompleteProvider::TYPE_SEARCH); | 387 AutocompleteProvider::TYPE_SEARCH); |
| 388 } | 388 } |
| 389 | 389 |
| 390 void AutocompleteProviderTest::ResetControllerWithKeywordProvider() { | 390 void AutocompleteProviderTest::ResetControllerWithKeywordProvider() { |
| 391 TemplateURLService* turl_model = client_->GetTemplateURLService(); | 391 TemplateURLService* turl_model = client_->GetTemplateURLService(); |
| 392 | 392 |
| 393 // Create a TemplateURL for KeywordProvider. | 393 // Create a TemplateURL for KeywordProvider. |
| 394 TemplateURLData data; | 394 TemplateURLData data; |
| 395 data.SetShortName(base::ASCIIToUTF16("foo.com")); | 395 data.SetShortName(base::ASCIIToUTF16("foo.com")); |
| 396 data.SetKeyword(base::ASCIIToUTF16("foo.com")); | 396 data.SetKeyword(base::ASCIIToUTF16("foo.com")); |
| 397 data.SetURL("http://foo.com/{searchTerms}"); | 397 data.SetURL("http://foo.com/{searchTerms}"); |
| 398 TemplateURL* keyword_t_url = new TemplateURL(data); | 398 TemplateURL* keyword_turl = |
| 399 turl_model->Add(keyword_t_url); | 399 turl_model->Add(base::MakeUnique<TemplateURL>(data)); |
| 400 ASSERT_NE(0, keyword_t_url->id()); | 400 ASSERT_NE(0, keyword_turl->id()); |
| 401 | 401 |
| 402 // Make a TemplateURL for KeywordProvider that a shorter version of the | 402 // Make a TemplateURL for KeywordProvider that a shorter version of the |
| 403 // first. | 403 // first. |
| 404 data.SetShortName(base::ASCIIToUTF16("f")); | 404 data.SetShortName(base::ASCIIToUTF16("f")); |
| 405 data.SetKeyword(base::ASCIIToUTF16("f")); | 405 data.SetKeyword(base::ASCIIToUTF16("f")); |
| 406 data.SetURL("http://f.com/{searchTerms}"); | 406 data.SetURL("http://f.com/{searchTerms}"); |
| 407 keyword_t_url = new TemplateURL(data); | 407 keyword_turl = turl_model->Add(base::MakeUnique<TemplateURL>(data)); |
| 408 turl_model->Add(keyword_t_url); | 408 ASSERT_NE(0, keyword_turl->id()); |
| 409 ASSERT_NE(0, keyword_t_url->id()); | |
| 410 | 409 |
| 411 // Create another TemplateURL for KeywordProvider. | 410 // Create another TemplateURL for KeywordProvider. |
| 412 data.SetShortName(base::ASCIIToUTF16("bar.com")); | 411 data.SetShortName(base::ASCIIToUTF16("bar.com")); |
| 413 data.SetKeyword(base::ASCIIToUTF16("bar.com")); | 412 data.SetKeyword(base::ASCIIToUTF16("bar.com")); |
| 414 data.SetURL("http://bar.com/{searchTerms}"); | 413 data.SetURL("http://bar.com/{searchTerms}"); |
| 415 keyword_t_url = new TemplateURL(data); | 414 keyword_turl = turl_model->Add(base::MakeUnique<TemplateURL>(data)); |
| 416 turl_model->Add(keyword_t_url); | 415 ASSERT_NE(0, keyword_turl->id()); |
| 417 ASSERT_NE(0, keyword_t_url->id()); | |
| 418 | 416 |
| 419 ResetControllerWithType(AutocompleteProvider::TYPE_KEYWORD); | 417 ResetControllerWithType(AutocompleteProvider::TYPE_KEYWORD); |
| 420 } | 418 } |
| 421 | 419 |
| 422 void AutocompleteProviderTest::ResetControllerWithType(int type) { | 420 void AutocompleteProviderTest::ResetControllerWithType(int type) { |
| 423 EXPECT_FALSE(client_owned_); | 421 EXPECT_FALSE(client_owned_); |
| 424 controller_.reset( | 422 controller_.reset( |
| 425 new AutocompleteController(base::WrapUnique(client_), nullptr, type)); | 423 new AutocompleteController(base::WrapUnique(client_), nullptr, type)); |
| 426 client_owned_ = true; | 424 client_owned_ = true; |
| 427 } | 425 } |
| (...skipping 338 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 766 EXPECT_FALSE(search_provider_field_trial_triggered_in_session()); | 764 EXPECT_FALSE(search_provider_field_trial_triggered_in_session()); |
| 767 url = GetDestinationURL(match, base::TimeDelta::FromMilliseconds(2456)); | 765 url = GetDestinationURL(match, base::TimeDelta::FromMilliseconds(2456)); |
| 768 EXPECT_EQ("//aqs=chrome.0.69i57j69i58j5l2j0l3j69i59.2456j0j4&", url.path()); | 766 EXPECT_EQ("//aqs=chrome.0.69i57j69i58j5l2j0l3j69i59.2456j0j4&", url.path()); |
| 769 | 767 |
| 770 // Test page classification and field trial triggered set. | 768 // Test page classification and field trial triggered set. |
| 771 set_search_provider_field_trial_triggered_in_session(true); | 769 set_search_provider_field_trial_triggered_in_session(true); |
| 772 EXPECT_TRUE(search_provider_field_trial_triggered_in_session()); | 770 EXPECT_TRUE(search_provider_field_trial_triggered_in_session()); |
| 773 url = GetDestinationURL(match, base::TimeDelta::FromMilliseconds(2456)); | 771 url = GetDestinationURL(match, base::TimeDelta::FromMilliseconds(2456)); |
| 774 EXPECT_EQ("//aqs=chrome.0.69i57j69i58j5l2j0l3j69i59.2456j1j4&", url.path()); | 772 EXPECT_EQ("//aqs=chrome.0.69i57j69i58j5l2j0l3j69i59.2456j1j4&", url.path()); |
| 775 } | 773 } |
| OLD | NEW |