OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 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 | 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_frame_builder.h" | 5 #include "net/spdy/spdy_frame_builder.h" |
6 | 6 |
7 #include "net/spdy/spdy_framer.h" | 7 #include "net/spdy/spdy_framer.h" |
8 #include "net/spdy/spdy_protocol.h" | 8 #include "net/spdy/spdy_protocol.h" |
9 #include "testing/platform_test.h" | 9 #include "testing/platform_test.h" |
10 | 10 |
(...skipping 28 matching lines...) Expand all Loading... |
39 } | 39 } |
40 | 40 |
41 TEST_P(SpdyFrameBuilderTest, RewriteLength) { | 41 TEST_P(SpdyFrameBuilderTest, RewriteLength) { |
42 // Create an empty SETTINGS frame both via framer and manually via builder. | 42 // Create an empty SETTINGS frame both via framer and manually via builder. |
43 // The one created via builder is initially given the incorrect length, but | 43 // The one created via builder is initially given the incorrect length, but |
44 // then is corrected via RewriteLength(). | 44 // then is corrected via RewriteLength(). |
45 SpdyFramer framer(spdy_version_); | 45 SpdyFramer framer(spdy_version_); |
46 SpdySettingsIR settings_ir; | 46 SpdySettingsIR settings_ir; |
47 scoped_ptr<SpdyFrame> expected(framer.SerializeSettings(settings_ir)); | 47 scoped_ptr<SpdyFrame> expected(framer.SerializeSettings(settings_ir)); |
48 SpdyFrameBuilder builder(expected->size() + 1, spdy_version_); | 48 SpdyFrameBuilder builder(expected->size() + 1, spdy_version_); |
49 if (spdy_version_ < 4) { | 49 if (spdy_version_ <= SPDY3) { |
50 builder.WriteControlFrameHeader(framer, SETTINGS, 0); | 50 builder.WriteControlFrameHeader(framer, SETTINGS, 0); |
51 builder.WriteUInt32(0); // Write the number of settings. | 51 builder.WriteUInt32(0); // Write the number of settings. |
52 } else { | 52 } else { |
53 builder.BeginNewFrame(framer, SETTINGS, 0, 0); | 53 builder.BeginNewFrame(framer, SETTINGS, 0, 0); |
54 } | 54 } |
55 EXPECT_TRUE(builder.GetWritableBuffer(1) != NULL); | 55 EXPECT_TRUE(builder.GetWritableBuffer(1) != NULL); |
56 builder.RewriteLength(framer); | 56 builder.RewriteLength(framer); |
57 scoped_ptr<SpdyFrame> built(builder.take()); | 57 scoped_ptr<SpdyFrame> built(builder.take()); |
58 EXPECT_EQ(base::StringPiece(expected->data(), expected->size()), | 58 EXPECT_EQ(base::StringPiece(expected->data(), expected->size()), |
59 base::StringPiece(built->data(), expected->size())); | 59 base::StringPiece(built->data(), expected->size())); |
60 } | 60 } |
61 | 61 |
62 TEST_P(SpdyFrameBuilderTest, OverwriteFlags) { | 62 TEST_P(SpdyFrameBuilderTest, OverwriteFlags) { |
63 // Create a HEADERS frame both via framer and manually via builder with | 63 // Create a HEADERS frame both via framer and manually via builder with |
64 // different flags set, then make them match using OverwriteFlags(). | 64 // different flags set, then make them match using OverwriteFlags(). |
65 SpdyFramer framer(spdy_version_); | 65 SpdyFramer framer(spdy_version_); |
66 if (spdy_version_ < SPDY4) { | 66 if (spdy_version_ <= SPDY3) { |
67 return; | 67 return; |
68 } | 68 } |
69 SpdyHeadersIR headers_ir(1); | 69 SpdyHeadersIR headers_ir(1); |
70 headers_ir.set_end_headers(false); | 70 headers_ir.set_end_headers(false); |
71 scoped_ptr<SpdyFrame> expected(framer.SerializeHeaders(headers_ir)); | 71 scoped_ptr<SpdyFrame> expected(framer.SerializeHeaders(headers_ir)); |
72 SpdyFrameBuilder builder(expected->size(), spdy_version_); | 72 SpdyFrameBuilder builder(expected->size(), spdy_version_); |
73 builder.BeginNewFrame(framer, HEADERS, HEADERS_FLAG_END_HEADERS, 1); | 73 builder.BeginNewFrame(framer, HEADERS, HEADERS_FLAG_END_HEADERS, 1); |
74 builder.OverwriteFlags(framer, 0); | 74 builder.OverwriteFlags(framer, 0); |
75 scoped_ptr<SpdyFrame> built(builder.take()); | 75 scoped_ptr<SpdyFrame> built(builder.take()); |
76 EXPECT_EQ(base::StringPiece(expected->data(), expected->size()), | 76 EXPECT_EQ(base::StringPiece(expected->data(), expected->size()), |
77 base::StringPiece(built->data(), built->size())); | 77 base::StringPiece(built->data(), built->size())); |
78 } | 78 } |
79 | 79 |
80 } // namespace net | 80 } // namespace net |
OLD | NEW |