OLD | NEW |
| (Empty) |
1 // Copyright (c) 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 "chrome/common/net/net_error_tracker.h" | |
6 | |
7 #include "testing/gtest/include/gtest/gtest.h" | |
8 | |
9 namespace { | |
10 | |
11 typedef NetErrorTracker::FrameType FrameType; | |
12 typedef NetErrorTracker::PageType PageType; | |
13 typedef NetErrorTracker::ErrorType ErrorType; | |
14 | |
15 const FrameType FRAME_SUB = NetErrorTracker::FRAME_SUB; | |
16 const FrameType FRAME_MAIN = NetErrorTracker::FRAME_MAIN; | |
17 | |
18 const PageType PAGE_NORMAL = NetErrorTracker::PAGE_NORMAL; | |
19 const PageType PAGE_ERROR = NetErrorTracker::PAGE_ERROR; | |
20 | |
21 const ErrorType ERROR_OTHER = NetErrorTracker::ERROR_OTHER; | |
22 const ErrorType ERROR_DNS = NetErrorTracker::ERROR_DNS; | |
23 | |
24 class NetErrorTrackerTest : public testing::Test { | |
25 public: | |
26 NetErrorTrackerTest() | |
27 : tracker_(base::Bind(&NetErrorTrackerTest::TrackerCallback, | |
28 base::Unretained(this))), | |
29 callback_state_(NetErrorTracker::DNS_ERROR_PAGE_NONE), | |
30 callback_count_(0) { | |
31 } | |
32 | |
33 protected: | |
34 NetErrorTracker tracker_; | |
35 NetErrorTracker::DnsErrorPageState callback_state_; | |
36 int callback_count_; | |
37 | |
38 private: | |
39 void TrackerCallback(NetErrorTracker::DnsErrorPageState state) { | |
40 callback_state_ = state; | |
41 ++callback_count_; | |
42 } | |
43 }; | |
44 | |
45 TEST_F(NetErrorTrackerTest, InitialState) { | |
46 EXPECT_EQ(0, callback_count_); | |
47 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
48 } | |
49 | |
50 TEST_F(NetErrorTrackerTest, SuccessfulMainFrameLoad) { | |
51 tracker_.OnStartProvisionalLoad(FRAME_MAIN, PAGE_NORMAL); | |
52 tracker_.OnCommitProvisionalLoad(FRAME_MAIN); | |
53 tracker_.OnFinishLoad(FRAME_MAIN); | |
54 | |
55 EXPECT_EQ(0, callback_count_); | |
56 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
57 } | |
58 | |
59 TEST_F(NetErrorTrackerTest, SuccessfulSubFrameLoad) { | |
60 tracker_.OnStartProvisionalLoad(FRAME_SUB, PAGE_NORMAL); | |
61 tracker_.OnCommitProvisionalLoad(FRAME_SUB); | |
62 tracker_.OnFinishLoad(FRAME_SUB); | |
63 | |
64 EXPECT_EQ(0, callback_count_); | |
65 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
66 } | |
67 | |
68 TEST_F(NetErrorTrackerTest, FailedMainFrameLoad) { | |
69 tracker_.OnStartProvisionalLoad(FRAME_MAIN, PAGE_NORMAL); | |
70 EXPECT_EQ(0, callback_count_); | |
71 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
72 | |
73 tracker_.OnFailProvisionalLoad(FRAME_MAIN, ERROR_DNS); | |
74 EXPECT_EQ(1, callback_count_); | |
75 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_PENDING, callback_state_); | |
76 | |
77 tracker_.OnStartProvisionalLoad(FRAME_MAIN, PAGE_ERROR); | |
78 tracker_.OnCommitProvisionalLoad(FRAME_MAIN); | |
79 tracker_.OnFinishLoad(FRAME_MAIN); | |
80 EXPECT_EQ(2, callback_count_); | |
81 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_LOADED, callback_state_); | |
82 | |
83 tracker_.OnStartProvisionalLoad(FRAME_MAIN, PAGE_NORMAL); | |
84 EXPECT_EQ(3, callback_count_); | |
85 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
86 } | |
87 | |
88 TEST_F(NetErrorTrackerTest, FailedSubFrameLoad) { | |
89 tracker_.OnStartProvisionalLoad(FRAME_SUB, PAGE_NORMAL); | |
90 tracker_.OnFailProvisionalLoad(FRAME_SUB, ERROR_DNS); | |
91 tracker_.OnStartProvisionalLoad(FRAME_SUB, PAGE_ERROR); | |
92 tracker_.OnCommitProvisionalLoad(FRAME_SUB); | |
93 tracker_.OnFinishLoad(FRAME_SUB); | |
94 EXPECT_EQ(0, callback_count_); | |
95 EXPECT_EQ(NetErrorTracker::DNS_ERROR_PAGE_NONE, callback_state_); | |
96 } | |
97 | |
98 } // namespace | |
OLD | NEW |