| 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 "net/dns/dns_test_util.h" | 5 #include "net/dns/dns_test_util.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/big_endian.h" | 9 #include "base/big_endian.h" |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| 11 #include "base/memory/weak_ptr.h" | 11 #include "base/memory/weak_ptr.h" |
| 12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
| 13 #include "base/sys_byteorder.h" | 13 #include "base/sys_byteorder.h" |
| 14 #include "net/base/dns_util.h" | 14 #include "net/base/dns_util.h" |
| 15 #include "net/base/io_buffer.h" | 15 #include "net/base/io_buffer.h" |
| 16 #include "net/base/net_errors.h" | 16 #include "net/base/net_errors.h" |
| 17 #include "net/dns/address_sorter.h" | 17 #include "net/dns/address_sorter.h" |
| 18 #include "net/dns/dns_query.h" | 18 #include "net/dns/dns_query.h" |
| 19 #include "net/dns/dns_response.h" | 19 #include "net/dns/dns_response.h" |
| 20 #include "net/dns/dns_transaction.h" | 20 #include "net/dns/dns_transaction.h" |
| 21 #include "testing/gtest/include/gtest/gtest.h" | 21 #include "testing/gtest/include/gtest/gtest.h" |
| 22 | 22 |
| 23 namespace net { | 23 namespace net { |
| 24 namespace { | 24 namespace { |
| 25 | 25 |
| 26 class MockAddressSorter : public AddressSorter { | 26 class MockAddressSorter : public AddressSorter { |
| 27 public: | 27 public: |
| 28 virtual ~MockAddressSorter() {} | 28 virtual ~MockAddressSorter() {} |
| 29 virtual void Sort(const AddressList& list, | 29 virtual void Sort(const AddressList& list, |
| 30 const CallbackType& callback) const OVERRIDE { | 30 const CallbackType& callback) const override { |
| 31 // Do nothing. | 31 // Do nothing. |
| 32 callback.Run(true, list); | 32 callback.Run(true, list); |
| 33 } | 33 } |
| 34 }; | 34 }; |
| 35 | 35 |
| 36 // A DnsTransaction which uses MockDnsClientRuleList to determine the response. | 36 // A DnsTransaction which uses MockDnsClientRuleList to determine the response. |
| 37 class MockTransaction : public DnsTransaction, | 37 class MockTransaction : public DnsTransaction, |
| 38 public base::SupportsWeakPtr<MockTransaction> { | 38 public base::SupportsWeakPtr<MockTransaction> { |
| 39 public: | 39 public: |
| 40 MockTransaction(const MockDnsClientRuleList& rules, | 40 MockTransaction(const MockDnsClientRuleList& rules, |
| (...skipping 12 matching lines...) Expand all Loading... |
| 53 if ((rules[i].qtype == qtype) && | 53 if ((rules[i].qtype == qtype) && |
| 54 (hostname.size() >= prefix.size()) && | 54 (hostname.size() >= prefix.size()) && |
| 55 (hostname.compare(0, prefix.size(), prefix) == 0)) { | 55 (hostname.compare(0, prefix.size(), prefix) == 0)) { |
| 56 result_ = rules[i].result; | 56 result_ = rules[i].result; |
| 57 delayed_ = rules[i].delay; | 57 delayed_ = rules[i].delay; |
| 58 break; | 58 break; |
| 59 } | 59 } |
| 60 } | 60 } |
| 61 } | 61 } |
| 62 | 62 |
| 63 virtual const std::string& GetHostname() const OVERRIDE { | 63 virtual const std::string& GetHostname() const override { |
| 64 return hostname_; | 64 return hostname_; |
| 65 } | 65 } |
| 66 | 66 |
| 67 virtual uint16 GetType() const OVERRIDE { | 67 virtual uint16 GetType() const override { |
| 68 return qtype_; | 68 return qtype_; |
| 69 } | 69 } |
| 70 | 70 |
| 71 virtual void Start() OVERRIDE { | 71 virtual void Start() override { |
| 72 EXPECT_FALSE(started_); | 72 EXPECT_FALSE(started_); |
| 73 started_ = true; | 73 started_ = true; |
| 74 if (delayed_) | 74 if (delayed_) |
| 75 return; | 75 return; |
| 76 // Using WeakPtr to cleanly cancel when transaction is destroyed. | 76 // Using WeakPtr to cleanly cancel when transaction is destroyed. |
| 77 base::MessageLoop::current()->PostTask( | 77 base::MessageLoop::current()->PostTask( |
| 78 FROM_HERE, base::Bind(&MockTransaction::Finish, AsWeakPtr())); | 78 FROM_HERE, base::Bind(&MockTransaction::Finish, AsWeakPtr())); |
| 79 } | 79 } |
| 80 | 80 |
| 81 void FinishDelayedTransaction() { | 81 void FinishDelayedTransaction() { |
| (...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 165 public: | 165 public: |
| 166 explicit MockTransactionFactory(const MockDnsClientRuleList& rules) | 166 explicit MockTransactionFactory(const MockDnsClientRuleList& rules) |
| 167 : rules_(rules) {} | 167 : rules_(rules) {} |
| 168 | 168 |
| 169 virtual ~MockTransactionFactory() {} | 169 virtual ~MockTransactionFactory() {} |
| 170 | 170 |
| 171 virtual scoped_ptr<DnsTransaction> CreateTransaction( | 171 virtual scoped_ptr<DnsTransaction> CreateTransaction( |
| 172 const std::string& hostname, | 172 const std::string& hostname, |
| 173 uint16 qtype, | 173 uint16 qtype, |
| 174 const DnsTransactionFactory::CallbackType& callback, | 174 const DnsTransactionFactory::CallbackType& callback, |
| 175 const BoundNetLog&) OVERRIDE { | 175 const BoundNetLog&) override { |
| 176 MockTransaction* transaction = | 176 MockTransaction* transaction = |
| 177 new MockTransaction(rules_, hostname, qtype, callback); | 177 new MockTransaction(rules_, hostname, qtype, callback); |
| 178 if (transaction->delayed()) | 178 if (transaction->delayed()) |
| 179 delayed_transactions_.push_back(transaction->AsWeakPtr()); | 179 delayed_transactions_.push_back(transaction->AsWeakPtr()); |
| 180 return scoped_ptr<DnsTransaction>(transaction); | 180 return scoped_ptr<DnsTransaction>(transaction); |
| 181 } | 181 } |
| 182 | 182 |
| 183 void CompleteDelayedTransactions() { | 183 void CompleteDelayedTransactions() { |
| 184 DelayedTransactionList old_delayed_transactions; | 184 DelayedTransactionList old_delayed_transactions; |
| 185 old_delayed_transactions.swap(delayed_transactions_); | 185 old_delayed_transactions.swap(delayed_transactions_); |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 220 | 220 |
| 221 AddressSorter* MockDnsClient::GetAddressSorter() { | 221 AddressSorter* MockDnsClient::GetAddressSorter() { |
| 222 return address_sorter_.get(); | 222 return address_sorter_.get(); |
| 223 } | 223 } |
| 224 | 224 |
| 225 void MockDnsClient::CompleteDelayedTransactions() { | 225 void MockDnsClient::CompleteDelayedTransactions() { |
| 226 factory_->CompleteDelayedTransactions(); | 226 factory_->CompleteDelayedTransactions(); |
| 227 } | 227 } |
| 228 | 228 |
| 229 } // namespace net | 229 } // namespace net |
| OLD | NEW |