Project import generated by Copybara.
PiperOrigin-RevId: 229942388
Change-Id: Ib5a23c152c95ed4294cece9f902227c21ce531ef
diff --git a/spdy/core/array_output_buffer_test.cc b/spdy/core/array_output_buffer_test.cc
new file mode 100644
index 0000000..369778d
--- /dev/null
+++ b/spdy/core/array_output_buffer_test.cc
@@ -0,0 +1,50 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/third_party/quiche/src/spdy/core/array_output_buffer.h"
+
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace spdy {
+namespace test {
+
+// This test verifies that ArrayOutputBuffer is initialized properly.
+TEST(ArrayOutputBufferTest, InitializedFromArray) {
+ char array[100];
+ ArrayOutputBuffer buffer(array, sizeof(array));
+ EXPECT_EQ(sizeof(array), buffer.BytesFree());
+ EXPECT_EQ(0u, buffer.Size());
+ EXPECT_EQ(array, buffer.Begin());
+}
+
+// This test verifies that Reset() causes an ArrayOutputBuffer's capacity and
+// size to be reset to the initial state.
+TEST(ArrayOutputBufferTest, WriteAndReset) {
+ char array[100];
+ ArrayOutputBuffer buffer(array, sizeof(array));
+
+ // Let's write some bytes.
+ char* dst;
+ int size;
+ buffer.Next(&dst, &size);
+ ASSERT_GT(size, 1);
+ ASSERT_NE(nullptr, dst);
+ const int64_t written = size / 2;
+ memset(dst, 'x', written);
+ buffer.AdvanceWritePtr(written);
+
+ // The buffer should be partially used.
+ EXPECT_EQ(static_cast<uint64_t>(size) - written, buffer.BytesFree());
+ EXPECT_EQ(static_cast<uint64_t>(written), buffer.Size());
+
+ buffer.Reset();
+
+ // After a reset, the buffer should regain its full capacity.
+ EXPECT_EQ(sizeof(array), buffer.BytesFree());
+ EXPECT_EQ(0u, buffer.Size());
+}
+
+} // namespace test
+} // namespace spdy
