streams_unittest.cc 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  1. // Copyright (c) 2009 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 "courgette/streams.h"
  5. #include <stddef.h>
  6. #include <stdint.h>
  7. #include <vector>
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. TEST(StreamsTest, SimpleWriteRead) {
  10. const unsigned int kValue1 = 12345;
  11. courgette::SinkStream sink;
  12. EXPECT_TRUE(sink.WriteVarint32(kValue1));
  13. const uint8_t* sink_buffer = sink.Buffer();
  14. size_t length = sink.Length();
  15. courgette::SourceStream source;
  16. source.Init(sink_buffer, length);
  17. unsigned int value;
  18. bool can_read = source.ReadVarint32(&value);
  19. EXPECT_TRUE(can_read);
  20. EXPECT_EQ(kValue1, value);
  21. EXPECT_EQ(0U, source.Remaining());
  22. }
  23. TEST(StreamsTest, SimpleWriteRead2) {
  24. courgette::SinkStream sink;
  25. EXPECT_TRUE(sink.Write("Hello", 5));
  26. const uint8_t* sink_buffer = sink.Buffer();
  27. size_t sink_length = sink.Length();
  28. courgette::SourceStream source;
  29. source.Init(sink_buffer, sink_length);
  30. char text[10] = {0};
  31. bool can_read = source.Read(text, 5);
  32. EXPECT_TRUE(can_read);
  33. EXPECT_EQ(0, memcmp("Hello", text, 5));
  34. EXPECT_EQ(0U, source.Remaining());
  35. }
  36. TEST(StreamsTest, StreamSetWriteRead) {
  37. courgette::SinkStreamSet out;
  38. out.Init(4);
  39. const unsigned int kValue1 = 12345;
  40. EXPECT_TRUE(out.stream(3)->WriteVarint32(kValue1));
  41. courgette::SinkStream collected;
  42. EXPECT_TRUE(out.CopyTo(&collected));
  43. const uint8_t* collected_buffer = collected.Buffer();
  44. size_t collected_length = collected.Length();
  45. courgette::SourceStreamSet in;
  46. bool can_init = in.Init(collected_buffer, collected_length);
  47. EXPECT_TRUE(can_init);
  48. uint32_t value;
  49. bool can_read = in.stream(3)->ReadVarint32(&value);
  50. EXPECT_TRUE(can_read);
  51. EXPECT_EQ(kValue1, value);
  52. EXPECT_EQ(0U, in.stream(3)->Remaining());
  53. EXPECT_EQ(0U, in.stream(2)->Remaining());
  54. }
  55. TEST(StreamsTest, StreamSetWriteRead2) {
  56. const size_t kNumberOfStreams = 4;
  57. const unsigned int kEnd = ~0U;
  58. courgette::SinkStreamSet out;
  59. out.Init(kNumberOfStreams);
  60. static const unsigned int data[] = {
  61. 3, 123, 3, 1000, 0, 100, 2, 100, 0, 999999,
  62. 0, 0, 0, 0, 1, 2, 1, 3, 1, 5, 0, 66,
  63. // varint32 edge case values:
  64. 1, 127, 1, 128, 1, 129, 1, 16383, 1, 16384,
  65. kEnd
  66. };
  67. for (size_t i = 0; data[i] != kEnd; i += 2) {
  68. size_t id = data[i];
  69. size_t datum = data[i + 1];
  70. EXPECT_TRUE(out.stream(id)->WriteVarint32(datum));
  71. }
  72. courgette::SinkStream collected;
  73. EXPECT_TRUE(out.CopyTo(&collected));
  74. courgette::SourceStreamSet in;
  75. bool can_init = in.Init(collected.Buffer(), collected.Length());
  76. EXPECT_TRUE(can_init);
  77. for (size_t i = 0; data[i] != kEnd; i += 2) {
  78. size_t id = data[i];
  79. size_t datum = data[i + 1];
  80. uint32_t value = 77;
  81. bool can_read = in.stream(id)->ReadVarint32(&value);
  82. EXPECT_TRUE(can_read);
  83. EXPECT_EQ(datum, value);
  84. }
  85. for (size_t i = 0; i < kNumberOfStreams; ++i) {
  86. EXPECT_EQ(0U, in.stream(i)->Remaining());
  87. }
  88. }
  89. TEST(StreamsTest, SignedVarint32) {
  90. courgette::SinkStream out;
  91. static const int32_t data[] = {0, 64, 128, 8192,
  92. 16384, 1 << 20, 1 << 21, 1 << 22,
  93. 1 << 27, 1 << 28, 0x7fffffff, -0x7fffffff};
  94. std::vector<int32_t> values;
  95. for (size_t i = 0; i < sizeof(data)/sizeof(data[0]); ++i) {
  96. int32_t basis = data[i];
  97. for (int delta = -4; delta <= 4; ++delta) {
  98. EXPECT_TRUE(out.WriteVarint32Signed(basis + delta));
  99. values.push_back(basis + delta);
  100. EXPECT_TRUE(out.WriteVarint32Signed(-basis + delta));
  101. values.push_back(-basis + delta);
  102. }
  103. }
  104. courgette::SourceStream in;
  105. in.Init(out);
  106. for (size_t i = 0; i < values.size(); ++i) {
  107. int written_value = values[i];
  108. int32_t datum;
  109. bool can_read = in.ReadVarint32Signed(&datum);
  110. EXPECT_TRUE(can_read);
  111. EXPECT_EQ(written_value, datum);
  112. }
  113. EXPECT_TRUE(in.Empty());
  114. }
  115. TEST(StreamsTest, StreamSetReadWrite) {
  116. courgette::SinkStreamSet out;
  117. { // Local scope for temporary stream sets.
  118. courgette::SinkStreamSet subset1;
  119. EXPECT_TRUE(subset1.stream(3)->WriteVarint32(30000));
  120. EXPECT_TRUE(subset1.stream(5)->WriteVarint32(50000));
  121. EXPECT_TRUE(out.WriteSet(&subset1));
  122. courgette::SinkStreamSet subset2;
  123. EXPECT_TRUE(subset2.stream(2)->WriteVarint32(20000));
  124. EXPECT_TRUE(subset2.stream(6)->WriteVarint32(60000));
  125. EXPECT_TRUE(out.WriteSet(&subset2));
  126. }
  127. courgette::SinkStream collected;
  128. EXPECT_TRUE(out.CopyTo(&collected));
  129. courgette::SourceStreamSet in;
  130. bool can_init_in = in.Init(collected.Buffer(), collected.Length());
  131. EXPECT_TRUE(can_init_in);
  132. courgette::SourceStreamSet subset1;
  133. bool can_read_1 = in.ReadSet(&subset1);
  134. EXPECT_TRUE(can_read_1);
  135. EXPECT_FALSE(in.Empty());
  136. courgette::SourceStreamSet subset2;
  137. bool can_read_2 = in.ReadSet(&subset2);
  138. EXPECT_TRUE(can_read_2);
  139. EXPECT_TRUE(in.Empty());
  140. courgette::SourceStreamSet subset3;
  141. bool can_read_3 = in.ReadSet(&subset3);
  142. EXPECT_FALSE(can_read_3);
  143. EXPECT_FALSE(subset1.Empty());
  144. EXPECT_FALSE(subset1.Empty());
  145. uint32_t datum;
  146. EXPECT_TRUE(subset1.stream(3)->ReadVarint32(&datum));
  147. EXPECT_EQ(30000U, datum);
  148. EXPECT_TRUE(subset1.stream(5)->ReadVarint32(&datum));
  149. EXPECT_EQ(50000U, datum);
  150. EXPECT_TRUE(subset1.Empty());
  151. EXPECT_TRUE(subset2.stream(2)->ReadVarint32(&datum));
  152. EXPECT_EQ(20000U, datum);
  153. EXPECT_TRUE(subset2.stream(6)->ReadVarint32(&datum));
  154. EXPECT_EQ(60000U, datum);
  155. EXPECT_TRUE(subset2.Empty());
  156. }