stack_test.cpp 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219
  1. // Copyright (C) 2011 Tim Blechmann
  2. //
  3. // Distributed under the Boost Software License, Version 1.0. (See
  4. // accompanying file LICENSE_1_0.txt or copy at
  5. // http://www.boost.org/LICENSE_1_0.txt)
  6. #include <boost/thread.hpp>
  7. #include <boost/lockfree/stack.hpp>
  8. #define BOOST_TEST_MAIN
  9. #ifdef BOOST_LOCKFREE_INCLUDE_TESTS
  10. #include <boost/test/included/unit_test.hpp>
  11. #else
  12. #include <boost/test/unit_test.hpp>
  13. #endif
  14. #include "test_helpers.hpp"
  15. BOOST_AUTO_TEST_CASE( simple_stack_test )
  16. {
  17. boost::lockfree::stack<long> stk(128);
  18. stk.push(1);
  19. stk.push(2);
  20. long out;
  21. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  22. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  23. BOOST_REQUIRE(!stk.pop(out));
  24. }
  25. BOOST_AUTO_TEST_CASE( unsafe_stack_test )
  26. {
  27. boost::lockfree::stack<long> stk(128);
  28. stk.unsynchronized_push(1);
  29. stk.unsynchronized_push(2);
  30. long out;
  31. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  32. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  33. BOOST_REQUIRE(!stk.unsynchronized_pop(out));
  34. }
  35. BOOST_AUTO_TEST_CASE( ranged_push_test )
  36. {
  37. boost::lockfree::stack<long> stk(128);
  38. long data[2] = {1, 2};
  39. BOOST_REQUIRE_EQUAL(stk.push(data, data + 2), data + 2);
  40. long out;
  41. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  42. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  43. BOOST_REQUIRE(!stk.unsynchronized_pop(out));
  44. }
  45. BOOST_AUTO_TEST_CASE( ranged_unsynchronized_push_test )
  46. {
  47. boost::lockfree::stack<long> stk(128);
  48. long data[2] = {1, 2};
  49. BOOST_REQUIRE_EQUAL(stk.unsynchronized_push(data, data + 2), data + 2);
  50. long out;
  51. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  52. BOOST_REQUIRE(stk.unsynchronized_pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  53. BOOST_REQUIRE(!stk.unsynchronized_pop(out));
  54. }
  55. BOOST_AUTO_TEST_CASE( fixed_size_stack_test )
  56. {
  57. boost::lockfree::stack<long, boost::lockfree::capacity<128> > stk;
  58. stk.push(1);
  59. stk.push(2);
  60. long out;
  61. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  62. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  63. BOOST_REQUIRE(!stk.pop(out));
  64. BOOST_REQUIRE(stk.empty());
  65. }
  66. BOOST_AUTO_TEST_CASE( fixed_size_stack_test_exhausted )
  67. {
  68. boost::lockfree::stack<long, boost::lockfree::capacity<2> > stk;
  69. stk.push(1);
  70. stk.push(2);
  71. BOOST_REQUIRE(!stk.push(3));
  72. long out;
  73. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  74. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  75. BOOST_REQUIRE(!stk.pop(out));
  76. BOOST_REQUIRE(stk.empty());
  77. }
  78. BOOST_AUTO_TEST_CASE( bounded_stack_test_exhausted )
  79. {
  80. boost::lockfree::stack<long> stk(2);
  81. stk.bounded_push(1);
  82. stk.bounded_push(2);
  83. BOOST_REQUIRE(!stk.bounded_push(3));
  84. long out;
  85. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 2);
  86. BOOST_REQUIRE(stk.pop(out)); BOOST_REQUIRE_EQUAL(out, 1);
  87. BOOST_REQUIRE(!stk.pop(out));
  88. BOOST_REQUIRE(stk.empty());
  89. }
  90. BOOST_AUTO_TEST_CASE( stack_consume_one_test )
  91. {
  92. boost::lockfree::stack<int> f(64);
  93. BOOST_WARN(f.is_lock_free());
  94. BOOST_REQUIRE(f.empty());
  95. f.push(1);
  96. f.push(2);
  97. #ifdef BOOST_NO_CXX11_LAMBDAS
  98. bool success1 = f.consume_one(test_equal(2));
  99. bool success2 = f.consume_one(test_equal(1));
  100. #else
  101. bool success1 = f.consume_one([] (int i) {
  102. BOOST_REQUIRE_EQUAL(i, 2);
  103. });
  104. bool success2 = f.consume_one([] (int i) {
  105. BOOST_REQUIRE_EQUAL(i, 1);
  106. });
  107. #endif
  108. BOOST_REQUIRE(success1);
  109. BOOST_REQUIRE(success2);
  110. BOOST_REQUIRE(f.empty());
  111. }
  112. BOOST_AUTO_TEST_CASE( stack_consume_all_test )
  113. {
  114. boost::lockfree::stack<int> f(64);
  115. BOOST_WARN(f.is_lock_free());
  116. BOOST_REQUIRE(f.empty());
  117. f.push(1);
  118. f.push(2);
  119. #ifdef BOOST_NO_CXX11_LAMBDAS
  120. size_t consumed = f.consume_all(dummy_functor());
  121. #else
  122. size_t consumed = f.consume_all([] (int i) {
  123. });
  124. #endif
  125. BOOST_REQUIRE_EQUAL(consumed, 2u);
  126. BOOST_REQUIRE(f.empty());
  127. }
  128. BOOST_AUTO_TEST_CASE( stack_consume_all_atomic_test )
  129. {
  130. boost::lockfree::stack<int> f(64);
  131. BOOST_WARN(f.is_lock_free());
  132. BOOST_REQUIRE(f.empty());
  133. f.push(1);
  134. f.push(2);
  135. f.push(3);
  136. #ifdef BOOST_NO_CXX11_LAMBDAS
  137. size_t consumed = f.consume_all_atomic(dummy_functor());
  138. #else
  139. size_t consumed = f.consume_all_atomic([] (int i) {
  140. });
  141. #endif
  142. BOOST_REQUIRE_EQUAL(consumed, 3u);
  143. BOOST_REQUIRE(f.empty());
  144. }
  145. BOOST_AUTO_TEST_CASE( stack_consume_all_atomic_reversed_test )
  146. {
  147. boost::lockfree::stack<int> f(64);
  148. BOOST_WARN(f.is_lock_free());
  149. BOOST_REQUIRE(f.empty());
  150. f.push(1);
  151. f.push(2);
  152. f.push(3);
  153. #ifdef BOOST_NO_CXX11_LAMBDAS
  154. size_t consumed = f.consume_all_atomic_reversed(dummy_functor());
  155. #else
  156. size_t consumed = f.consume_all_atomic_reversed([] (int i) {
  157. });
  158. #endif
  159. BOOST_REQUIRE_EQUAL(consumed, 3u);
  160. BOOST_REQUIRE(f.empty());
  161. }
  162. BOOST_AUTO_TEST_CASE( reserve_test )
  163. {
  164. typedef boost::lockfree::stack< void* > memory_stack;
  165. memory_stack ms(1);
  166. ms.reserve(1);
  167. ms.reserve_unsafe(1);
  168. }