Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(41)

Side by Side Diff: net/spdy/spdy_stream_test_util.h

Issue 992733002: Remove //net (except for Android test stuff) and sdch (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/spdy/spdy_stream.cc ('k') | net/spdy/spdy_stream_test_util.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #ifndef NET_SPDY_SPDY_STREAM_TEST_UTIL_H_
6 #define NET_SPDY_SPDY_STREAM_TEST_UTIL_H_
7
8 #include "base/compiler_specific.h"
9 #include "base/memory/ref_counted.h"
10 #include "base/memory/scoped_ptr.h"
11 #include "base/strings/string_piece.h"
12 #include "net/base/io_buffer.h"
13 #include "net/base/test_completion_callback.h"
14 #include "net/spdy/spdy_read_queue.h"
15 #include "net/spdy/spdy_stream.h"
16
17 namespace net {
18
19 namespace test {
20
21 // Delegate that calls Close() on |stream_| on OnClose. Used by tests
22 // to make sure that such an action is harmless.
23 class ClosingDelegate : public SpdyStream::Delegate {
24 public:
25 explicit ClosingDelegate(const base::WeakPtr<SpdyStream>& stream);
26 ~ClosingDelegate() override;
27
28 // SpdyStream::Delegate implementation.
29 void OnRequestHeadersSent() override;
30 SpdyResponseHeadersStatus OnResponseHeadersUpdated(
31 const SpdyHeaderBlock& response_headers) override;
32 void OnDataReceived(scoped_ptr<SpdyBuffer> buffer) override;
33 void OnDataSent() override;
34 void OnClose(int status) override;
35
36 // Returns whether or not the stream is closed.
37 bool StreamIsClosed() const { return !stream_.get(); }
38
39 private:
40 base::WeakPtr<SpdyStream> stream_;
41 };
42
43 // Base class with shared functionality for test delegate
44 // implementations below.
45 class StreamDelegateBase : public SpdyStream::Delegate {
46 public:
47 explicit StreamDelegateBase(const base::WeakPtr<SpdyStream>& stream);
48 ~StreamDelegateBase() override;
49
50 void OnRequestHeadersSent() override;
51 SpdyResponseHeadersStatus OnResponseHeadersUpdated(
52 const SpdyHeaderBlock& response_headers) override;
53 void OnDataReceived(scoped_ptr<SpdyBuffer> buffer) override;
54 void OnDataSent() override;
55 void OnClose(int status) override;
56
57 // Waits for the stream to be closed and returns the status passed
58 // to OnClose().
59 int WaitForClose();
60
61 // Drains all data from the underlying read queue and returns it as
62 // a string.
63 std::string TakeReceivedData();
64
65 // Returns whether or not the stream is closed.
66 bool StreamIsClosed() const { return !stream_.get(); }
67
68 // Returns the stream's ID. If called when the stream is closed,
69 // returns the stream's ID when it was open.
70 SpdyStreamId stream_id() const { return stream_id_; }
71
72 std::string GetResponseHeaderValue(const std::string& name) const;
73 bool send_headers_completed() const { return send_headers_completed_; }
74
75 protected:
76 const base::WeakPtr<SpdyStream>& stream() { return stream_; }
77
78 private:
79 base::WeakPtr<SpdyStream> stream_;
80 SpdyStreamId stream_id_;
81 TestCompletionCallback callback_;
82 bool send_headers_completed_;
83 SpdyHeaderBlock response_headers_;
84 SpdyReadQueue received_data_queue_;
85 };
86
87 // Test delegate that does nothing. Used to capture data about the
88 // stream, e.g. its id when it was open.
89 class StreamDelegateDoNothing : public StreamDelegateBase {
90 public:
91 StreamDelegateDoNothing(const base::WeakPtr<SpdyStream>& stream);
92 ~StreamDelegateDoNothing() override;
93 };
94
95 // Test delegate that sends data immediately in OnResponseHeadersUpdated().
96 class StreamDelegateSendImmediate : public StreamDelegateBase {
97 public:
98 // |data| can be NULL.
99 StreamDelegateSendImmediate(const base::WeakPtr<SpdyStream>& stream,
100 base::StringPiece data);
101 ~StreamDelegateSendImmediate() override;
102
103 SpdyResponseHeadersStatus OnResponseHeadersUpdated(
104 const SpdyHeaderBlock& response_headers) override;
105
106 private:
107 base::StringPiece data_;
108 };
109
110 // Test delegate that sends body data.
111 class StreamDelegateWithBody : public StreamDelegateBase {
112 public:
113 StreamDelegateWithBody(const base::WeakPtr<SpdyStream>& stream,
114 base::StringPiece data);
115 ~StreamDelegateWithBody() override;
116
117 void OnRequestHeadersSent() override;
118
119 private:
120 scoped_refptr<StringIOBuffer> buf_;
121 };
122
123 // Test delegate that closes stream in OnResponseHeadersUpdated().
124 class StreamDelegateCloseOnHeaders : public StreamDelegateBase {
125 public:
126 StreamDelegateCloseOnHeaders(const base::WeakPtr<SpdyStream>& stream);
127 ~StreamDelegateCloseOnHeaders() override;
128
129 SpdyResponseHeadersStatus OnResponseHeadersUpdated(
130 const SpdyHeaderBlock& response_headers) override;
131 };
132
133 } // namespace test
134
135 } // namespace net
136
137 #endif // NET_SPDY_SPDY_STREAM_TEST_UTIL_H_
OLDNEW
« no previous file with comments | « net/spdy/spdy_stream.cc ('k') | net/spdy/spdy_stream_test_util.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698