| 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 #include "net/spdy/spdy_frame_builder.h" | |
| 6 | |
| 7 #include "net/spdy/array_output_buffer.h" | |
| 8 #include "net/spdy/spdy_framer.h" | |
| 9 #include "net/spdy/spdy_protocol.h" | |
| 10 #include "testing/platform_test.h" | |
| 11 | |
| 12 namespace net { | |
| 13 | |
| 14 namespace { | |
| 15 | |
| 16 const int64_t kSize = 64 * 1024; | |
| 17 char output_buffer[kSize] = ""; | |
| 18 | |
| 19 } // namespace | |
| 20 | |
| 21 // Verifies that SpdyFrameBuilder::GetWritableBuffer() can be used to build a | |
| 22 // SpdySerializedFrame. | |
| 23 TEST(SpdyFrameBuilderTest, GetWritableBuffer) { | |
| 24 const size_t kBuilderSize = 10; | |
| 25 SpdyFrameBuilder builder(kBuilderSize); | |
| 26 char* writable_buffer = builder.GetWritableBuffer(kBuilderSize); | |
| 27 memset(writable_buffer, ~1, kBuilderSize); | |
| 28 EXPECT_TRUE(builder.Seek(kBuilderSize)); | |
| 29 SpdySerializedFrame frame(builder.take()); | |
| 30 char expected[kBuilderSize]; | |
| 31 memset(expected, ~1, kBuilderSize); | |
| 32 EXPECT_EQ(SpdyStringPiece(expected, kBuilderSize), | |
| 33 SpdyStringPiece(frame.data(), kBuilderSize)); | |
| 34 } | |
| 35 | |
| 36 // Verifies that SpdyFrameBuilder::GetWritableBuffer() can be used to build a | |
| 37 // SpdySerializedFrame to the output buffer. | |
| 38 TEST(SpdyFrameBuilderTest, GetWritableOutput) { | |
| 39 ArrayOutputBuffer output(output_buffer, kSize); | |
| 40 const size_t kBuilderSize = 10; | |
| 41 SpdyFrameBuilder builder(kBuilderSize, &output); | |
| 42 size_t actual_size = 0; | |
| 43 char* writable_buffer = builder.GetWritableOutput(kBuilderSize, &actual_size); | |
| 44 memset(writable_buffer, ~1, kBuilderSize); | |
| 45 EXPECT_TRUE(builder.Seek(kBuilderSize)); | |
| 46 SpdySerializedFrame frame(output.Begin(), kBuilderSize, false); | |
| 47 char expected[kBuilderSize]; | |
| 48 memset(expected, ~1, kBuilderSize); | |
| 49 EXPECT_EQ(SpdyStringPiece(expected, kBuilderSize), | |
| 50 SpdyStringPiece(frame.data(), kBuilderSize)); | |
| 51 } | |
| 52 | |
| 53 // Verifies the case that the buffer's capacity is too small. | |
| 54 TEST(SpdyFrameBuilderTest, GetWritableOutputNegative) { | |
| 55 size_t small_cap = 1; | |
| 56 ArrayOutputBuffer output(output_buffer, small_cap); | |
| 57 const size_t kBuilderSize = 10; | |
| 58 SpdyFrameBuilder builder(kBuilderSize, &output); | |
| 59 size_t actual_size = 0; | |
| 60 char* writable_buffer = builder.GetWritableOutput(kBuilderSize, &actual_size); | |
| 61 builder.GetWritableOutput(kBuilderSize, &actual_size); | |
| 62 EXPECT_EQ((uint64_t)0, actual_size); | |
| 63 EXPECT_EQ(nullptr, writable_buffer); | |
| 64 } | |
| 65 | |
| 66 } // namespace net | |
| OLD | NEW |