Server push cancellation: Use TestPushDelegate in QuicChromiumClientSessionTest to cancel push.

net/spdy/spdy_test_utils.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/spdy/spdy_test_utils.h"
 
 #include <cstring>
 #include <memory>
 #include <vector>
 
@@ skipping 171 matching lines @@
 void TestHeadersHandler::OnHeaderBlockEnd(size_t header_bytes_parsed) {
   header_bytes_parsed_ = header_bytes_parsed;
 }
 
 void TestHeadersHandler::OnHeaderBlockEnd(
     size_t header_bytes_parsed,
     size_t /* compressed_header_bytes_parsed */) {
   header_bytes_parsed_ = header_bytes_parsed;
 }
 
+TestServerPushDelegate::TestServerPushDelegate() {}
+
+TestServerPushDelegate::~TestServerPushDelegate() {}
+
+void TestServerPushDelegate::OnPush(
+    std::unique_ptr<ServerPushHelper> push_helper) {
+  push_helpers[push_helper->GetURL()] = std::move(push_helper);
+}
+
+bool TestServerPushDelegate::CancelPush(GURL url) {
+  auto itr = push_helpers.find(url);
+  if (itr == push_helpers.end())
+    return false;

[Ryan Hamilton, 2016/11/11 01:22:57]

Are there any cases where this should be called with a url that is not in the
map? I'd be inclined to error out in this case.

[Zhongyi Shi, 2016/11/11 01:38:45]

It shouldn't be, unless someone is writing wrong test case. That's a good
suggestion to help write correct testcases.

+
+  itr->second->Cancel();
+  push_helpers.erase(itr);
+  return true;
+}
+
 }  // namespace test
 }  // namespace net