chunked_byte_buffer_unittest.cc 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. // Copyright (c) 2012 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. #include <stdint.h>
  5. #include <string>
  6. #include <vector>
  7. #include "components/speech/chunked_byte_buffer.h"
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. namespace speech {
  10. typedef std::vector<uint8_t> ByteVector;
  11. TEST(ChunkedByteBufferTest, BasicTest) {
  12. ChunkedByteBuffer buffer;
  13. const uint8_t kChunks[] = {
  14. 0x00, 0x00, 0x00, 0x04, 0x01, 0x02, 0x03, 0x04, // Chunk 1: 4 bytes
  15. 0x00, 0x00, 0x00, 0x02, 0x05, 0x06, // Chunk 2: 2 bytes
  16. 0x00, 0x00, 0x00, 0x01, 0x07 // Chunk 3: 1 bytes
  17. };
  18. EXPECT_EQ(0U, buffer.GetTotalLength());
  19. EXPECT_FALSE(buffer.HasChunks());
  20. // Append partially chunk 1.
  21. buffer.Append(kChunks, 2);
  22. EXPECT_EQ(2U, buffer.GetTotalLength());
  23. EXPECT_FALSE(buffer.HasChunks());
  24. // Complete chunk 1.
  25. buffer.Append(kChunks + 2, 6);
  26. EXPECT_EQ(8U, buffer.GetTotalLength());
  27. EXPECT_TRUE(buffer.HasChunks());
  28. // Append fully chunk 2.
  29. buffer.Append(kChunks + 8, 6);
  30. EXPECT_EQ(14U, buffer.GetTotalLength());
  31. EXPECT_TRUE(buffer.HasChunks());
  32. // Remove and check chunk 1.
  33. std::unique_ptr<ByteVector> chunk;
  34. chunk = buffer.PopChunk();
  35. EXPECT_TRUE(chunk != nullptr);
  36. EXPECT_EQ(4U, chunk->size());
  37. EXPECT_EQ(0, std::char_traits<uint8_t>::compare(kChunks + 4, &(*chunk)[0],
  38. chunk->size()));
  39. EXPECT_EQ(6U, buffer.GetTotalLength());
  40. EXPECT_TRUE(buffer.HasChunks());
  41. // Read and check chunk 2.
  42. chunk = buffer.PopChunk();
  43. EXPECT_TRUE(chunk != nullptr);
  44. EXPECT_EQ(2U, chunk->size());
  45. EXPECT_EQ(0, std::char_traits<uint8_t>::compare(kChunks + 12, &(*chunk)[0],
  46. chunk->size()));
  47. EXPECT_EQ(0U, buffer.GetTotalLength());
  48. EXPECT_FALSE(buffer.HasChunks());
  49. // Append fully chunk 3.
  50. buffer.Append(kChunks + 14, 5);
  51. EXPECT_EQ(5U, buffer.GetTotalLength());
  52. // Remove and check chunk 3.
  53. chunk = buffer.PopChunk();
  54. EXPECT_TRUE(chunk != nullptr);
  55. EXPECT_EQ(1U, chunk->size());
  56. EXPECT_EQ((*chunk)[0], kChunks[18]);
  57. EXPECT_EQ(0U, buffer.GetTotalLength());
  58. EXPECT_FALSE(buffer.HasChunks());
  59. }
  60. } // namespace speech