| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 ********************************************************************** | |
| 3 * Copyright (C) 2002-2011, International Business Machines | |
| 4 * Corporation and others. All Rights Reserved. | |
| 5 ********************************************************************** | |
| 6 * file name: iotest.cpp | |
| 7 * encoding: US-ASCII | |
| 8 * tab size: 8 (not used) | |
| 9 * indentation:4 | |
| 10 * | |
| 11 * created on: 2002feb21 | |
| 12 * created by: George Rhoten | |
| 13 */ | |
| 14 | |
| 15 | |
| 16 #include "unicode/ustream.h" | |
| 17 | |
| 18 #include "unicode/ucnv.h" | |
| 19 #include "unicode/ustring.h" | |
| 20 #include "ustr_cnv.h" | |
| 21 #include "iotest.h" | |
| 22 | |
| 23 #if U_IOSTREAM_SOURCE >= 199711 | |
| 24 #if defined(__GNUC__) && __GNUC__ >= 4 | |
| 25 #define USE_SSTREAM 1 | |
| 26 #include <sstream> | |
| 27 #else | |
| 28 // <strstream> is deprecated on some platforms, and the compiler complains very
loudly if you use it. | |
| 29 #include <strstream> | |
| 30 #endif | |
| 31 #include <fstream> | |
| 32 #include <iomanip> | |
| 33 using namespace std; | |
| 34 | |
| 35 #include <string.h> | |
| 36 | |
| 37 U_CDECL_BEGIN | |
| 38 #if U_PLATFORM_USES_ONLY_WIN32_API | |
| 39 const UChar NEW_LINE[] = {0x0d,0x0a,0}; | |
| 40 const char C_NEW_LINE[] = {0x0d,0x0a,0}; | |
| 41 #define UTF8_NEW_LINE "\x0d\x0a" | |
| 42 #else | |
| 43 const UChar NEW_LINE[] = {0x0a,0}; | |
| 44 const char C_NEW_LINE[] = {'\n',0}; | |
| 45 #define UTF8_NEW_LINE "\x0a" | |
| 46 #endif | |
| 47 U_CDECL_END | |
| 48 | |
| 49 U_CDECL_BEGIN | |
| 50 static void U_CALLCONV TestStream(void) | |
| 51 { | |
| 52 const UChar thisMu[] = { 0x74, 0x48, 0x69, 0x73, 0x3BC, 0}; | |
| 53 const UChar mu[] = { 0x6D, 0x75, 0}; | |
| 54 UnicodeString str1 = UNICODE_STRING_SIMPLE("str1"); | |
| 55 UnicodeString str2 = UNICODE_STRING_SIMPLE(" <<"); | |
| 56 UnicodeString str3 = UNICODE_STRING_SIMPLE("2"); | |
| 57 UnicodeString str4 = UNICODE_STRING_SIMPLE(" UTF-8 "); | |
| 58 UnicodeString inStr = UNICODE_STRING_SIMPLE(" UTF-8 "); | |
| 59 UnicodeString inStr2; | |
| 60 char defConvName[UCNV_MAX_CONVERTER_NAME_LENGTH*2]; | |
| 61 char inStrC[128]; | |
| 62 UErrorCode status = U_ZERO_ERROR; | |
| 63 UConverter *defConv; | |
| 64 static const char testStr[] = "\x42\x65\x67\x69\x6E\x6E\x69\x6E\x67\x20\x6F\
x66\x20\x74\x65\x73\x74\x20\x73\x74\x72\x31\x20\x20\x20\x3C\x3C\x32\x31\x20" UTF
8_NEW_LINE "\x20\x55\x54\x46\x2D\x38\x20\xCE\xBC\xF0\x90\x80\x81\xF0\x90\x80\x82
"; | |
| 65 | |
| 66 str4.append((UChar32)0x03BC); /* mu */ | |
| 67 str4.append((UChar32)0x10001); | |
| 68 str4.append((UChar32)0x10002); | |
| 69 | |
| 70 /* release the default converter and use utf-8 for a bit */ | |
| 71 defConv = u_getDefaultConverter(&status); | |
| 72 if (U_FAILURE(status)) { | |
| 73 log_err("Can't get default converter\n"); | |
| 74 return; | |
| 75 } | |
| 76 ucnv_close(defConv); | |
| 77 strncpy(defConvName, ucnv_getDefaultName(), sizeof(defConvName)/sizeof(defCo
nvName[0])); | |
| 78 ucnv_setDefaultName("UTF-8"); | |
| 79 | |
| 80 static const char * const TESTSTRING = "\x20\x74\x48\x69\x73\xCE\xBC\xE2\x80
\x82\x20\x6D\x75\x20\x77\x6F\x72\x6C\x64"; | |
| 81 #ifdef USE_SSTREAM | |
| 82 ostringstream outTestStream; | |
| 83 istringstream inTestStream(TESTSTRING); | |
| 84 #else | |
| 85 char testStreamBuf[512]; | |
| 86 ostrstream outTestStream(testStreamBuf, sizeof(testStreamBuf)); | |
| 87 istrstream inTestStream(TESTSTRING, 0); | |
| 88 | |
| 89 /* initialize testStreamBuf */ | |
| 90 memset(testStreamBuf, '*', sizeof(testStreamBuf)); | |
| 91 testStreamBuf[sizeof(testStreamBuf)-1] = 0; | |
| 92 #endif | |
| 93 | |
| 94 outTestStream << "\x42\x65\x67\x69\x6E\x6E\x69\x6E\x67\x20\x6F\x66\x20\x74\x
65\x73\x74\x20"; | |
| 95 outTestStream << str1 << "\x20\x20" << str2 << str3 << "\x31\x20" << UTF8_NE
W_LINE << str4 << ends; | |
| 96 #ifdef USE_SSTREAM | |
| 97 string tempStr = outTestStream.str(); | |
| 98 const char *testStreamBuf = tempStr.c_str(); | |
| 99 #endif | |
| 100 if (strcmp(testStreamBuf, testStr) != 0) { | |
| 101 log_err("Got: \"%s\", Expected: \"%s\"\n", testStreamBuf, testStr); | |
| 102 } | |
| 103 | |
| 104 inTestStream >> inStr >> inStr2; | |
| 105 if (inStr.compare(thisMu) != 0) { | |
| 106 u_austrncpy(inStrC, inStr.getBuffer(), inStr.length()); | |
| 107 inStrC[inStr.length()] = 0; | |
| 108 log_err("Got: \"%s\", Expected: \"tHis\\u03BC\"\n", inStrC); | |
| 109 } | |
| 110 if (inStr2.compare(mu) != 0) { | |
| 111 u_austrncpy(inStrC, inStr.getBuffer(), inStr.length()); | |
| 112 inStrC[inStr.length()] = 0; | |
| 113 log_err("Got: \"%s\", Expected: \"mu\"\n", inStrC); | |
| 114 } | |
| 115 | |
| 116 /* return the default converter to the original state. */ | |
| 117 ucnv_setDefaultName(defConvName); | |
| 118 defConv = u_getDefaultConverter(&status); | |
| 119 if (U_FAILURE(status)) { | |
| 120 log_err("Can't get default converter"); | |
| 121 return; | |
| 122 } | |
| 123 | |
| 124 /* Test formatting when using '<<' and UnicodeString */ | |
| 125 #ifdef USE_SSTREAM | |
| 126 ostringstream outFormatStream; | |
| 127 #else | |
| 128 char testFormatStreamBuf[512]; | |
| 129 memset(testFormatStreamBuf, 0, sizeof(testFormatStreamBuf)); | |
| 130 ostrstream outFormatStream(testFormatStreamBuf, sizeof(testFormatStreamBuf))
; | |
| 131 #endif | |
| 132 UnicodeString ustr("string"); | |
| 133 | |
| 134 outFormatStream << "1234567890" << setw(10) << left << ustr << " " << "01234
56789"; | |
| 135 | |
| 136 #ifdef USE_SSTREAM | |
| 137 tempStr = outFormatStream.str(); | |
| 138 const char *testFormatStreamBuf = tempStr.c_str(); | |
| 139 #endif | |
| 140 const char *format_test_expected = "1234567890string 0123456789"; | |
| 141 if (strcmp(format_test_expected, testFormatStreamBuf) != 0) { | |
| 142 log_err("UnicodeString format test using << operator Got: '%s' Expected:
'%s'\n", testFormatStreamBuf, format_test_expected); | |
| 143 } | |
| 144 | |
| 145 /* Test large buffer (size > 200) when using '<<' and UnicodeString */ | |
| 146 #ifdef USE_SSTREAM | |
| 147 ostringstream outLargeStream; | |
| 148 #else | |
| 149 char testLargeStreamBuf[512]; | |
| 150 memset(testLargeStreamBuf, 0, sizeof(testLargeStreamBuf)); | |
| 151 ostrstream outLargeStream(testLargeStreamBuf, sizeof(testLargeStreamBuf)); | |
| 152 #endif | |
| 153 UChar large_array[200]; | |
| 154 int32_t large_array_length = sizeof(large_array)/sizeof(UChar); | |
| 155 for (int32_t i = 0; i < large_array_length; i++) { | |
| 156 large_array[i] = 0x41; | |
| 157 } | |
| 158 UnicodeString large_array_unistr(large_array, large_array_length); | |
| 159 | |
| 160 outLargeStream << large_array_unistr; | |
| 161 | |
| 162 #ifdef USE_SSTREAM | |
| 163 string tmpString = outLargeStream.str(); | |
| 164 const char *testLargeStreamBuf = tmpString.c_str(); | |
| 165 #endif | |
| 166 char expectedLargeStreamBuf[300]; | |
| 167 int32_t expectedBufLength = sizeof(expectedLargeStreamBuf); | |
| 168 | |
| 169 ucnv_fromUChars(defConv, expectedLargeStreamBuf, expectedBufLength, large_ar
ray, large_array_length, &status); | |
| 170 if (U_SUCCESS(status)) { | |
| 171 if (strcmp(testLargeStreamBuf, expectedLargeStreamBuf) != 0) { | |
| 172 log_err("Large UnicodeString operator << output incorrect.\n"); | |
| 173 } | |
| 174 } else { | |
| 175 log_err("Error converting string for large stream buffer testing.\n"); | |
| 176 } | |
| 177 ucnv_close(defConv); | |
| 178 } | |
| 179 | |
| 180 #define IOSTREAM_GOOD_SHIFT 3 | |
| 181 #define IOSTREAM_GOOD (1<<IOSTREAM_GOOD_SHIFT) | |
| 182 #define IOSTREAM_BAD_SHIFT 2 | |
| 183 #define IOSTREAM_BAD (1<<IOSTREAM_BAD_SHIFT) | |
| 184 #define IOSTREAM_EOF_SHIFT 1 | |
| 185 #define IOSTREAM_EOF (1<<IOSTREAM_EOF_SHIFT) | |
| 186 #define IOSTREAM_FAIL_SHIFT 0 | |
| 187 #define IOSTREAM_FAIL (1<<IOSTREAM_FAIL_SHIFT) | |
| 188 | |
| 189 static int32_t getBitStatus(const iostream& stream) { | |
| 190 return (stream.good()<<IOSTREAM_GOOD_SHIFT) | |
| 191 | (stream.bad()<<IOSTREAM_BAD_SHIFT) | |
| 192 | (stream.eof()<<IOSTREAM_EOF_SHIFT) | |
| 193 | (stream.fail()<<IOSTREAM_FAIL_SHIFT); | |
| 194 } | |
| 195 | |
| 196 void | |
| 197 printBits(const iostream& stream) | |
| 198 { | |
| 199 int32_t status = getBitStatus(stream); | |
| 200 log_verbose("status 0x%02X (", status); | |
| 201 if (status & IOSTREAM_GOOD) { | |
| 202 log_verbose("good"); | |
| 203 } | |
| 204 if (status & IOSTREAM_BAD) { | |
| 205 log_verbose("bad"); | |
| 206 } | |
| 207 if (status & IOSTREAM_EOF) { | |
| 208 log_verbose("eof"); | |
| 209 } | |
| 210 if (status & IOSTREAM_FAIL) { | |
| 211 log_verbose("fail"); | |
| 212 } | |
| 213 log_verbose(")\n"); | |
| 214 } | |
| 215 | |
| 216 void | |
| 217 testString( | |
| 218 UnicodeString& str, | |
| 219 const char* testString, | |
| 220 const UChar* expectedString, | |
| 221 int32_t expectedStatus) | |
| 222 { | |
| 223 #ifdef USE_SSTREAM | |
| 224 stringstream sstrm; | |
| 225 #else | |
| 226 strstream sstrm; | |
| 227 #endif | |
| 228 | |
| 229 sstrm << testString; | |
| 230 | |
| 231 /*log_verbose("iostream before operator::>>() call \"%s\" ", testString); | |
| 232 printBits(sstrm);*/ | |
| 233 | |
| 234 sstrm >> str; | |
| 235 | |
| 236 log_verbose("iostream after operator::>>() call \"%s\" ", testString); | |
| 237 printBits(sstrm); | |
| 238 | |
| 239 if (getBitStatus(sstrm) != expectedStatus) { | |
| 240 printBits(sstrm); | |
| 241 log_err("Expected status %d, Got %d. See verbose output for details\n",
expectedStatus, getBitStatus(sstrm)); | |
| 242 } | |
| 243 if (str != UnicodeString(expectedString)) { | |
| 244 log_err("Did not get expected results from \"%s\", expected \"%s\"\n", t
estString, expectedString); | |
| 245 } | |
| 246 } | |
| 247 | |
| 248 | |
| 249 static void U_CALLCONV TestStreamEOF(void) | |
| 250 { | |
| 251 UnicodeString dest; | |
| 252 fstream fs(STANDARD_TEST_FILE, fstream::in | fstream::out | fstream::trunc); | |
| 253 #ifdef USE_SSTREAM | |
| 254 stringstream ss; | |
| 255 #else | |
| 256 strstream ss; | |
| 257 #endif | |
| 258 | |
| 259 fs << "EXAMPLE"; | |
| 260 fs.seekg(0); | |
| 261 ss << "EXAMPLE"; | |
| 262 | |
| 263 if (!(fs >> dest)) { | |
| 264 log_err("Reading of file did not return expected status result\n"); | |
| 265 } | |
| 266 if (dest != "EXAMPLE") { | |
| 267 log_err("Reading of file did not return expected string\n"); | |
| 268 } | |
| 269 | |
| 270 if (!(ss >> dest)) { | |
| 271 log_err("Reading of string did not return expected status result\n"); | |
| 272 } | |
| 273 if (dest != "EXAMPLE") { | |
| 274 log_err("Reading of string did not return expected string\n"); | |
| 275 } | |
| 276 fs.close(); | |
| 277 | |
| 278 log_verbose("Testing operator >> for UnicodeString...\n"); | |
| 279 | |
| 280 /* The test cases needs to be converted to the default codepage. However, t
he stream operator needs char* so U_STRING_* is called. */ | |
| 281 U_STRING_DECL(testCase1, "", 0); | |
| 282 U_STRING_INIT(testCase1, "", 0); | |
| 283 U_STRING_DECL(testCase2, "foo", 3); | |
| 284 U_STRING_INIT(testCase2, "foo", 3); | |
| 285 U_STRING_DECL(testCase3, " ", 3); | |
| 286 U_STRING_INIT(testCase3, " ", 3); | |
| 287 U_STRING_DECL(testCase4, " bar", 6); | |
| 288 U_STRING_INIT(testCase4, " bar", 6); | |
| 289 U_STRING_DECL(testCase5, "bar ", 6); | |
| 290 U_STRING_INIT(testCase5, "bar ", 6); | |
| 291 U_STRING_DECL(testCase6, " bar ", 9); | |
| 292 U_STRING_INIT(testCase6, " bar ", 9); | |
| 293 | |
| 294 | |
| 295 U_STRING_DECL(expectedResultA, "", 0); | |
| 296 U_STRING_INIT(expectedResultA, "", 0); | |
| 297 U_STRING_DECL(expectedResultB, "foo", 3); | |
| 298 U_STRING_INIT(expectedResultB, "foo", 3); | |
| 299 U_STRING_DECL(expectedResultC, "unchanged", 9); | |
| 300 U_STRING_INIT(expectedResultC, "unchanged", 9); | |
| 301 U_STRING_DECL(expectedResultD, "bar", 3); | |
| 302 U_STRING_INIT(expectedResultD, "bar", 3); | |
| 303 | |
| 304 | |
| 305 UnicodeString UStr; | |
| 306 UnicodeString expectedResults; | |
| 307 char testcase[10]; | |
| 308 testString(UStr, u_austrcpy(testcase, testCase1), expectedResultA, IOSTREAM_
EOF|IOSTREAM_FAIL); | |
| 309 testString(UStr, u_austrcpy(testcase, testCase2), expectedResultB, IOSTREAM_
EOF); | |
| 310 UStr = UnicodeString(expectedResultC); | |
| 311 testString(UStr, u_austrcpy(testcase, testCase3), expectedResultC, IOSTREAM_
EOF|IOSTREAM_FAIL); | |
| 312 testString(UStr, u_austrcpy(testcase, testCase4), expectedResultD, IOSTREAM_
EOF); | |
| 313 testString(UStr, u_austrcpy(testcase, testCase5), expectedResultD, IOSTREAM_
GOOD); | |
| 314 testString(UStr, u_austrcpy(testcase, testCase6), expectedResultD, IOSTREAM_
GOOD); | |
| 315 } | |
| 316 U_CDECL_END | |
| 317 | |
| 318 U_CFUNC void addStreamTests(TestNode** root) { | |
| 319 addTest(root, &TestStream, "stream/TestStream"); | |
| 320 addTest(root, &TestStreamEOF, "stream/TestStreamEOF"); | |
| 321 } | |
| 322 #endif | |
| OLD | NEW |