Index: net/dns/dns_test_util.cc |
diff --git a/net/dns/dns_test_util.cc b/net/dns/dns_test_util.cc |
index 051f595e5a87df15c5b3806970690b9f06f03bdf..bae5fb1b701d252a659f013034e1fead8c08eb36 100644 |
--- a/net/dns/dns_test_util.cc |
+++ b/net/dns/dns_test_util.cc |
@@ -14,6 +14,7 @@ |
#include "net/base/dns_util.h" |
#include "net/base/io_buffer.h" |
#include "net/base/net_errors.h" |
+#include "net/dns/address_sorter.h" |
#include "net/dns/dns_client.h" |
#include "net/dns/dns_config_service.h" |
#include "net/dns/dns_protocol.h" |
@@ -133,6 +134,15 @@ class MockTransactionFactory : public DnsTransactionFactory { |
} |
}; |
+class MockAddressSorter : public AddressSorter { |
+ public: |
+ virtual ~MockAddressSorter() {} |
+ virtual bool Sort(AddressList* list) const OVERRIDE { |
+ // Do nothing. |
+ return true; |
+ } |
+}; |
+ |
// MockDnsClient provides MockTransactionFactory. |
class MockDnsClient : public DnsClient { |
public: |
@@ -151,9 +161,14 @@ class MockDnsClient : public DnsClient { |
return config_.IsValid() ? &factory_ : NULL; |
} |
+ virtual AddressSorter* GetAddressSorter() OVERRIDE { |
+ return &address_sorter_; |
+ } |
+ |
private: |
DnsConfig config_; |
MockTransactionFactory factory_; |
+ MockAddressSorter address_sorter_; |
}; |
} // namespace |