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

net/spdy/spdy_test_utils.h

 // 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.
 
 #ifndef NET_SPDY_SPDY_TEST_UTILS_H_
 #define NET_SPDY_SPDY_TEST_UTILS_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <string>
 
 #include "base/strings/string_piece.h"
+#include "net/spdy/server_push_delegate.h"
 #include "net/spdy/spdy_bug_tracker.h"
 #include "net/spdy/spdy_header_block.h"
 #include "net/spdy/spdy_headers_handler_interface.h"
 #include "net/spdy/spdy_protocol.h"
 #include "net/test/gtest_util.h"
 
 #define EXPECT_SPDY_BUG EXPECT_DFATAL
 
 namespace net {
 
@@ skipping 58 matching lines @@
   const SpdyHeaderBlock& decoded_block() const { return block_; }
   size_t header_bytes_parsed() { return header_bytes_parsed_; }
 
  private:
   SpdyHeaderBlock block_;
   size_t header_bytes_parsed_;
 
   DISALLOW_COPY_AND_ASSIGN(TestHeadersHandler);
 };
 
+// A test implementation of ServerPushDelegate that caches all the pushed
+// request and provides a interface to cancel the push given url.
+class TestServerPushDelegate : public ServerPushDelegate {
+ public:
+  explicit TestServerPushDelegate();
+  ~TestServerPushDelegate() override;
+
+  void OnPush(std::unique_ptr<ServerPushHelper> push_helper) override;
+
+  bool CancelPush(GURL url);
+
+ private:
+  std::map<GURL, std::unique_ptr<ServerPushHelper>> push_helpers;
+};
+
 }  // namespace test
 }  // namespace net
 
 #endif  // NET_SPDY_SPDY_TEST_UTILS_H_