OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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/net/net_error_tab_helper.h" | 5 #include "chrome/browser/net/net_error_tab_helper.h" |
6 | 6 |
7 #include "chrome/common/net/net_error_info.h" | 7 #include "chrome/common/net/net_error_info.h" |
8 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 8 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
9 #include "content/public/browser/browser_thread.h" | 9 #include "content/public/browser/browser_thread.h" |
10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
(...skipping 15 matching lines...) Expand all Loading... |
26 EXPECT_TRUE(mock_probe_running_); | 26 EXPECT_TRUE(mock_probe_running_); |
27 OnDnsProbeFinished(status); | 27 OnDnsProbeFinished(status); |
28 mock_probe_running_ = false; | 28 mock_probe_running_ = false; |
29 } | 29 } |
30 | 30 |
31 bool mock_probe_running() const { return mock_probe_running_; } | 31 bool mock_probe_running() const { return mock_probe_running_; } |
32 DnsProbeStatus last_status_sent() const { return last_status_sent_; } | 32 DnsProbeStatus last_status_sent() const { return last_status_sent_; } |
33 int mock_sent_count() const { return mock_sent_count_; } | 33 int mock_sent_count() const { return mock_sent_count_; } |
34 | 34 |
35 private: | 35 private: |
36 virtual void StartDnsProbe() OVERRIDE { | 36 virtual void StartDnsProbe() override { |
37 EXPECT_FALSE(mock_probe_running_); | 37 EXPECT_FALSE(mock_probe_running_); |
38 mock_probe_running_ = true; | 38 mock_probe_running_ = true; |
39 } | 39 } |
40 | 40 |
41 virtual void SendInfo() OVERRIDE { | 41 virtual void SendInfo() override { |
42 last_status_sent_ = dns_probe_status(); | 42 last_status_sent_ = dns_probe_status(); |
43 mock_sent_count_++; | 43 mock_sent_count_++; |
44 } | 44 } |
45 | 45 |
46 bool mock_probe_running_; | 46 bool mock_probe_running_; |
47 DnsProbeStatus last_status_sent_; | 47 DnsProbeStatus last_status_sent_; |
48 int mock_sent_count_; | 48 int mock_sent_count_; |
49 }; | 49 }; |
50 | 50 |
51 class NetErrorTabHelperTest : public ChromeRenderViewHostTestHarness { | 51 class NetErrorTabHelperTest : public ChromeRenderViewHostTestHarness { |
52 protected: | 52 protected: |
53 enum MainFrame { SUB_FRAME, MAIN_FRAME }; | 53 enum MainFrame { SUB_FRAME, MAIN_FRAME }; |
54 enum ErrorPage { NORMAL_PAGE, ERROR_PAGE }; | 54 enum ErrorPage { NORMAL_PAGE, ERROR_PAGE }; |
55 enum ErrorType { DNS_ERROR, OTHER_ERROR }; | 55 enum ErrorType { DNS_ERROR, OTHER_ERROR }; |
56 | 56 |
57 virtual void SetUp() OVERRIDE { | 57 virtual void SetUp() override { |
58 ChromeRenderViewHostTestHarness::SetUp(); | 58 ChromeRenderViewHostTestHarness::SetUp(); |
59 subframe_ = content::RenderFrameHostTester::For(main_rfh()) | 59 subframe_ = content::RenderFrameHostTester::For(main_rfh()) |
60 ->AppendChild("subframe"); | 60 ->AppendChild("subframe"); |
61 | 61 |
62 tab_helper_.reset(new TestNetErrorTabHelper); | 62 tab_helper_.reset(new TestNetErrorTabHelper); |
63 NetErrorTabHelper::set_state_for_testing( | 63 NetErrorTabHelper::set_state_for_testing( |
64 NetErrorTabHelper::TESTING_FORCE_ENABLED); | 64 NetErrorTabHelper::TESTING_FORCE_ENABLED); |
65 } | 65 } |
66 | 66 |
67 void StartProvisionalLoad(MainFrame main_frame, ErrorPage error_page) { | 67 void StartProvisionalLoad(MainFrame main_frame, ErrorPage error_page) { |
(...skipping 305 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
373 CommitProvisionalLoad(MAIN_FRAME); | 373 CommitProvisionalLoad(MAIN_FRAME); |
374 EXPECT_TRUE(probe_running()); | 374 EXPECT_TRUE(probe_running()); |
375 EXPECT_EQ(3, sent_count()); | 375 EXPECT_EQ(3, sent_count()); |
376 EXPECT_EQ(chrome_common_net::DNS_PROBE_STARTED, last_status_sent()); | 376 EXPECT_EQ(chrome_common_net::DNS_PROBE_STARTED, last_status_sent()); |
377 | 377 |
378 FinishProbe(chrome_common_net::DNS_PROBE_FINISHED_NXDOMAIN); | 378 FinishProbe(chrome_common_net::DNS_PROBE_FINISHED_NXDOMAIN); |
379 EXPECT_FALSE(probe_running()); | 379 EXPECT_FALSE(probe_running()); |
380 EXPECT_EQ(4, sent_count()); | 380 EXPECT_EQ(4, sent_count()); |
381 EXPECT_EQ(chrome_common_net::DNS_PROBE_FINISHED_NXDOMAIN, last_status_sent()); | 381 EXPECT_EQ(chrome_common_net::DNS_PROBE_FINISHED_NXDOMAIN, last_status_sent()); |
382 } | 382 } |
OLD | NEW |