| 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 <time.h> | 5 #include <time.h> |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <sstream> | 8 #include <sstream> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 21 matching lines...) Expand all Loading... |
| 32 #include "testing/gtest/include/gtest/gtest.h" | 32 #include "testing/gtest/include/gtest/gtest.h" |
| 33 | 33 |
| 34 using base::Time; | 34 using base::Time; |
| 35 using base::TimeDelta; | 35 using base::TimeDelta; |
| 36 using content::BrowserThread; | 36 using content::BrowserThread; |
| 37 | 37 |
| 38 namespace chrome_browser_net { | 38 namespace chrome_browser_net { |
| 39 | 39 |
| 40 class WaitForResolutionHelper; | 40 class WaitForResolutionHelper; |
| 41 | 41 |
| 42 typedef base::RepeatingTimer<WaitForResolutionHelper> HelperTimer; | |
| 43 | |
| 44 class WaitForResolutionHelper { | 42 class WaitForResolutionHelper { |
| 45 public: | 43 public: |
| 46 WaitForResolutionHelper(Predictor* predictor, const UrlList& hosts, | 44 WaitForResolutionHelper(Predictor* predictor, |
| 47 HelperTimer* timer, int checks_until_quit) | 45 const UrlList& hosts, |
| 46 base::RepeatingTimer* timer, |
| 47 int checks_until_quit) |
| 48 : predictor_(predictor), | 48 : predictor_(predictor), |
| 49 hosts_(hosts), | 49 hosts_(hosts), |
| 50 timer_(timer), | 50 timer_(timer), |
| 51 checks_until_quit_(checks_until_quit) { | 51 checks_until_quit_(checks_until_quit) {} |
| 52 } | |
| 53 | 52 |
| 54 void CheckIfResolutionsDone() { | 53 void CheckIfResolutionsDone() { |
| 55 if (--checks_until_quit_ > 0) { | 54 if (--checks_until_quit_ > 0) { |
| 56 for (UrlList::const_iterator i = hosts_.begin(); i != hosts_.end(); ++i) | 55 for (UrlList::const_iterator i = hosts_.begin(); i != hosts_.end(); ++i) |
| 57 if (predictor_->GetResolutionDuration(*i) == | 56 if (predictor_->GetResolutionDuration(*i) == |
| 58 UrlInfo::NullDuration()) | 57 UrlInfo::NullDuration()) |
| 59 return; // We don't have resolution for that host. | 58 return; // We don't have resolution for that host. |
| 60 } | 59 } |
| 61 | 60 |
| 62 // When all hostnames have been resolved, or we've hit the limit, | 61 // When all hostnames have been resolved, or we've hit the limit, |
| 63 // exit the loop. | 62 // exit the loop. |
| 64 timer_->Stop(); | 63 timer_->Stop(); |
| 65 base::MessageLoop::current()->Quit(); | 64 base::MessageLoop::current()->Quit(); |
| 66 delete timer_; | 65 delete timer_; |
| 67 delete this; | 66 delete this; |
| 68 } | 67 } |
| 69 | 68 |
| 70 private: | 69 private: |
| 71 Predictor* predictor_; | 70 Predictor* predictor_; |
| 72 const UrlList hosts_; | 71 const UrlList hosts_; |
| 73 HelperTimer* timer_; | 72 base::RepeatingTimer* timer_; |
| 74 int checks_until_quit_; | 73 int checks_until_quit_; |
| 75 }; | 74 }; |
| 76 | 75 |
| 77 class PredictorTest : public testing::Test { | 76 class PredictorTest : public testing::Test { |
| 78 public: | 77 public: |
| 79 PredictorTest() | 78 PredictorTest() |
| 80 : ui_thread_(BrowserThread::UI, &loop_), | 79 : ui_thread_(BrowserThread::UI, &loop_), |
| 81 io_thread_(BrowserThread::IO, &loop_), | 80 io_thread_(BrowserThread::IO, &loop_), |
| 82 host_resolver_(new net::MockCachingHostResolver()) { | 81 host_resolver_(new net::MockCachingHostResolver()) { |
| 83 } | 82 } |
| (...skipping 11 matching lines...) Expand all Loading... |
| 95 // only be incurred by the first request, after which the result will be | 94 // only be incurred by the first request, after which the result will be |
| 96 // cached internally by |host_resolver_|. | 95 // cached internally by |host_resolver_|. |
| 97 net::RuleBasedHostResolverProc* rules = host_resolver_->rules(); | 96 net::RuleBasedHostResolverProc* rules = host_resolver_->rules(); |
| 98 rules->AddRuleWithLatency("www.google.com", "127.0.0.1", 50); | 97 rules->AddRuleWithLatency("www.google.com", "127.0.0.1", 50); |
| 99 rules->AddRuleWithLatency("gmail.google.com.com", "127.0.0.1", 70); | 98 rules->AddRuleWithLatency("gmail.google.com.com", "127.0.0.1", 70); |
| 100 rules->AddRuleWithLatency("mail.google.com", "127.0.0.1", 44); | 99 rules->AddRuleWithLatency("mail.google.com", "127.0.0.1", 44); |
| 101 rules->AddRuleWithLatency("gmail.com", "127.0.0.1", 63); | 100 rules->AddRuleWithLatency("gmail.com", "127.0.0.1", 63); |
| 102 } | 101 } |
| 103 | 102 |
| 104 void WaitForResolution(Predictor* predictor, const UrlList& hosts) { | 103 void WaitForResolution(Predictor* predictor, const UrlList& hosts) { |
| 105 HelperTimer* timer = new HelperTimer(); | 104 base::RepeatingTimer* timer = new base::RepeatingTimer(); |
| 106 // By default allow the loop to run for a minute -- 600 iterations. | 105 // By default allow the loop to run for a minute -- 600 iterations. |
| 107 timer->Start(FROM_HERE, TimeDelta::FromMilliseconds(100), | 106 timer->Start(FROM_HERE, TimeDelta::FromMilliseconds(100), |
| 108 new WaitForResolutionHelper(predictor, hosts, timer, 600), | 107 new WaitForResolutionHelper(predictor, hosts, timer, 600), |
| 109 &WaitForResolutionHelper::CheckIfResolutionsDone); | 108 &WaitForResolutionHelper::CheckIfResolutionsDone); |
| 110 base::MessageLoop::current()->Run(); | 109 base::MessageLoop::current()->Run(); |
| 111 } | 110 } |
| 112 | 111 |
| 113 void WaitForResolutionWithLimit( | 112 void WaitForResolutionWithLimit( |
| 114 Predictor* predictor, const UrlList& hosts, int limit) { | 113 Predictor* predictor, const UrlList& hosts, int limit) { |
| 115 HelperTimer* timer = new HelperTimer(); | 114 base::RepeatingTimer* timer = new base::RepeatingTimer(); |
| 116 timer->Start(FROM_HERE, TimeDelta::FromMilliseconds(100), | 115 timer->Start(FROM_HERE, TimeDelta::FromMilliseconds(100), |
| 117 new WaitForResolutionHelper(predictor, hosts, timer, limit), | 116 new WaitForResolutionHelper(predictor, hosts, timer, limit), |
| 118 &WaitForResolutionHelper::CheckIfResolutionsDone); | 117 &WaitForResolutionHelper::CheckIfResolutionsDone); |
| 119 base::MessageLoop::current()->Run(); | 118 base::MessageLoop::current()->Run(); |
| 120 } | 119 } |
| 121 | 120 |
| 122 private: | 121 private: |
| 123 // IMPORTANT: do not move this below |host_resolver_|; the host resolver | 122 // IMPORTANT: do not move this below |host_resolver_|; the host resolver |
| 124 // must not outlive the message loop, otherwise bad things can happen | 123 // must not outlive the message loop, otherwise bad things can happen |
| 125 // (like posting to a deleted message loop). | 124 // (like posting to a deleted message loop). |
| (...skipping 712 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 838 testing_master.Resolve(goog, UrlInfo::OMNIBOX_MOTIVATED); | 837 testing_master.Resolve(goog, UrlInfo::OMNIBOX_MOTIVATED); |
| 839 | 838 |
| 840 // Proxy may not be in use (the PAC script has not yet been evaluated), so the | 839 // Proxy may not be in use (the PAC script has not yet been evaluated), so the |
| 841 // name has been registered for pre-resolve. | 840 // name has been registered for pre-resolve. |
| 842 EXPECT_FALSE(testing_master.work_queue_.IsEmpty()); | 841 EXPECT_FALSE(testing_master.work_queue_.IsEmpty()); |
| 843 | 842 |
| 844 testing_master.Shutdown(); | 843 testing_master.Shutdown(); |
| 845 } | 844 } |
| 846 | 845 |
| 847 } // namespace chrome_browser_net | 846 } // namespace chrome_browser_net |
| OLD | NEW |