string_ref_test2.cpp 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323
  1. /*
  2. Copyright (c) Marshall Clow 2012-2012.
  3. Distributed under the Boost Software License, Version 1.0. (See accompanying
  4. file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  5. For more information, see http://www.boost.org
  6. */
  7. #include <iostream>
  8. #include <cstring> // for std::strchr
  9. #include <boost/utility/string_ref.hpp>
  10. #include <boost/core/lightweight_test.hpp>
  11. typedef boost::string_ref string_ref;
  12. void ends_with ( const char *arg ) {
  13. const size_t sz = std::strlen ( arg );
  14. string_ref sr ( arg );
  15. string_ref sr2 ( arg );
  16. const char *p = arg;
  17. while ( *p ) {
  18. BOOST_TEST ( sr.ends_with ( p ));
  19. ++p;
  20. }
  21. while ( !sr2.empty ()) {
  22. BOOST_TEST ( sr.ends_with ( sr2 ));
  23. sr2.remove_prefix (1);
  24. }
  25. sr2 = arg;
  26. while ( !sr2.empty ()) {
  27. BOOST_TEST ( sr.ends_with ( sr2 ));
  28. sr2.remove_prefix (1);
  29. }
  30. char ch = sz == 0 ? '\0' : arg [ sz - 1 ];
  31. sr2 = arg;
  32. if ( sz > 0 )
  33. BOOST_TEST ( sr2.ends_with ( ch ));
  34. BOOST_TEST ( !sr2.ends_with ( ++ch ));
  35. BOOST_TEST ( sr2.ends_with ( string_ref ()));
  36. }
  37. void starts_with ( const char *arg ) {
  38. const size_t sz = std::strlen ( arg );
  39. string_ref sr ( arg );
  40. string_ref sr2 ( arg );
  41. const char *p = arg + std::strlen ( arg ) - 1;
  42. while ( p >= arg ) {
  43. std::string foo ( arg, p + 1 );
  44. BOOST_TEST ( sr.starts_with ( foo ));
  45. --p;
  46. }
  47. while ( !sr2.empty ()) {
  48. BOOST_TEST ( sr.starts_with ( sr2 ));
  49. sr2.remove_suffix (1);
  50. }
  51. char ch = *arg;
  52. sr2 = arg;
  53. if ( sz > 0 )
  54. BOOST_TEST ( sr2.starts_with ( ch ));
  55. BOOST_TEST ( !sr2.starts_with ( ++ch ));
  56. BOOST_TEST ( sr2.starts_with ( string_ref ()));
  57. }
  58. void reverse ( const char *arg ) {
  59. // Round trip
  60. string_ref sr1 ( arg );
  61. std::string string1 ( sr1.rbegin (), sr1.rend ());
  62. string_ref sr2 ( string1 );
  63. std::string string2 ( sr2.rbegin (), sr2.rend ());
  64. BOOST_TEST ( std::equal ( sr2.rbegin (), sr2.rend (), arg ));
  65. BOOST_TEST ( string2 == arg );
  66. BOOST_TEST ( std::equal ( sr1.begin (), sr1.end (), string2.begin ()));
  67. }
  68. // This helper function eliminates signed vs. unsigned warnings
  69. string_ref::size_type ptr_diff ( const char *res, const char *base ) {
  70. BOOST_TEST ( res >= base );
  71. return static_cast<string_ref::size_type> ( res - base );
  72. }
  73. void find ( const char *arg ) {
  74. string_ref sr1;
  75. string_ref sr2;
  76. const char *p;
  77. // When we search for the empty string, we find it at position 0
  78. BOOST_TEST ( sr1.find (sr2) == 0 );
  79. BOOST_TEST ( sr1.rfind(sr2) == 0 );
  80. // Look for each character in the string(searching from the start)
  81. p = arg;
  82. sr1 = arg;
  83. while ( *p ) {
  84. string_ref::size_type pos = sr1.find(*p);
  85. BOOST_TEST ( pos != string_ref::npos && ( pos <= ptr_diff ( p, arg )));
  86. ++p;
  87. }
  88. // Look for each character in the string (searching from the end)
  89. p = arg;
  90. sr1 = arg;
  91. while ( *p ) {
  92. string_ref::size_type pos = sr1.rfind(*p);
  93. BOOST_TEST ( pos != string_ref::npos && pos < sr1.size () && ( pos >= ptr_diff ( p, arg )));
  94. ++p;
  95. }
  96. // Look for pairs on characters (searching from the start)
  97. sr1 = arg;
  98. p = arg;
  99. while ( *p && *(p+1)) {
  100. string_ref sr3 ( p, 2 );
  101. string_ref::size_type pos = sr1.find ( sr3 );
  102. BOOST_TEST ( pos != string_ref::npos && pos <= static_cast<string_ref::size_type>( p - arg ));
  103. p++;
  104. }
  105. sr1 = arg;
  106. p = arg;
  107. // for all possible chars, see if we find them in the right place.
  108. // Note that strchr will/might do the _wrong_ thing if we search for NULL
  109. for ( int ch = 1; ch < 256; ++ch ) {
  110. string_ref::size_type pos = sr1.find(ch);
  111. const char *strp = std::strchr ( arg, ch );
  112. BOOST_TEST (( strp == NULL ) == ( pos == string_ref::npos ));
  113. if ( strp != NULL )
  114. BOOST_TEST ( ptr_diff ( strp, arg ) == pos );
  115. }
  116. sr1 = arg;
  117. p = arg;
  118. // for all possible chars, see if we find them in the right place.
  119. // Note that strchr will/might do the _wrong_ thing if we search for NULL
  120. for ( int ch = 1; ch < 256; ++ch ) {
  121. string_ref::size_type pos = sr1.rfind(ch);
  122. const char *strp = std::strrchr ( arg, ch );
  123. BOOST_TEST (( strp == NULL ) == ( pos == string_ref::npos ));
  124. if ( strp != NULL )
  125. BOOST_TEST ( ptr_diff ( strp, arg ) == pos );
  126. }
  127. // Find everything at the start
  128. p = arg;
  129. sr1 = arg;
  130. while ( !sr1.empty ()) {
  131. string_ref::size_type pos = sr1.find(*p);
  132. BOOST_TEST ( pos == 0 );
  133. sr1.remove_prefix (1);
  134. ++p;
  135. }
  136. // Find everything at the end
  137. sr1 = arg;
  138. p = arg + std::strlen ( arg ) - 1;
  139. while ( !sr1.empty ()) {
  140. string_ref::size_type pos = sr1.rfind(*p);
  141. BOOST_TEST ( pos == sr1.size () - 1 );
  142. sr1.remove_suffix (1);
  143. --p;
  144. }
  145. // Find everything at the start
  146. sr1 = arg;
  147. p = arg;
  148. while ( !sr1.empty ()) {
  149. string_ref::size_type pos = sr1.find_first_of(*p);
  150. BOOST_TEST ( pos == 0 );
  151. sr1.remove_prefix (1);
  152. ++p;
  153. }
  154. // Find everything at the end
  155. sr1 = arg;
  156. p = arg + std::strlen ( arg ) - 1;
  157. while ( !sr1.empty ()) {
  158. string_ref::size_type pos = sr1.find_last_of(*p);
  159. BOOST_TEST ( pos == sr1.size () - 1 );
  160. sr1.remove_suffix (1);
  161. --p;
  162. }
  163. // Basic sanity checking for "find_first_of / find_first_not_of"
  164. sr1 = arg;
  165. sr2 = arg;
  166. while ( !sr1.empty() ) {
  167. BOOST_TEST ( sr1.find_first_of ( sr2 ) == 0 );
  168. BOOST_TEST ( sr1.find_first_not_of ( sr2 ) == string_ref::npos );
  169. sr1.remove_prefix ( 1 );
  170. }
  171. p = arg;
  172. sr1 = arg;
  173. while ( *p ) {
  174. string_ref::size_type pos1 = sr1.find_first_of(*p);
  175. string_ref::size_type pos2 = sr1.find_first_not_of(*p);
  176. BOOST_TEST ( pos1 != string_ref::npos && pos1 < sr1.size () && pos1 <= ptr_diff ( p, arg ));
  177. if ( pos2 != string_ref::npos ) {
  178. for ( size_t i = 0 ; i < pos2; ++i )
  179. BOOST_TEST ( sr1[i] == *p );
  180. BOOST_TEST ( sr1 [ pos2 ] != *p );
  181. }
  182. BOOST_TEST ( pos2 != pos1 );
  183. ++p;
  184. }
  185. // Basic sanity checking for "find_last_of / find_last_not_of"
  186. sr1 = arg;
  187. sr2 = arg;
  188. while ( !sr1.empty() ) {
  189. BOOST_TEST ( sr1.find_last_of ( sr2 ) == ( sr1.size () - 1 ));
  190. BOOST_TEST ( sr1.find_last_not_of ( sr2 ) == string_ref::npos );
  191. sr1.remove_suffix ( 1 );
  192. }
  193. p = arg;
  194. sr1 = arg;
  195. while ( *p ) {
  196. string_ref::size_type pos1 = sr1.find_last_of(*p);
  197. string_ref::size_type pos2 = sr1.find_last_not_of(*p);
  198. BOOST_TEST ( pos1 != string_ref::npos && pos1 < sr1.size () && pos1 >= ptr_diff ( p, arg ));
  199. BOOST_TEST ( pos2 == string_ref::npos || pos1 < sr1.size ());
  200. if ( pos2 != string_ref::npos ) {
  201. for ( size_t i = sr1.size () -1 ; i > pos2; --i )
  202. BOOST_TEST ( sr1[i] == *p );
  203. BOOST_TEST ( sr1 [ pos2 ] != *p );
  204. }
  205. BOOST_TEST ( pos2 != pos1 );
  206. ++p;
  207. }
  208. }
  209. void to_string ( const char *arg ) {
  210. string_ref sr1;
  211. std::string str1;
  212. std::string str2;
  213. str1.assign ( arg );
  214. sr1 = arg;
  215. // str2 = sr1.to_string<std::allocator<char> > ();
  216. str2 = sr1.to_string ();
  217. BOOST_TEST ( str1 == str2 );
  218. #ifndef BOOST_NO_CXX11_EXPLICIT_CONVERSION_OPERATORS
  219. std::string str3 = static_cast<std::string> ( sr1 );
  220. BOOST_TEST ( str1 == str3 );
  221. #endif
  222. }
  223. void compare ( const char *arg ) {
  224. string_ref sr1;
  225. std::string str1;
  226. std::string str2 = str1;
  227. str1.assign ( arg );
  228. sr1 = arg;
  229. BOOST_TEST ( sr1 == sr1); // compare string_ref and string_ref
  230. BOOST_TEST ( sr1 == str1); // compare string and string_ref
  231. BOOST_TEST ( str1 == sr1 ); // compare string_ref and string
  232. BOOST_TEST ( sr1 == arg ); // compare string_ref and pointer
  233. BOOST_TEST ( arg == sr1 ); // compare pointer and string_ref
  234. if ( sr1.size () > 0 ) {
  235. (*str1.rbegin())++;
  236. BOOST_TEST ( sr1 != str1 );
  237. BOOST_TEST ( str1 != sr1 );
  238. BOOST_TEST ( sr1 < str1 );
  239. BOOST_TEST ( sr1 <= str1 );
  240. BOOST_TEST ( str1 > sr1 );
  241. BOOST_TEST ( str1 >= sr1 );
  242. (*str1.rbegin()) -= 2;
  243. BOOST_TEST ( sr1 != str1 );
  244. BOOST_TEST ( str1 != sr1 );
  245. BOOST_TEST ( sr1 > str1 );
  246. BOOST_TEST ( sr1 >= str1 );
  247. BOOST_TEST ( str1 < sr1 );
  248. BOOST_TEST ( str1 <= sr1 );
  249. }
  250. }
  251. const char *test_strings [] = {
  252. "",
  253. "0",
  254. "abc",
  255. "AAA", // all the same
  256. "adsfadadiaef;alkdg;aljt;j agl;sjrl;tjs;lga;lretj;srg[w349u5209dsfadfasdfasdfadsf",
  257. "abc\0asdfadsfasf",
  258. NULL
  259. };
  260. int main()
  261. {
  262. const char **p = &test_strings[0];
  263. while ( *p != NULL ) {
  264. starts_with ( *p );
  265. ends_with ( *p );
  266. reverse ( *p );
  267. find ( *p );
  268. to_string ( *p );
  269. compare ( *p );
  270. p++;
  271. }
  272. return boost::report_errors();
  273. }