Add Http2PushPromiseIndexTest.

Bug: 554220
Change-Id: Id4652e6c43de606dacfb4a32d01d0cb3e057a226
Reviewed-on: https://chromium-review.googlesource.com/700403Reviewed-by: Helen Li <xunjieli@chromium.org>
Commit-Queue: Bence Béky <bnc@chromium.org>
Cr-Commit-Position: refs/heads/master@{#506710}

net/BUILD.gn

@@ -5076,6 +5076,7 @@ test("net_unittests") {
     "spdy/chromium/buffered_spdy_framer_unittest.cc",
     "spdy/chromium/header_coalescer_test.cc",
     "spdy/chromium/http2_priority_dependencies_unittest.cc",
+    "spdy/chromium/http2_push_promise_index_test.cc",
     "spdy/chromium/spdy_buffer_unittest.cc",
     "spdy/chromium/spdy_http_stream_unittest.cc",
     "spdy/chromium/spdy_http_utils_unittest.cc",

net/spdy/chromium/http2_push_promise_index_test.cc

+// Copyright 2017 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/chromium/http2_push_promise_index.h"
+
+#include "base/run_loop.h"
+#include "net/base/host_port_pair.h"
+#include "net/base/privacy_mode.h"
+#include "net/log/test_net_log.h"
+#include "net/socket/socket_test_util.h"
+#include "net/spdy/chromium/spdy_session.h"
+#include "net/spdy/chromium/spdy_test_util_common.h"
+#include "net/test/cert_test_util.h"
+#include "net/test/test_data_directory.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+namespace net {
+namespace test {
+
+class Http2PushPromiseIndexTest : public testing::Test {
+ protected:
+  Http2PushPromiseIndexTest()
+      : url1_("https://www.example.org"),
+        url2_("https://mail.example.com"),
+        host_port_pair1_(HostPortPair::FromURL(url1_)),
+        host_port_pair2_(HostPortPair::FromURL(url2_)),
+        key1_(host_port_pair1_, ProxyServer::Direct(), PRIVACY_MODE_ENABLED),
+        key2_(host_port_pair2_, ProxyServer::Direct(), PRIVACY_MODE_ENABLED),
+        http_network_session_(
+            SpdySessionDependencies::SpdyCreateSession(&session_deps_)) {}
+
+  NetLogWithSource log_;
+  const GURL url1_;
+  const GURL url2_;
+  const HostPortPair host_port_pair1_;
+  const HostPortPair host_port_pair2_;
+  const SpdySessionKey key1_;
+  const SpdySessionKey key2_;
+  SpdySessionDependencies session_deps_;
+  std::unique_ptr<HttpNetworkSession> http_network_session_;
+  Http2PushPromiseIndex index_;
+};
+
+TEST_F(Http2PushPromiseIndexTest, Empty) {
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+}
+
+TEST_F(Http2PushPromiseIndexTest, FindMultipleSessionsWithDifferentUrl) {
+  MockRead reads[] = {MockRead(SYNCHRONOUS, ERR_IO_PENDING, 0)};
+  SSLSocketDataProvider ssl(SYNCHRONOUS, OK);
+  ssl.cert = ImportCertFromFile(GetTestCertsDirectory(), "spdy_pooling.pem");
+  // For first session.
+  SequencedSocketData data1(reads, arraysize(reads), nullptr, 0);
+  session_deps_.socket_factory->AddSSLSocketDataProvider(&ssl);
+  session_deps_.socket_factory->AddSocketDataProvider(&data1);
+  // For second session.
+  SequencedSocketData data2(reads, arraysize(reads), nullptr, 0);
+  session_deps_.socket_factory->AddSSLSocketDataProvider(&ssl);
+  session_deps_.socket_factory->AddSocketDataProvider(&data2);
+
+  base::WeakPtr<SpdySession> spdy_session1 =
+      CreateSpdySession(http_network_session_.get(), key1_, log_);
+  base::WeakPtr<SpdySession> spdy_session2 =
+      CreateSpdySession(http_network_session_.get(), key2_, log_);
+  // Read hanging socket data.
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.RegisterUnclaimedPushedStream(url1_, spdy_session1);
+
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key1_, url1_).get());
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.RegisterUnclaimedPushedStream(url2_, spdy_session2);
+
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key1_, url1_).get());
+  EXPECT_EQ(spdy_session2.get(), index_.Find(key2_, url2_).get());
+
+  index_.UnregisterUnclaimedPushedStream(url1_, spdy_session1.get());
+
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_EQ(spdy_session2.get(), index_.Find(key2_, url2_).get());
+
+  index_.UnregisterUnclaimedPushedStream(url2_, spdy_session2.get());
+
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  // SpdySession weak pointers must still be valid,
+  // otherwise comparisons above are not meaningful.
+  EXPECT_TRUE(spdy_session1);
+  EXPECT_TRUE(spdy_session2);
+
+  EXPECT_TRUE(data1.AllReadDataConsumed());
+  EXPECT_TRUE(data1.AllWriteDataConsumed());
+  EXPECT_TRUE(data2.AllReadDataConsumed());
+  EXPECT_TRUE(data2.AllWriteDataConsumed());
+}
+
+TEST_F(Http2PushPromiseIndexTest, MultipleSessionsForSingleUrl) {
+  MockRead reads[] = {MockRead(SYNCHRONOUS, ERR_IO_PENDING, 0)};
+  SSLSocketDataProvider ssl(SYNCHRONOUS, OK);
+  ssl.cert = ImportCertFromFile(GetTestCertsDirectory(), "spdy_pooling.pem");
+  // For first session.
+  SequencedSocketData data1(reads, arraysize(reads), nullptr, 0);
+  session_deps_.socket_factory->AddSSLSocketDataProvider(&ssl);
+  session_deps_.socket_factory->AddSocketDataProvider(&data1);
+  // For second session.
+  SequencedSocketData data2(reads, arraysize(reads), nullptr, 0);
+  session_deps_.socket_factory->AddSSLSocketDataProvider(&ssl);
+  session_deps_.socket_factory->AddSocketDataProvider(&data2);
+
+  base::WeakPtr<SpdySession> spdy_session1 =
+      CreateSpdySession(http_network_session_.get(), key1_, log_);
+  base::WeakPtr<SpdySession> spdy_session2 =
+      CreateSpdySession(http_network_session_.get(), key2_, log_);
+  // Read hanging socket data.
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_FALSE(index_.Find(key2_, url1_));
+  EXPECT_FALSE(index_.Find(key1_, url2_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.RegisterUnclaimedPushedStream(url1_, spdy_session1);
+
+  // Note that Find() only uses its SpdySessionKey argument to verify proxy and
+  // privacy mode.  Cross-origin pooling is supported, therefore HostPortPair of
+  // SpdySessionKey does not matter.
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key1_, url1_).get());
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key2_, url1_).get());
+  EXPECT_FALSE(index_.Find(key1_, url2_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.RegisterUnclaimedPushedStream(url1_, spdy_session2);
+
+  // Find returns the first SpdySession if there are multiple for the same URL.
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key1_, url1_).get());
+  EXPECT_EQ(spdy_session1.get(), index_.Find(key2_, url1_).get());
+  EXPECT_FALSE(index_.Find(key1_, url2_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.UnregisterUnclaimedPushedStream(url1_, spdy_session1.get());
+
+  EXPECT_EQ(spdy_session2.get(), index_.Find(key1_, url1_).get());
+  EXPECT_EQ(spdy_session2.get(), index_.Find(key2_, url1_).get());
+  EXPECT_FALSE(index_.Find(key1_, url2_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  index_.UnregisterUnclaimedPushedStream(url1_, spdy_session2.get());
+
+  EXPECT_FALSE(index_.Find(key1_, url1_));
+  EXPECT_FALSE(index_.Find(key2_, url1_));
+  EXPECT_FALSE(index_.Find(key1_, url2_));
+  EXPECT_FALSE(index_.Find(key2_, url2_));
+
+  // SpdySession weak pointers must still be valid,
+  // otherwise comparisons above are not meaningful.
+  EXPECT_TRUE(spdy_session1);
+  EXPECT_TRUE(spdy_session2);
+
+  EXPECT_TRUE(data1.AllReadDataConsumed());
+  EXPECT_TRUE(data1.AllWriteDataConsumed());
+  EXPECT_TRUE(data2.AllReadDataConsumed());
+  EXPECT_TRUE(data2.AllWriteDataConsumed());
+}
+
+}  // namespace test
+}  // namespace net