Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(270)

Unified Diff: net/dns/single_request_host_resolver_unittest.cc

Issue 2116983002: Change HostResolver::Resolve() to take an std::unique_ptr<Request>* rather than a RequestHandle* (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: http_stream_factory_impl_job_controller_unittest RequestHandle* to unique_ptr Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: net/dns/single_request_host_resolver_unittest.cc
diff --git a/net/dns/single_request_host_resolver_unittest.cc b/net/dns/single_request_host_resolver_unittest.cc
index c0b3a7b82cda8a8c031580ad4adb3b85d4bbfb58..559e411a243dab7feba98f99ee901b60802fa895 100644
--- a/net/dns/single_request_host_resolver_unittest.cc
+++ b/net/dns/single_request_host_resolver_unittest.cc
@@ -25,6 +25,9 @@ namespace {
// It checks that only one request is outstanding at a time, and that
// it is cancelled before the class is destroyed.
class HangingHostResolver : public HostResolver {
+ private:
+ class RequestImpl;
+
public:
HangingHostResolver() : outstanding_request_(NULL) {}
@@ -38,11 +41,11 @@ class HangingHostResolver : public HostResolver {
RequestPriority priority,
AddressList* addresses,
const CompletionCallback& callback,
- RequestHandle* out_req,
+ std::unique_ptr<Request>* request,
const BoundNetLog& net_log) override {
EXPECT_FALSE(has_outstanding_request());
- outstanding_request_ = reinterpret_cast<RequestHandle>(0x1234);
- *out_req = outstanding_request_;
+ outstanding_request_ = new RequestImpl(this);
+ request->reset(outstanding_request_);
// Never complete this request! Caller is expected to cancel it
// before destroying the resolver.
@@ -56,14 +59,29 @@ class HangingHostResolver : public HostResolver {
return ERR_UNEXPECTED;
}
- void CancelRequest(RequestHandle req) override {
+ void RemoveRequest(RequestImpl* request) {
EXPECT_TRUE(has_outstanding_request());
- EXPECT_EQ(req, outstanding_request_);
- outstanding_request_ = NULL;
+ EXPECT_EQ(outstanding_request_, request);
+ outstanding_request_ = nullptr;
}
private:
- RequestHandle outstanding_request_;
+ class RequestImpl : public HostResolver::Request {
+ public:
+ explicit RequestImpl(HangingHostResolver* resolver) : resolver_(resolver) {}
+
+ ~RequestImpl() override {
+ DCHECK(resolver_);
+ resolver_->RemoveRequest(this);
+ }
+
+ void ChangeRequestPriority(RequestPriority priority) override {}
+
+ private:
+ HangingHostResolver* resolver_;
+ };
+
+ HostResolver::Request* outstanding_request_;
DISALLOW_COPY_AND_ASSIGN(HangingHostResolver);
};

Powered by Google App Engine
This is Rietveld 408576698