| 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 |