OLD | NEW |
| (Empty) |
1 // Copyright (c) 2013 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_WEBSOCKET_TEST_UTIL_H_ | |
6 #define NET_SPDY_SPDY_WEBSOCKET_TEST_UTIL_H_ | |
7 | |
8 #include "net/base/request_priority.h" | |
9 #include "net/spdy/spdy_header_block.h" | |
10 #include "net/spdy/spdy_protocol.h" | |
11 #include "net/spdy/spdy_test_util_common.h" | |
12 | |
13 namespace net { | |
14 | |
15 class SpdyWebSocketTestUtil { | |
16 public: | |
17 explicit SpdyWebSocketTestUtil(NextProto protocol); | |
18 | |
19 // Returns the value corresponding to the given key (passed through | |
20 // GetHeaderKey()), or the empty string if none exists. | |
21 std::string GetHeader(const SpdyHeaderBlock& headers, | |
22 const std::string& key) const; | |
23 | |
24 // Adds the given key/value pair to |headers|, passing the key | |
25 // through GetHeaderKey(). | |
26 void SetHeader(const std::string& key, | |
27 const std::string& value, | |
28 SpdyHeaderBlock* headers) const; | |
29 | |
30 // Constructs a standard SPDY SYN_STREAM frame for WebSocket over | |
31 // SPDY opening handshake. | |
32 SpdyFrame* ConstructSpdyWebSocketSynStream(int stream_id, | |
33 const char* path, | |
34 const char* host, | |
35 const char* origin); | |
36 | |
37 // Constructs a standard SPDY SYN_REPLY packet to match the | |
38 // WebSocket over SPDY opening handshake. | |
39 SpdyFrame* ConstructSpdyWebSocketSynReply(int stream_id); | |
40 | |
41 // Constructs a WebSocket over SPDY handshake request packet. | |
42 SpdyFrame* ConstructSpdyWebSocketHandshakeRequestFrame( | |
43 scoped_ptr<SpdyHeaderBlock> headers, | |
44 SpdyStreamId stream_id, | |
45 RequestPriority request_priority); | |
46 | |
47 // Constructs a WebSocket over SPDY handshake response packet. | |
48 SpdyFrame* ConstructSpdyWebSocketHandshakeResponseFrame( | |
49 scoped_ptr<SpdyHeaderBlock> headers, | |
50 SpdyStreamId stream_id, | |
51 RequestPriority request_priority); | |
52 | |
53 // Constructs a SPDY HEADERS frame for a WebSocket frame over SPDY. | |
54 SpdyFrame* ConstructSpdyWebSocketHeadersFrame(int stream_id, | |
55 const char* length, | |
56 bool fin); | |
57 | |
58 // Constructs a WebSocket over SPDY data packet. | |
59 SpdyFrame* ConstructSpdyWebSocketDataFrame(const char* data, | |
60 int len, | |
61 SpdyStreamId stream_id, | |
62 bool fin); | |
63 | |
64 // Forwards to |spdy_util_|. | |
65 SpdyFrame* ConstructSpdySettings(const SettingsMap& settings) const; | |
66 SpdyFrame* ConstructSpdySettingsAck() const; | |
67 SpdyMajorVersion spdy_version() const; | |
68 | |
69 private: | |
70 // Modify the header key based on the SPDY version and return it. | |
71 std::string GetHeaderKey(const std::string& key) const; | |
72 | |
73 SpdyTestUtil spdy_util_; | |
74 }; | |
75 | |
76 } // namespace net | |
77 | |
78 #endif // NET_SPDY_SPDY_WEBSOCKET_TEST_UTIL_H_ | |
OLD | NEW |