OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/spdy/spdy_stream.h" | 5 #include "net/spdy/spdy_stream.h" |
6 #include "base/ref_counted.h" | 6 #include "base/ref_counted.h" |
7 #include "net/spdy/spdy_session.h" | 7 #include "net/spdy/spdy_session.h" |
8 #include "net/spdy/spdy_test_util.h" | 8 #include "net/spdy/spdy_test_util.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
43 virtual ~TestSpdyStreamDelegate() {} | 43 virtual ~TestSpdyStreamDelegate() {} |
44 | 44 |
45 virtual bool OnSendHeadersComplete(int status) { | 45 virtual bool OnSendHeadersComplete(int status) { |
46 send_headers_completed_ = true; | 46 send_headers_completed_ = true; |
47 return true; | 47 return true; |
48 } | 48 } |
49 virtual int OnSendBody() { | 49 virtual int OnSendBody() { |
50 ADD_FAILURE() << "OnSendBody should not be called"; | 50 ADD_FAILURE() << "OnSendBody should not be called"; |
51 return ERR_UNEXPECTED; | 51 return ERR_UNEXPECTED; |
52 } | 52 } |
53 virtual bool OnSendBodyComplete(int status) { | 53 virtual bool OnSendBodyComplete(int* status) { |
54 ADD_FAILURE() << "OnSendBodyComplete should not be called"; | 54 ADD_FAILURE() << "OnSendBodyComplete should not be called"; |
55 return true; | 55 return true; |
56 } | 56 } |
57 | 57 |
58 virtual int OnResponseReceived(const spdy::SpdyHeaderBlock& response, | 58 virtual int OnResponseReceived(const spdy::SpdyHeaderBlock& response, |
59 base::Time response_time, | 59 base::Time response_time, |
60 int status) { | 60 int status) { |
61 EXPECT_TRUE(send_headers_completed_); | 61 EXPECT_TRUE(send_headers_completed_); |
62 *response_ = response; | 62 *response_ = response; |
63 if (buf_) { | 63 if (buf_) { |
64 EXPECT_EQ(ERR_IO_PENDING, | 64 EXPECT_EQ(ERR_IO_PENDING, |
65 stream_->WriteStreamData(buf_.get(), buf_->size(), | 65 stream_->WriteStreamData(buf_.get(), buf_->size(), |
66 spdy::DATA_FLAG_NONE)); | 66 spdy::DATA_FLAG_NONE)); |
67 } | 67 } |
68 return status; | 68 return status; |
69 } | 69 } |
70 virtual void OnDataReceived(const char* buffer, int bytes) { | 70 virtual void OnDataReceived(const char* buffer, int bytes) { |
71 received_data_ += std::string(buffer, bytes); | 71 received_data_ += std::string(buffer, bytes); |
72 } | 72 } |
73 virtual void OnDataSent(int length) { | 73 virtual void OnDataSent(int length) { |
74 data_sent_ += length; | 74 data_sent_ += length; |
75 } | 75 } |
76 virtual void OnClose(int status) { | 76 virtual void OnClose(int status) { |
77 closed_ = true; | 77 closed_ = true; |
78 CompletionCallback* callback = callback_; | 78 CompletionCallback* callback = callback_; |
79 callback_ = NULL; | 79 callback_ = NULL; |
80 callback->Run(OK); | 80 callback->Run(OK); |
81 } | 81 } |
| 82 virtual void set_chunk_callback(net::ChunkCallback *) {} |
82 | 83 |
83 bool send_headers_completed() const { return send_headers_completed_; } | 84 bool send_headers_completed() const { return send_headers_completed_; } |
84 const linked_ptr<spdy::SpdyHeaderBlock>& response() const { | 85 const linked_ptr<spdy::SpdyHeaderBlock>& response() const { |
85 return response_; | 86 return response_; |
86 } | 87 } |
87 const std::string& received_data() const { return received_data_; } | 88 const std::string& received_data() const { return received_data_; } |
88 int data_sent() const { return data_sent_; } | 89 int data_sent() const { return data_sent_; } |
89 bool closed() const { return closed_; } | 90 bool closed() const { return closed_; } |
90 | 91 |
91 private: | 92 private: |
(...skipping 179 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
271 stream->OnHeaders(headers); | 272 stream->OnHeaders(headers); |
272 | 273 |
273 stream->set_response_received(); | 274 stream->set_response_received(); |
274 EXPECT_TRUE(stream->response_received()); | 275 EXPECT_TRUE(stream->response_received()); |
275 EXPECT_TRUE(stream->HasUrl()); | 276 EXPECT_TRUE(stream->HasUrl()); |
276 EXPECT_EQ(kStreamUrl, stream->GetUrl().spec()); | 277 EXPECT_EQ(kStreamUrl, stream->GetUrl().spec()); |
277 } | 278 } |
278 | 279 |
279 | 280 |
280 } // namespace net | 281 } // namespace net |
OLD | NEW |