| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/tools/quic/quic_spdy_client_stream.h" | 5 #include "net/tools/quic/quic_spdy_client_stream.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
| 11 #include "net/quic/quic_utils.h" | 11 #include "net/quic/quic_utils.h" |
| 12 #include "net/quic/spdy_utils.h" | 12 #include "net/quic/spdy_utils.h" |
| 13 #include "net/quic/test_tools/crypto_test_utils.h" | 13 #include "net/quic/test_tools/crypto_test_utils.h" |
| 14 #include "net/quic/test_tools/quic_test_utils.h" | 14 #include "net/quic/test_tools/quic_test_utils.h" |
| 15 #include "net/tools/quic/quic_client_session.h" | 15 #include "net/tools/quic/quic_client_session.h" |
| 16 #include "net/tools/quic/spdy_balsa_utils.h" | 16 #include "net/tools/quic/spdy_balsa_utils.h" |
| 17 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 19 |
| 20 using net::test::CryptoTestUtils; | 20 using net::test::CryptoTestUtils; |
| 21 using net::test::DefaultQuicConfig; | 21 using net::test::DefaultQuicConfig; |
| 22 using net::test::MockConnection; | 22 using net::test::MockQuicConnection; |
| 23 using net::test::MockConnectionHelper; | 23 using net::test::MockQuicConnectionHelper; |
| 24 using net::test::SupportedVersions; | 24 using net::test::SupportedVersions; |
| 25 using net::test::kClientDataStreamId1; | 25 using net::test::kClientDataStreamId1; |
| 26 using net::test::kServerDataStreamId1; | 26 using net::test::kServerDataStreamId1; |
| 27 using net::test::kInitialSessionFlowControlWindowForTest; | 27 using net::test::kInitialSessionFlowControlWindowForTest; |
| 28 using net::test::kInitialStreamFlowControlWindowForTest; | 28 using net::test::kInitialStreamFlowControlWindowForTest; |
| 29 using net::test::ValueRestore; | 29 using net::test::ValueRestore; |
| 30 | 30 |
| 31 using std::string; | 31 using std::string; |
| 32 using testing::StrictMock; | 32 using testing::StrictMock; |
| 33 using testing::TestWithParam; | 33 using testing::TestWithParam; |
| (...skipping 22 matching lines...) Expand all Loading... |
| 56 QuicCryptoClientConfig crypto_config_; | 56 QuicCryptoClientConfig crypto_config_; |
| 57 | 57 |
| 58 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); | 58 DISALLOW_COPY_AND_ASSIGN(MockQuicClientSession); |
| 59 }; | 59 }; |
| 60 | 60 |
| 61 class QuicSpdyClientStreamTest : public ::testing::Test { | 61 class QuicSpdyClientStreamTest : public ::testing::Test { |
| 62 public: | 62 public: |
| 63 class StreamVisitor; | 63 class StreamVisitor; |
| 64 | 64 |
| 65 QuicSpdyClientStreamTest() | 65 QuicSpdyClientStreamTest() |
| 66 : connection_(new StrictMock<MockConnection>(&helper_, | 66 : connection_(new StrictMock<MockQuicConnection>(&helper_, |
| 67 &alarm_factory_, | 67 &alarm_factory_, |
| 68 Perspective::IS_CLIENT)), | 68 Perspective::IS_CLIENT)), |
| 69 session_(connection_, &push_promise_index_), | 69 session_(connection_, &push_promise_index_), |
| 70 body_("hello world") { | 70 body_("hello world") { |
| 71 session_.Initialize(); | 71 session_.Initialize(); |
| 72 | 72 |
| 73 headers_.SetResponseFirstlineFromStringPieces("HTTP/1.1", "200", "Ok"); | 73 headers_.SetResponseFirstlineFromStringPieces("HTTP/1.1", "200", "Ok"); |
| 74 headers_.ReplaceOrAppendHeader("content-length", "11"); | 74 headers_.ReplaceOrAppendHeader("content-length", "11"); |
| 75 | 75 |
| 76 headers_string_ = net::SpdyBalsaUtils::SerializeResponseHeaders(headers_); | 76 headers_string_ = net::SpdyBalsaUtils::SerializeResponseHeaders(headers_); |
| 77 | 77 |
| 78 stream_.reset(new QuicSpdyClientStream(kClientDataStreamId1, &session_)); | 78 stream_.reset(new QuicSpdyClientStream(kClientDataStreamId1, &session_)); |
| 79 stream_visitor_.reset(new StreamVisitor()); | 79 stream_visitor_.reset(new StreamVisitor()); |
| 80 stream_->set_visitor(stream_visitor_.get()); | 80 stream_->set_visitor(stream_visitor_.get()); |
| 81 } | 81 } |
| 82 | 82 |
| 83 class StreamVisitor : public QuicSpdyClientStream::Visitor { | 83 class StreamVisitor : public QuicSpdyClientStream::Visitor { |
| 84 void OnClose(QuicSpdyStream* stream) override { | 84 void OnClose(QuicSpdyStream* stream) override { |
| 85 DVLOG(1) << "stream " << stream->id(); | 85 DVLOG(1) << "stream " << stream->id(); |
| 86 } | 86 } |
| 87 }; | 87 }; |
| 88 | 88 |
| 89 MockConnectionHelper helper_; | 89 MockQuicConnectionHelper helper_; |
| 90 MockAlarmFactory alarm_factory_; | 90 MockAlarmFactory alarm_factory_; |
| 91 StrictMock<MockConnection>* connection_; | 91 StrictMock<MockQuicConnection>* connection_; |
| 92 QuicClientPushPromiseIndex push_promise_index_; | 92 QuicClientPushPromiseIndex push_promise_index_; |
| 93 | 93 |
| 94 MockQuicClientSession session_; | 94 MockQuicClientSession session_; |
| 95 std::unique_ptr<QuicSpdyClientStream> stream_; | 95 std::unique_ptr<QuicSpdyClientStream> stream_; |
| 96 std::unique_ptr<StreamVisitor> stream_visitor_; | 96 std::unique_ptr<StreamVisitor> stream_visitor_; |
| 97 BalsaHeaders headers_; | 97 BalsaHeaders headers_; |
| 98 string headers_string_; | 98 string headers_string_; |
| 99 string body_; | 99 string body_; |
| 100 }; | 100 }; |
| 101 | 101 |
| (...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 176 // Now send the body, which should close the stream as the FIN has been | 176 // Now send the body, which should close the stream as the FIN has been |
| 177 // received, as well as all data. | 177 // received, as well as all data. |
| 178 EXPECT_CALL(session_, CloseStream(stream_->id())); | 178 EXPECT_CALL(session_, CloseStream(stream_->id())); |
| 179 stream_->OnStreamFrame( | 179 stream_->OnStreamFrame( |
| 180 QuicStreamFrame(stream_->id(), /*fin=*/false, /*offset=*/0, body_)); | 180 QuicStreamFrame(stream_->id(), /*fin=*/false, /*offset=*/0, body_)); |
| 181 } | 181 } |
| 182 | 182 |
| 183 } // namespace | 183 } // namespace |
| 184 } // namespace test | 184 } // namespace test |
| 185 } // namespace net | 185 } // namespace net |
| OLD | NEW |