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 "chrome/browser/predictors/autocomplete_action_predictor.h" | 5 #include "chrome/browser/predictors/autocomplete_action_predictor.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include "base/auto_reset.h" | 9 #include "base/auto_reset.h" |
10 #include "base/command_line.h" | 10 #include "base/command_line.h" |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
84 | 84 |
85 namespace predictors { | 85 namespace predictors { |
86 | 86 |
87 class AutocompleteActionPredictorTest : public testing::Test { | 87 class AutocompleteActionPredictorTest : public testing::Test { |
88 public: | 88 public: |
89 AutocompleteActionPredictorTest() | 89 AutocompleteActionPredictorTest() |
90 : ui_thread_(BrowserThread::UI, &loop_), | 90 : ui_thread_(BrowserThread::UI, &loop_), |
91 db_thread_(BrowserThread::DB, &loop_), | 91 db_thread_(BrowserThread::DB, &loop_), |
92 file_thread_(BrowserThread::FILE, &loop_), | 92 file_thread_(BrowserThread::FILE, &loop_), |
93 profile_(new TestingProfile()), | 93 profile_(new TestingProfile()), |
94 predictor_(new AutocompleteActionPredictor(profile_.get())) { | 94 predictor_(nullptr) { |
95 } | 95 } |
96 | 96 |
97 ~AutocompleteActionPredictorTest() override { | 97 ~AutocompleteActionPredictorTest() override { |
98 predictor_.reset(NULL); | 98 predictor_.reset(NULL); |
99 profile_.reset(NULL); | 99 profile_.reset(NULL); |
100 base::RunLoop().RunUntilIdle(); | 100 base::RunLoop().RunUntilIdle(); |
101 } | 101 } |
102 | 102 |
103 void SetUp() override { | 103 void SetUp() override { |
104 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( | 104 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
105 switches::kPrerenderFromOmnibox, | 105 switches::kPrerenderFromOmnibox, |
106 switches::kPrerenderFromOmniboxSwitchValueEnabled); | 106 switches::kPrerenderFromOmniboxSwitchValueEnabled); |
107 | 107 |
| 108 ASSERT_TRUE(profile_->CreateHistoryService(true, false)); |
| 109 predictor_.reset(new AutocompleteActionPredictor(profile_.get())); |
108 predictor_->CreateLocalCachesFromDatabase(); | 110 predictor_->CreateLocalCachesFromDatabase(); |
109 ASSERT_TRUE(profile_->CreateHistoryService(true, false)); | |
110 profile_->BlockUntilHistoryProcessesPendingRequests(); | 111 profile_->BlockUntilHistoryProcessesPendingRequests(); |
111 | 112 |
112 ASSERT_TRUE(predictor_->initialized_); | 113 ASSERT_TRUE(predictor_->initialized_); |
113 ASSERT_TRUE(db_cache()->empty()); | 114 ASSERT_TRUE(db_cache()->empty()); |
114 ASSERT_TRUE(db_id_cache()->empty()); | 115 ASSERT_TRUE(db_id_cache()->empty()); |
115 } | 116 } |
116 | 117 |
117 void TearDown() override { | 118 void TearDown() override { |
| 119 predictor_->Shutdown(); |
118 profile_->DestroyHistoryService(); | 120 profile_->DestroyHistoryService(); |
119 predictor_->Shutdown(); | |
120 } | 121 } |
121 | 122 |
122 protected: | 123 protected: |
123 typedef AutocompleteActionPredictor::DBCacheKey DBCacheKey; | 124 typedef AutocompleteActionPredictor::DBCacheKey DBCacheKey; |
124 typedef AutocompleteActionPredictor::DBCacheValue DBCacheValue; | 125 typedef AutocompleteActionPredictor::DBCacheValue DBCacheValue; |
125 typedef AutocompleteActionPredictor::DBCacheMap DBCacheMap; | 126 typedef AutocompleteActionPredictor::DBCacheMap DBCacheMap; |
126 typedef AutocompleteActionPredictor::DBIdCacheMap DBIdCacheMap; | 127 typedef AutocompleteActionPredictor::DBIdCacheMap DBIdCacheMap; |
127 | 128 |
128 void AddAllRowsToHistory() { | 129 void AddAllRowsToHistory() { |
129 for (size_t i = 0; i < arraysize(test_url_db); ++i) | 130 for (size_t i = 0; i < arraysize(test_url_db); ++i) |
(...skipping 250 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
380 AutocompleteActionPredictor::ACTION_PRERENDER) ? | 381 AutocompleteActionPredictor::ACTION_PRERENDER) ? |
381 AutocompleteActionPredictor::ACTION_PRECONNECT : | 382 AutocompleteActionPredictor::ACTION_PRECONNECT : |
382 test_url_db[i].expected_action; | 383 test_url_db[i].expected_action; |
383 EXPECT_EQ(expected_action, | 384 EXPECT_EQ(expected_action, |
384 predictor()->RecommendAction(test_url_db[i].user_text, match)) | 385 predictor()->RecommendAction(test_url_db[i].user_text, match)) |
385 << "Unexpected action for " << match.destination_url; | 386 << "Unexpected action for " << match.destination_url; |
386 } | 387 } |
387 } | 388 } |
388 | 389 |
389 } // namespace predictors | 390 } // namespace predictors |
OLD | NEW |