| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | |
| 3 * | |
| 4 * Use of this source code is governed by a BSD-style license | |
| 5 * that can be found in the LICENSE file in the root of the source | |
| 6 * tree. An additional intellectual property rights grant can be found | |
| 7 * in the file PATENTS. All contributing project authors may | |
| 8 * be found in the AUTHORS file in the root of the source tree. | |
| 9 */ | |
| 10 | |
| 11 #include "webrtc/base/autodetectproxy.h" | |
| 12 #include "webrtc/base/gunit.h" | |
| 13 #include "webrtc/base/httpcommon.h" | |
| 14 #include "webrtc/base/httpcommon-inl.h" | |
| 15 | |
| 16 namespace rtc { | |
| 17 | |
| 18 static const char kUserAgent[] = ""; | |
| 19 static const char kPath[] = "/"; | |
| 20 static const char kHost[] = "relay.google.com"; | |
| 21 static const uint16_t kPort = 443; | |
| 22 static const bool kSecure = true; | |
| 23 // At most, AutoDetectProxy should take ~6 seconds. Each connect step is | |
| 24 // allotted 2 seconds, with the initial resolution + connect given an | |
| 25 // extra 2 seconds. The slowest case is: | |
| 26 // 1) Resolution + HTTPS takes full 4 seconds and fails (but resolution | |
| 27 // succeeds). | |
| 28 // 2) SOCKS5 takes the full 2 seconds. | |
| 29 // Socket creation time seems unbounded, and has been observed to take >1 second | |
| 30 // on a linux machine under load. As such, we allow for 10 seconds for timeout, | |
| 31 // though could still end up with some flakiness. | |
| 32 static const int kTimeoutMs = 10000; | |
| 33 | |
| 34 class AutoDetectProxyTest : public testing::Test, public sigslot::has_slots<> { | |
| 35 public: | |
| 36 AutoDetectProxyTest() : auto_detect_proxy_(NULL), done_(false) {} | |
| 37 | |
| 38 protected: | |
| 39 bool Create(const std::string& user_agent, | |
| 40 const std::string& path, | |
| 41 const std::string& host, | |
| 42 uint16_t port, | |
| 43 bool secure, | |
| 44 bool startnow) { | |
| 45 auto_detect_proxy_ = new AutoDetectProxy(user_agent); | |
| 46 EXPECT_TRUE(auto_detect_proxy_ != NULL); | |
| 47 if (!auto_detect_proxy_) { | |
| 48 return false; | |
| 49 } | |
| 50 Url<char> host_url(path, host, port); | |
| 51 host_url.set_secure(secure); | |
| 52 auto_detect_proxy_->set_server_url(host_url.url()); | |
| 53 auto_detect_proxy_->SignalWorkDone.connect( | |
| 54 this, | |
| 55 &AutoDetectProxyTest::OnWorkDone); | |
| 56 if (startnow) { | |
| 57 auto_detect_proxy_->Start(); | |
| 58 } | |
| 59 return true; | |
| 60 } | |
| 61 | |
| 62 bool Run(int timeout_ms) { | |
| 63 EXPECT_TRUE_WAIT(done_, timeout_ms); | |
| 64 return done_; | |
| 65 } | |
| 66 | |
| 67 void SetProxy(const SocketAddress& proxy) { | |
| 68 auto_detect_proxy_->set_proxy(proxy); | |
| 69 } | |
| 70 | |
| 71 void Start() { | |
| 72 auto_detect_proxy_->Start(); | |
| 73 } | |
| 74 | |
| 75 void TestCopesWithProxy(const SocketAddress& proxy) { | |
| 76 // Tests that at least autodetect doesn't crash for a given proxy address. | |
| 77 ASSERT_TRUE(Create(kUserAgent, | |
| 78 kPath, | |
| 79 kHost, | |
| 80 kPort, | |
| 81 kSecure, | |
| 82 false)); | |
| 83 SetProxy(proxy); | |
| 84 Start(); | |
| 85 ASSERT_TRUE(Run(kTimeoutMs)); | |
| 86 } | |
| 87 | |
| 88 private: | |
| 89 void OnWorkDone(rtc::SignalThread *thread) { | |
| 90 AutoDetectProxy *auto_detect_proxy = | |
| 91 static_cast<rtc::AutoDetectProxy *>(thread); | |
| 92 EXPECT_TRUE(auto_detect_proxy == auto_detect_proxy_); | |
| 93 auto_detect_proxy_ = NULL; | |
| 94 auto_detect_proxy->Release(); | |
| 95 done_ = true; | |
| 96 } | |
| 97 | |
| 98 AutoDetectProxy *auto_detect_proxy_; | |
| 99 bool done_; | |
| 100 }; | |
| 101 | |
| 102 TEST_F(AutoDetectProxyTest, TestDetectUnresolvedProxy) { | |
| 103 TestCopesWithProxy(rtc::SocketAddress("localhost", 9999)); | |
| 104 } | |
| 105 | |
| 106 TEST_F(AutoDetectProxyTest, TestDetectUnresolvableProxy) { | |
| 107 TestCopesWithProxy(rtc::SocketAddress("invalid", 9999)); | |
| 108 } | |
| 109 | |
| 110 TEST_F(AutoDetectProxyTest, TestDetectIPv6Proxy) { | |
| 111 TestCopesWithProxy(rtc::SocketAddress("::1", 9999)); | |
| 112 } | |
| 113 | |
| 114 TEST_F(AutoDetectProxyTest, TestDetectIPv4Proxy) { | |
| 115 TestCopesWithProxy(rtc::SocketAddress("127.0.0.1", 9999)); | |
| 116 } | |
| 117 | |
| 118 // Test that proxy detection completes successfully. (Does not actually verify | |
| 119 // the correct detection result since we don't know what proxy to expect on an | |
| 120 // arbitrary machine.) | |
| 121 TEST_F(AutoDetectProxyTest, TestProxyDetection) { | |
| 122 ASSERT_TRUE(Create(kUserAgent, | |
| 123 kPath, | |
| 124 kHost, | |
| 125 kPort, | |
| 126 kSecure, | |
| 127 true)); | |
| 128 ASSERT_TRUE(Run(kTimeoutMs)); | |
| 129 } | |
| 130 | |
| 131 } // namespace rtc | |
| OLD | NEW |