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
13namespace net {
14
15class 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_
79