test/std/input.output/iostream.format/quoted.manip/quoted.pass.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is dual licensed under the MIT and the University of Illinois Open
   6 // Source Licenses. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 // <iomanip>
  11
  12 // quoted
  13
  14 #include <iomanip>
  15 #include <sstream>
  16 #include <string>
  17 #include <cassert>
  18
  19 #include "test_macros.h"
  20
  21 #if TEST_STD_VER > 11
  22
  23 template <class CharT, class Traits>
  24 bool is_skipws ( const std::basic_istream<CharT, Traits>& is ) {
  25     return ( is.flags() & std::ios_base::skipws ) != 0;
  26     }
  27
  28 template <class CharT, class Traits = std::char_traits<CharT>>
  29 void both_ways ( const CharT *p ) {
  30     std::basic_string<CharT, Traits> str(p);
  31     auto q = std::quoted(str);
  32
  33     std::basic_stringstream<CharT, Traits> ss;
  34     bool skippingws = is_skipws ( ss );
  35     ((void)skippingws); // Prevent unused warning
  36     ss << q;
  37     ss >> q;
  38     }
  39
  40 template <class CharT, class Traits = std::char_traits<CharT>>
  41 void round_trip ( const CharT *p ) {
  42     std::basic_stringstream<CharT, Traits> ss;
  43     bool skippingws = is_skipws ( ss );
  44
  45     ss << std::quoted(p);
  46     std::basic_string<CharT, Traits> s;
  47     ss >> std::quoted(s);
  48     assert ( s == p );
  49     assert ( skippingws == is_skipws ( ss ));
  50     }
  51
  52
  53 template <class CharT, class Traits = std::char_traits<CharT>>
  54 void round_trip_ws ( const CharT *p ) {
  55     std::basic_stringstream<CharT, Traits> ss;
  56     std::noskipws ( ss );
  57     bool skippingws = is_skipws ( ss );
  58
  59     ss << std::quoted(p);
  60     std::basic_string<CharT, Traits> s;
  61     ss >> std::quoted(s);
  62     assert ( s == p );
  63     assert ( skippingws == is_skipws ( ss ));
  64     }
  65
  66 template <class CharT, class Traits = std::char_traits<CharT>>
  67 void round_trip_d ( const CharT *p, char delim ) {
  68     std::basic_stringstream<CharT, Traits> ss;
  69     CharT d(delim);
  70
  71     ss << std::quoted(p, d);
  72     std::basic_string<CharT, Traits> s;
  73     ss >> std::quoted(s, d);
  74     assert ( s == p );
  75     }
  76
  77 template <class CharT, class Traits = std::char_traits<CharT>>
  78 void round_trip_e ( const CharT *p, char escape ) {
  79     std::basic_stringstream<CharT, Traits> ss;
  80     CharT e(escape);
  81
  82     ss << std::quoted(p, CharT('"'), e );
  83     std::basic_string<CharT, Traits> s;
  84     ss >> std::quoted(s, CharT('"'), e );
  85     assert ( s == p );
  86     }
  87
  88
  89 template <class CharT, class Traits = std::char_traits<CharT>>
  90 std::basic_string<CharT, Traits> quote ( const CharT *p, char delim='"', char escape='\\' ) {
  91     std::basic_stringstream<CharT, Traits> ss;
  92     CharT d(delim);
  93     CharT e(escape);
  94     ss << std::quoted(p, d, e);
  95     std::basic_string<CharT, Traits> s;
  96     ss >> s;    // no quote
  97     return s;
  98 }
  99
 100 template <class CharT, class Traits = std::char_traits<CharT>>
 101 std::basic_string<CharT, Traits> unquote ( const CharT *p, char delim='"', char escape='\\' ) {
 102     std::basic_stringstream<CharT, Traits> ss;
 103     ss << p;
 104
 105     CharT d(delim);
 106     CharT e(escape);
 107     std::basic_string<CharT, Traits> s;
 108     ss >> std::quoted(s, d, e);
 109     return s;
 110 }
 111
 112 void test_padding () {
 113     {
 114     std::stringstream ss;
 115     ss << std::left << std::setw(10) << std::setfill('!') << std::quoted("abc", '`');
 116     assert ( ss.str() == "`abc`!!!!!" );
 117     }
 118
 119     {
 120     std::stringstream ss;
 121     ss << std::right << std::setw(10) << std::setfill('!') << std::quoted("abc", '`');
 122     assert ( ss.str() == "!!!!!`abc`" );
 123     }
 124 }
 125
 126
 127 int main()
 128 {
 129     both_ways ( "" );   // This is a compilation check
 130
 131     round_trip    (  "" );
 132     round_trip_ws (  "" );
 133     round_trip_d  (  "", 'q' );
 134     round_trip_e  (  "", 'q' );
 135
 136     round_trip    ( L"" );
 137     round_trip_ws ( L"" );
 138     round_trip_d  ( L"", 'q' );
 139     round_trip_e  ( L"", 'q' );
 140
 141     round_trip    (  "Hi" );
 142     round_trip_ws (  "Hi" );
 143     round_trip_d  (  "Hi", '!' );
 144     round_trip_e  (  "Hi", '!' );
 145     assert ( quote ( "Hi", '!' ) == "!Hi!" );
 146     assert ( quote ( "Hi!", '!' ) == R"(!Hi\!!)" );
 147
 148     round_trip    ( L"Hi" );
 149     round_trip_ws ( L"Hi" );
 150     round_trip_d  ( L"Hi", '!' );
 151     round_trip_e  ( L"Hi", '!' );
 152     assert ( quote ( L"Hi", '!' )  == L"!Hi!" );
 153     assert ( quote ( L"Hi!", '!' ) == LR"(!Hi\!!)" );
 154
 155     round_trip    (  "Hi Mom" );
 156     round_trip_ws (  "Hi Mom" );
 157     round_trip    ( L"Hi Mom" );
 158     round_trip_ws ( L"Hi Mom" );
 159
 160     assert ( quote (  "" )  ==  "\"\"" );
 161     assert ( quote ( L"" )  == L"\"\"" );
 162     assert ( quote (  "a" ) ==  "\"a\"" );
 163     assert ( quote ( L"a" ) == L"\"a\"" );
 164
 165 //  missing end quote - must not hang
 166     assert ( unquote (  "\"abc" ) ==  "abc" );
 167     assert ( unquote ( L"\"abc" ) == L"abc" );
 168
 169     assert ( unquote (  "abc" ) == "abc" ); // no delimiter
 170     assert ( unquote ( L"abc" ) == L"abc" ); // no delimiter
 171     assert ( unquote (  "abc def" ) ==  "abc" ); // no delimiter
 172     assert ( unquote ( L"abc def" ) == L"abc" ); // no delimiter
 173
 174     assert ( unquote (  "" ) ==  "" ); // nothing there
 175     assert ( unquote ( L"" ) == L"" ); // nothing there
 176     test_padding ();
 177     }
 178
 179 #else
 180 int main() {}
 181 #endif