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 |