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

net/spdy/spdy_session_unittest.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_session.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/base64.h"
@@ skipping 25 matching lines @@
 #include "net/spdy/spdy_test_util_common.h"
 #include "net/spdy/spdy_test_utils.h"
 #include "net/test/cert_test_util.h"
 #include "net/test/gtest_util.h"
 #include "net/test/test_data_directory.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/platform_test.h"
 
 using net::test::IsError;
 using net::test::IsOk;
+using net::test::TestServerPushDelegate;
 
 namespace net {
 
 namespace {
 
 const char kHttpURLFromAnotherOrigin[] = "http://www.example2.org/a.dat";
 const char kHttpsURLFromAnotherOrigin[] = "https://www.example2.org/b.dat";
 
 const char kBodyData[] = "Body data";
 const size_t kBodyDataSize = arraysize(kBodyData);
@@ skipping 15 matching lines @@
 base::TimeTicks InstantaneousReads() {
   return g_time_now;
 }
 
 class MockRequireCTDelegate : public TransportSecurityState::RequireCTDelegate {
  public:
   MOCK_METHOD1(IsCTRequiredForHost,
                CTRequirementLevel(const std::string& host));
 };
 
-class TestServerPushDelegate : public ServerPushDelegate {
- public:
-  explicit TestServerPushDelegate() {}
-
-  void OnPush(std::unique_ptr<ServerPushHelper> push_helper) override {
-    push_helpers[push_helper->GetURL()] = std::move(push_helper);
-  }
-
-  bool CancelPush(GURL url) {
-    auto itr = push_helpers.find(url);
-    if (itr == push_helpers.end())
-      return false;
-
-    itr->second->Cancel();
-    push_helpers.erase(itr);
-    return true;
-  }
-
- private:
-  std::map<GURL, std::unique_ptr<ServerPushHelper>> push_helpers;
-};
-
 }  // namespace
 
 class SpdySessionTest : public PlatformTest {
  public:
   // Functions used with RunResumeAfterUnstallTest().
 
   void StallSessionOnly(SpdyStream* stream) { StallSessionSend(); }
 
   void StallStreamOnly(SpdyStream* stream) { StallStreamSend(stream); }
 
@@ skipping 5791 matching lines @@
   ssl_info.cert = ImportCertFromFile(GetTestCertsDirectory(),
                                      "spdy_pooling.pem");
   ssl_info.is_issued_by_known_root = true;
   ssl_info.public_key_hashes.push_back(test::GetTestHashValue(primary_pin));
 
   EXPECT_TRUE(SpdySession::CanPool(
       &tss, ssl_info, "www.example.org", "mail.example.org"));
 }
 
 }  // namespace net