• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2013 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/prefs/pref_service.h"
6 #include "chrome/browser/autocomplete/autocomplete_controller.h"
7 #include "chrome/browser/autocomplete/autocomplete_provider.h"
8 #include "chrome/browser/ui/omnibox/omnibox_controller.h"
9 #include "chrome/common/pref_names.h"
10 #include "chrome/test/base/testing_profile.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12 
13 class OmniboxControllerTest : public testing::Test {
14  protected:
15   OmniboxControllerTest();
16   virtual ~OmniboxControllerTest();
17 
18   void CreateController();
19   void AssertProviders(int expected_providers);
20 
GetPrefs()21   PrefService* GetPrefs() { return profile_.GetPrefs(); }
GetAutocompleteProviders() const22   const ACProviders* GetAutocompleteProviders() const {
23     return omnibox_controller_->autocomplete_controller()->providers();
24   }
25 
26  private:
27   TestingProfile profile_;
28   scoped_ptr<OmniboxController> omnibox_controller_;
29 
30   DISALLOW_COPY_AND_ASSIGN(OmniboxControllerTest);
31 };
32 
OmniboxControllerTest()33 OmniboxControllerTest::OmniboxControllerTest() {
34 }
35 
~OmniboxControllerTest()36 OmniboxControllerTest::~OmniboxControllerTest() {
37 }
38 
CreateController()39 void OmniboxControllerTest::CreateController() {
40   omnibox_controller_.reset(new OmniboxController(NULL, &profile_));
41 }
42 
43 // Checks that the list of autocomplete providers used by the OmniboxController
44 // matches the one in the |expected_providers| bit field.
AssertProviders(int expected_providers)45 void OmniboxControllerTest::AssertProviders(int expected_providers) {
46   const ACProviders* providers = GetAutocompleteProviders();
47 
48   for (size_t i = 0; i < providers->size(); ++i) {
49     // Ensure this is a provider we wanted.
50     int type = providers->at(i)->type();
51     ASSERT_TRUE(expected_providers & type);
52 
53     // Remove it from expectations so we fail if it's there twice.
54     expected_providers &= ~type;
55   }
56 
57   // Ensure we saw all the providers we expected.
58   ASSERT_EQ(0, expected_providers);
59 }
60 
TEST_F(OmniboxControllerTest,CheckDefaultAutocompleteProviders)61 TEST_F(OmniboxControllerTest, CheckDefaultAutocompleteProviders) {
62   CreateController();
63   // First collect the basic providers.
64   int observed_providers = 0;
65   const ACProviders* providers = GetAutocompleteProviders();
66   for (size_t i = 0; i < providers->size(); ++i)
67     observed_providers |= providers->at(i)->type();
68   // Ensure we have at least one provider.
69   ASSERT_NE(0, observed_providers);
70 
71   // Ensure instant extended includes all the provides in classic Chrome.
72   int providers_with_instant_extended = observed_providers;
73   // TODO(beaudoin): remove TYPE_SEARCH once it's no longer needed to pass
74   // the Instant suggestion through via FinalizeInstantQuery.
75   CreateController();
76   AssertProviders(providers_with_instant_extended);
77 }
78