PageRenderTime 36ms CodeModel.GetById 24ms RepoModel.GetById 1ms app.codeStats 0ms

/mordor/tests/memory_stream.cpp

http://github.com/mozy/mordor
C++ | 178 lines | 165 code | 11 blank | 2 comment | 60 complexity | c213cb7d4665e6ee20fd7748405c8fc3 MD5 | raw file
Possible License(s): BSD-3-Clause 
    

  1. // Copyright (c) 2009 - Mozy, Inc.
    2. 

  2. 

  3. #include "mordor/streams/memory.h"
    4. 

  4. #include "mordor/version.h"
    5. 

  5. #include "mordor/test/test.h"
    6. 

  6. 

  7. using namespace Mordor;
    8. 

  8. using namespace Mordor::Test;
    9. 

  9. 

  10. MORDOR_UNITTEST(MemoryStream, basic)
    11. 

  11. {
    12. 

  12.     MemoryStream stream;
    13. 

  13.     Buffer buffer;
    14. 

  14.     MORDOR_TEST_ASSERT_EQUAL(stream.size(), 0);
    15. 

  15.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 0);
    16. 

  16.     MORDOR_TEST_ASSERT_EQUAL(stream.write("cody", 4), 4u);
    17. 

  17.     MORDOR_TEST_ASSERT_EQUAL(stream.size(), 4);
    18. 

  18.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    19. 

  19.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    20. 

  20.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    21. 

  21.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(0, Stream::BEGIN), 0);
    22. 

  22.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 0);
    23. 

  23.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    24. 

  24.     MORDOR_TEST_ASSERT(stream.readBuffer() == "cody");
    25. 

  25.     MORDOR_TEST_ASSERT_EQUAL(stream.read(buffer, 2), 2u);
    26. 

  26.     MORDOR_TEST_ASSERT_EQUAL(stream.size(), 4);
    27. 

  27.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 2);
    28. 

  28.     MORDOR_TEST_ASSERT(buffer == "co");
    29. 

  29.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    30. 

  30.     MORDOR_TEST_ASSERT(stream.readBuffer() == "dy");
    31. 

  31. }
    32. 

  32. 

  33. MORDOR_UNITTEST(MemoryStream, absoluteSeek)
    34. 

  34. {
    35. 

  35.     MemoryStream stream;
    36. 

  36.     stream.write("cody", 4u);
    37. 

  37.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    38. 

  38.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    39. 

  39.     MORDOR_TEST_ASSERT_EXCEPTION(stream.seek(-1, Stream::BEGIN), std::invalid_argument);
    40. 

  40. #ifdef X86
    41. 

  41.     MORDOR_TEST_ASSERT_EXCEPTION(stream.seek(0x100000000ull, Stream::BEGIN), std::invalid_argument);
    42. 

  42. #endif
    43. 

  43.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(0, Stream::BEGIN), 0);
    44. 

  44.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    45. 

  45.     MORDOR_TEST_ASSERT(stream.readBuffer() == "cody");
    46. 

  46.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(2, Stream::BEGIN), 2);
    47. 

  47.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    48. 

  48.     MORDOR_TEST_ASSERT(stream.readBuffer() == "dy");
    49. 

  49.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(4, Stream::BEGIN), 4);
    50. 

  50.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    51. 

  51.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    52. 

  52.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(6, Stream::BEGIN), 6);
    53. 

  53.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    54. 

  54.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    55. 

  55. }
    56. 

  56. 

  57. MORDOR_UNITTEST(MemoryStream, forwardSeek)
    58. 

  58. {
    59. 

  59.     MemoryStream stream;
    60. 

  60.     stream.write("cody", 4u);
    61. 

  61.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    62. 

  62.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    63. 

  63.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(2, Stream::BEGIN), 2);
    64. 

  64.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    65. 

  65.     MORDOR_TEST_ASSERT(stream.readBuffer() == "dy");
    66. 

  66.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::CURRENT), 3);
    67. 

  67.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    68. 

  68.     MORDOR_TEST_ASSERT(stream.readBuffer() == "y");
    69. 

  69.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::CURRENT), 4);
    70. 

  70.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    71. 

  71.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    72. 

  72.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::CURRENT), 5);
    73. 

  73.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    74. 

  74.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    75. 

  75. #ifdef X86
    76. 

  76.     MORDOR_TEST_ASSERT_EXCEPTION(stream.seek(0x100000000ull, Stream::CURRENT), std::invalid_argument);
    77. 

  77. #endif
    78. 

  78. }
    79. 

  79. 

  80. MORDOR_UNITTEST(MemoryStream, backwardSeek)
    81. 

  81. {
    82. 

  82.     MemoryStream stream;
    83. 

  83.     stream.write("cody", 4u);
    84. 

  84.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    85. 

  85.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    86. 

  86.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(2, Stream::BEGIN), 2);
    87. 

  87.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    88. 

  88.     MORDOR_TEST_ASSERT(stream.readBuffer() == "dy");
    89. 

  89.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(-1, Stream::CURRENT), 1);
    90. 

  90.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    91. 

  91.     MORDOR_TEST_ASSERT(stream.readBuffer() == "ody");
    92. 

  92.     MORDOR_TEST_ASSERT_EXCEPTION(stream.seek(-2, Stream::CURRENT), std::invalid_argument);
    93. 

  93. }
    94. 

  94. 

  95. MORDOR_UNITTEST(MemoryStream, seekFromEnd)
    96. 

  96. {
    97. 

  97.     MemoryStream stream;
    98. 

  98.     stream.write("cody", 4u);
    99. 

  99.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    100. 

  100.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    101. 

  101.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(0, Stream::END), 4);
    102. 

  102.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    103. 

  103.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    104. 

  104.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::END), 5);
    105. 

  105.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    106. 

  106.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    107. 

  107.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(-2, Stream::END), 2);
    108. 

  108.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    109. 

  109.     MORDOR_TEST_ASSERT(stream.readBuffer() == "dy");
    110. 

  110.     // Should catch an optimized forward seek
    111. 

  111.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(-1, Stream::END), 3);
    112. 

  112.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    113. 

  113.     MORDOR_TEST_ASSERT(stream.readBuffer() == "y");
    114. 

  114.     MORDOR_TEST_ASSERT_EXCEPTION(stream.seek(-5, Stream::CURRENT), std::invalid_argument);
    115. 

  115. }
    116. 

  116. 

  117. MORDOR_UNITTEST(MemoryStream, truncate)
    118. 

  118. {
    119. 

  119.     MemoryStream stream;
    120. 

  120.     stream.write("cody", 4u);
    121. 

  121.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    122. 

  122.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    123. 

  123.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    124. 

  124.     stream.truncate(4);
    125. 

  125.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    126. 

  126.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    127. 

  127.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    128. 

  128.     stream.truncate(3);
    129. 

  129.     MORDOR_TEST_ASSERT(stream.buffer() == "cod");
    130. 

  130.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    131. 

  131.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    132. 

  132.     stream.truncate(5);
    133. 

  133.     MORDOR_TEST_ASSERT(stream.buffer() == std::string("cod\0\0", 5));
    134. 

  134.     MORDOR_TEST_ASSERT(stream.readBuffer() == std::string("\0", 1));
    135. 

  135.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    136. 

  136.     stream.truncate(0);
    137. 

  137.     MORDOR_TEST_ASSERT(stream.buffer() == "");
    138. 

  138.     MORDOR_TEST_ASSERT(stream.readBuffer() == "");
    139. 

  139.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 4);
    140. 

  140. #ifdef X86
    141. 

  141.     MORDOR_TEST_ASSERT_EXCEPTION(stream.truncate(0x100000000ull), std::invalid_argument);
    142. 

  142. #endif
    143. 

  143. }
    144. 

  144. 

  145. MORDOR_UNITTEST(MemoryStream, writeExtension)
    146. 

  146. {
    147. 

  147.     MemoryStream stream;
    148. 

  148.     stream.write("cody", 4u);
    149. 

  149.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    150. 

  150.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::END), 5);
    151. 

  151.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    152. 

  152.     MORDOR_TEST_ASSERT_EQUAL(stream.write("cutrer", 6), 6u);
    153. 

  153.     MORDOR_TEST_ASSERT_EQUAL(stream.size(), 11);
    154. 

  154.     MORDOR_TEST_ASSERT(stream.buffer() == std::string("cody\0cutrer", 11));
    155. 

  155. }
    156. 

  156. 

  157. MORDOR_UNITTEST(MemoryStream, writep1)
    158. 

  158. {
    159. 

  159.     MemoryStream stream;
    160. 

  160.     stream.write("cody", 4u);
    161. 

  161.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    162. 

  162.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::BEGIN), 1);
    163. 

  163.     MORDOR_TEST_ASSERT_EQUAL(stream.write("c", 1), 1u);
    164. 

  164.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 2);
    165. 

  165.     MORDOR_TEST_ASSERT(stream.buffer() == "ccdy");
    166. 

  166. }
    167. 

  167. 

  168. MORDOR_UNITTEST(MemoryStream, writep2)
    169. 

  169. {
    170. 

  170.     MemoryStream stream;
    171. 

  171.     stream.write("cody", 4u);
    172. 

  172.     MORDOR_TEST_ASSERT(stream.buffer() == "cody");
    173. 

  173.     MORDOR_TEST_ASSERT_EQUAL(stream.seek(1, Stream::BEGIN), 1);
    174. 

  174.     MORDOR_TEST_ASSERT_EQUAL(stream.write("cutrer", 6), 6u);
    175. 

  175.     MORDOR_TEST_ASSERT_EQUAL(stream.tell(), 7);
    176. 

  176.     MORDOR_TEST_ASSERT_EQUAL(stream.size(), 7);
    177. 

  177.     MORDOR_TEST_ASSERT(stream.buffer() == "ccutrer");
    178. 

  178. }
    179. 

  179.