PageRenderTime 2882ms CodeModel.GetById 16ms RepoModel.GetById 0ms app.codeStats 0ms

/branches/1.3-wix3.0/src/CustomActions/StringTools/StringToolsImplUnitTests.cpp

#
C++ | 341 lines | 269 code | 66 blank | 6 comment | 32 complexity | d3ea5c96a4c16be978cdaa1a0b1f8eef MD5 | raw file
Possible License(s): LGPL-2.0, EPL-1.0, LGPL-2.1 
    

  1. #include "StdAfxUnitTests.h"

    2. 

  2. #include "StringToolsImplUnitTests.h"

    3. 

  3. #include <String/StringUtils.h>

    4. 

  4. 

    5. 

  5. CPPUNIT_TEST_SUITE_REGISTRATION(AppSecInc::UnitTests::StringTools::StringToolsImplUnitTests);

    6. 

  6. 

    7. 

  7. using namespace AppSecInc::Msi;

    8. 

  8. using namespace AppSecInc::UnitTests::StringTools;

    9. 

  9. 

    10. 

  10. void StringToolsImplUnitTests::Test_EntryPoints()

    11. 

  11. {

    12. 

  12. 	HMODULE h = ::LoadLibrary(L"StringTools.dll");

    13. 

  13. 	CPPUNIT_ASSERT_MESSAGE("Missing StringTools.dll", h != NULL);

    14. 

  14. 	CPPUNIT_ASSERT(NULL != GetProcAddress(h, "String_Replace"));

    15. 

  15. 	CPPUNIT_ASSERT_MESSAGE("Missing String_ToLower", NULL != GetProcAddress(h, "String_ToLower"));

    16. 

  16. 	CPPUNIT_ASSERT_MESSAGE("Missing String_ToUpper", NULL != GetProcAddress(h, "String_ToUpper"));

    17. 

  17. 	CPPUNIT_ASSERT_MESSAGE("Missing String_Trim", NULL != GetProcAddress(h, "String_Trim"));

    18. 

  18. 	CPPUNIT_ASSERT_MESSAGE("Missing String_LTrim", NULL != GetProcAddress(h, "String_LTrim"));

    19. 

  19. 	CPPUNIT_ASSERT_MESSAGE("Missing String_RTrim", NULL != GetProcAddress(h, "String_RTrim"));

    20. 

  20. 	::FreeLibrary(h);

    21. 

  21. }

    22. 

  22. 

    23. 

  23. void StringToolsImplUnitTests::Test_StringReplace()

    24. 

  24. {

    25. 

  25. 	AppSecInc::Msi::MsiShim hInstall;

    26. 

  26.     MsiInstall msiInstall(hInstall);

    27. 

  27. 

    28. 

  28. 	typedef struct  

    29. 

  29. 	{

    30. 

  30. 		LPCSTR testIn;

    31. 

  31. 		LPCSTR testFrom;

    32. 

  32. 		LPCSTR testTo;

    33. 

  33. 		int testExpectedResult;

    34. 

  34. 		LPCSTR testExpectedOutput;

    35. 

  35. 	} TestData;

    36. 

  36. 

    37. 

  37. 	TestData testData[] = 

    38. 

  38. 	{

    39. 

  39. 		{ "", "", "", 0, "" },

    40. 

  40. 		{ "x", "x", "y", 1, "y" },

    41. 

  41. 		{ "x", "xx", "y", 0, "x" },

    42. 

  42. 		{ "foo", "fo", "f", 1, "fo" },

    43. 

  43. 		{ "{source}\r", "source", "target", 1, "{target}\r" },

    44. 

  44. 		{ "{source}\r", "source", "target", 1, "{target}\r" },

    45. 

  45. 		{ "\n{source}\r", "source", "target", 1, "\n{target}\r" },

    46. 

  46. 		{ "\n{source}{source}\r", "source", "target", 2, "\n{target}{target}\r" },

    47. 

  47. 		{ "\n{source}XZY{source}\r", "source", "target", 2, "\n{target}XZY{target}\r" }

    48. 

  48. 	};

    49. 

  49. 

    50. 

  50. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    51. 

  51. 	{

    52. 

  52. 		std::stringstream message;

    53. 

  53. 		message << "'" << testData[i].testIn << "'.replace('" << 

    54. 

  54. 			testData[i].testFrom << "', '" << testData[i].testTo << "') didn't return '" << 

    55. 

  55. 			testData[i].testExpectedResult << "'";

    56. 

  56. 		

    57. 

  57. 		msiInstall.SetProperty("STRING_VALUE_FROM", testData[i].testIn);

    58. 

  58. 		msiInstall.SetProperty("STRING_REPLACE_FROM", testData[i].testFrom);

    59. 

  59. 		msiInstall.SetProperty("STRING_REPLACE_TO", testData[i].testTo);

    60. 

  60. 

    61. 

  61. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_Replace"));

    62. 

  62. 

    63. 

  63. 		int replacement_count = atoi(msiInstall.GetProperty("STRING_REPLACE_COUNT").c_str());

    64. 

  64. 		std::string output = msiInstall.GetProperty("STRING_VALUE_TO");

    65. 

  65. 

    66. 

  66. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(),

    67. 

  67. 			testData[i].testExpectedResult == replacement_count);

    68. 

  68. 

    69. 

  69. 		message << "'" << testData[i].testIn << "'.replace('" << 

    70. 

  70. 			testData[i].testFrom << "', '" << testData[i].testTo << "') didn't return '" << 

    71. 

  71. 			testData[i].testExpectedOutput << "' (returned '" << output << "')";

    72. 

  72. 

    73. 

  73. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(),

    74. 

  74. 			testData[i].testExpectedOutput == output);

    75. 

  75. 	}

    76. 

  76. }

    77. 

  77. 

    78. 

  78. void StringToolsImplUnitTests::Test_StringTrim()

    79. 

  79. {

    80. 

  80. 	AppSecInc::Msi::MsiShim hInstall;

    81. 

  81.     MsiInstall msiInstall(hInstall);

    82. 

  82. 

    83. 

  83. 	typedef struct  

    84. 

  84. 	{

    85. 

  85. 		LPCSTR testInput;

    86. 

  86. 		LPCSTR testWhitespaces;

    87. 

  87. 		LPCSTR testExpectedResult;

    88. 

  88. 	} TestData;

    89. 

  89. 

    90. 

  90. 	TestData testData[] = 

    91. 

  91. 	{

    92. 

  92. 		{ "     this is a string whith whitespaces\t     ", "", "this is a string whith whitespaces" },

    93. 

  93. 		{ "     this is a string whith whitespaces     ", " ", "this is a string whith whitespaces" },

    94. 

  94. 		{ "this string doesn't have whitespaces", "", "this string doesn't have whitespaces" },

    95. 

  95. 		{ "%%test string%%", "%", "test string" },

    96. 

  96. 		{ " test string     ", "", "test string" }

    97. 

  97. 	};

    98. 

  98. 

    99. 

  99. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    100. 

  100. 	{		

    101. 

  101. 		msiInstall.SetProperty("STRING_TRIM_INPUT", testData[i].testInput);

    102. 

  102. 		msiInstall.SetProperty("STRING_TRIM_WHITESPACES", testData[i].testWhitespaces);

    103. 

  103. 

    104. 

  104. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_Trim"));

    105. 

  105. 

    106. 

  106. 		std::string stringTrimResult = msiInstall.GetProperty("STRING_TRIM_RESULT");

    107. 

  107. 		std::stringstream message;

    108. 

  108. 

    109. 

  109. 		message << "StringTrim('" << testData[i].testInput <<

    110. 

  110. 			"', '" << testData[i].testWhitespaces << "') returned '" << stringTrimResult <<

    111. 

  111. 			"'. It was expected: '" << testData[i].testExpectedResult << "'";

    112. 

  112. 

    113. 

  113. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(), testData[i].testExpectedResult == stringTrimResult);

    114. 

  114. 	}

    115. 

  115. }

    116. 

  116. 

    117. 

  117. void StringToolsImplUnitTests::Test_StringLTrim()

    118. 

  118. {

    119. 

  119. 	AppSecInc::Msi::MsiShim hInstall;

    120. 

  120.     MsiInstall msiInstall(hInstall);

    121. 

  121. 

    122. 

  122. 	typedef struct  

    123. 

  123. 	{

    124. 

  124. 		LPCSTR testInput;

    125. 

  125. 		LPCSTR testWhitespaces;

    126. 

  126. 		LPCSTR testExpectedResult;

    127. 

  127. 	} TestData;

    128. 

  128. 

    129. 

  129. 	TestData testData[] = 

    130. 

  130. 	{

    131. 

  131. 		{ "     this is a string whith whitespaces\t     ", "", "this is a string whith whitespaces\t     " },

    132. 

  132. 		{ "     this is a string whith whitespaces     ", " ", "this is a string whith whitespaces     " },

    133. 

  133. 		{ "this string doesn't have whitespaces", "", "this string doesn't have whitespaces" },

    134. 

  134. 		{ "%%test string%%", "%", "test string%%" },

    135. 

  135. 		{ " test string     ", "", "test string     " }

    136. 

  136. 	};

    137. 

  137. 

    138. 

  138. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    139. 

  139. 	{		

    140. 

  140. 		msiInstall.SetProperty("STRING_LTRIM_INPUT", testData[i].testInput);

    141. 

  141. 		msiInstall.SetProperty("STRING_LTRIM_WHITESPACES", testData[i].testWhitespaces);

    142. 

  142. 

    143. 

  143. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_LTrim"));

    144. 

  144. 

    145. 

  145. 		std::string stringLTrimResult = msiInstall.GetProperty("STRING_LTRIM_RESULT");

    146. 

  146. 		std::stringstream message;

    147. 

  147. 

    148. 

  148. 		message << "StringLTrim('" << testData[i].testInput <<

    149. 

  149. 			"', '" << testData[i].testWhitespaces << "') returned '" << stringLTrimResult <<

    150. 

  150. 			"'. It was expected: '" << testData[i].testExpectedResult << "'";

    151. 

  151. 

    152. 

  152. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(), testData[i].testExpectedResult == stringLTrimResult);

    153. 

  153. 	}

    154. 

  154. }

    155. 

  155. 

    156. 

  156. void StringToolsImplUnitTests::Test_StringRTrim()

    157. 

  157. {

    158. 

  158. 	AppSecInc::Msi::MsiShim hInstall;

    159. 

  159.     MsiInstall msiInstall(hInstall);

    160. 

  160. 

    161. 

  161. 	typedef struct  

    162. 

  162. 	{

    163. 

  163. 		LPCSTR testInput;

    164. 

  164. 		LPCSTR testWhitespaces;

    165. 

  165. 		LPCSTR testExpectedResult;

    166. 

  166. 	} TestData;

    167. 

  167. 

    168. 

  168. 	TestData testData[] = 

    169. 

  169. 	{

    170. 

  170. 		{ "     this is a string whith whitespaces\t     ", "", "     this is a string whith whitespaces" },

    171. 

  171. 		{ "     this is a string whith whitespaces     ", " ", "     this is a string whith whitespaces" },

    172. 

  172. 		{ "this string doesn't have whitespaces", "", "this string doesn't have whitespaces" },

    173. 

  173. 		{ "%%test string%%", "%", "%%test string" },

    174. 

  174. 		{ " test string     ", "", " test string" }

    175. 

  175. 	};

    176. 

  176. 

    177. 

  177. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    178. 

  178. 	{		

    179. 

  179. 		msiInstall.SetProperty("STRING_RTRIM_INPUT", testData[i].testInput);

    180. 

  180. 		msiInstall.SetProperty("STRING_RTRIM_WHITESPACES", testData[i].testWhitespaces);

    181. 

  181. 

    182. 

  182. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_RTrim"));

    183. 

  183. 

    184. 

  184. 		std::string stringRTrimResult = msiInstall.GetProperty("STRING_RTRIM_RESULT");

    185. 

  185. 		std::stringstream message;

    186. 

  186. 

    187. 

  187. 		message << "StringRTrim('" << testData[i].testInput <<

    188. 

  188. 			"', '" << testData[i].testWhitespaces << "') returned '" << stringRTrimResult <<

    189. 

  189. 			"'. It was expected: '" << testData[i].testExpectedResult << "'";

    190. 

  190. 

    191. 

  191. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(), testData[i].testExpectedResult == stringRTrimResult);

    192. 

  192. 	}

    193. 

  193. }

    194. 

  194. 

    195. 

  195. void StringToolsImplUnitTests::Test_RegexMatch()

    196. 

  196. {

    197. 

  197. 	AppSecInc::Msi::MsiShim hInstall;

    198. 

  198.     MsiInstall msiInstall(hInstall);

    199. 

  199. 

    200. 

  200. 	typedef struct  

    201. 

  201. 	{

    202. 

  202. 		LPCSTR testInputString;

    203. 

  203. 		LPCSTR testExpression;

    204. 

  204. 		LPCSTR testExpectedResult;

    205. 

  205. 	} TestData;

    206. 

  206. 

    207. 

  207. 	TestData testData[] = 

    208. 

  208. 	{

    209. 

  209. 		{ "<TAG>one<TAG>two</TAG>one</TAG>", "<TAG\\b[^>]*>(.*?)</TAG>", "1" },

    210. 

  210. 		{ "<TAG>one<TAG>two</TAG>one", "<TAG\\b[^>]*>(.*?)</TAG>", "0" },

    211. 

  211. 		{ "125", "^([1-9]|[1-9][0-9]|[1-9][0-9][0-9])$", "1" },

    212. 

  212. 		{ "1005", "^([1-9]|[1-9][0-9]|[1-9][0-9][0-9])$", "0" },

    213. 

  213. 		{ "2008-07-31", "(19|20)\\d\\d[- /.](0[1-9]|1[012])[- /.](0[1-9]|[12][0-9]|3[01])", "1" },

    214. 

  214. 		{ "2008-15-20", "(19|20)\\d\\d[- /.](0[1-9]|1[012])[- /.](0[1-9]|[12][0-9]|3[01])", "0" },

    215. 

  215. 		{ "myname@host.com", "\\b[a-z0-9._%+-]+@[a-z0-9.-]+\\.[a-z]{2,4}\\b", "1" },

    216. 

  216. 		{ "invalid email@host.com", "\\b[a-z0-9._%+-]+@[a-z0-9.-]+\\.[a-z]{2,4}\\b", "0" }

    217. 

  217. 	};

    218. 

  218. 

    219. 

  219. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    220. 

  220. 	{		

    221. 

  221. 		msiInstall.SetProperty("REGEX_MATCH_INPUT_STRING", testData[i].testInputString);

    222. 

  222. 		msiInstall.SetProperty("REGEX_MATCH_EXPRESSION", testData[i].testExpression);

    223. 

  223. 

    224. 

  224. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"Regex_Match"));

    225. 

  225. 

    226. 

  226. 		std::string regexMatchResult = msiInstall.GetProperty("REGEX_MATCH_RESULT");

    227. 

  227. 		std::stringstream message;

    228. 

  228. 

    229. 

  229. 		message << "RegexMatch('" << testData[i].testInputString <<

    230. 

  230. 			"', '" << testData[i].testExpression << "') returned '" << regexMatchResult <<

    231. 

  231. 			"'. It was expected: '" << testData[i].testExpectedResult << "'";

    232. 

  232. 

    233. 

  233. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(), testData[i].testExpectedResult == regexMatchResult);

    234. 

  234. 	}

    235. 

  235. }

    236. 

  236. 

    237. 

  237. void StringToolsImplUnitTests::Test_RegexReplace()

    238. 

  238. {

    239. 

  239. 	AppSecInc::Msi::MsiShim hInstall;

    240. 

  240.     MsiInstall msiInstall(hInstall);

    241. 

  241. 

    242. 

  242. 	typedef struct  

    243. 

  243. 	{

    244. 

  244. 		LPCSTR testInputString;

    245. 

  245. 		LPCSTR testExpression;

    246. 

  246. 		LPCSTR testFormat;

    247. 

  247. 		LPCSTR testExpectedResult;

    248. 

  248. 	} TestData;

    249. 

  249. 

    250. 

  250. 	TestData testData[] = 

    251. 

  251. 	{

    252. 

  252. 		{ "I have a dog. Also a machine.", "\\b(cat|dog)\\b", "pet", "I have a pet. Also a machine." },

    253. 

  253. 		{ "I have a spider. Also a machine.", "\\b(cat|dog)\\b", "pet", "I have a spider. Also a machine." },

    254. 

  254. 		{ "user@domain", "([^@]*)@(.*)", "$2\\\\$1", "domain\\user" }

    255. 

  255. 	};

    256. 

  256. 

    257. 

  257. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    258. 

  258. 	{		

    259. 

  259. 		msiInstall.SetProperty("REGEX_REPLACE_INPUT_STRING", testData[i].testInputString);

    260. 

  260. 		msiInstall.SetProperty("REGEX_REPLACE_EXPRESSION", testData[i].testExpression);

    261. 

  261. 		msiInstall.SetProperty("REGEX_REPLACE_FORMAT", testData[i].testFormat);

    262. 

  262. 

    263. 

  263. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"Regex_Replace"));

    264. 

  264. 

    265. 

  265. 		std::string regexReplaceResult = msiInstall.GetProperty("REGEX_REPLACE_RESULT");

    266. 

  266. 		std::stringstream message;

    267. 

  267. 

    268. 

  268. 		message << "RegexReplace('" << testData[i].testInputString <<

    269. 

  269. 			"', '" << testData[i].testExpression <<

    270. 

  270. 			"', '" << testData[i].testFormat <<

    271. 

  271. 			"') returned: '" << regexReplaceResult <<

    272. 

  272. 			"'. It was expected: " << testData[i].testExpectedResult << "'";

    273. 

  273. 

    274. 

  274. 		CPPUNIT_ASSERT_MESSAGE(message.str().c_str(), testData[i].testExpectedResult == regexReplaceResult);

    275. 

  275. 	}

    276. 

  276. }

    277. 

  277. 

    278. 

  278. 

    279. 

  279. void StringToolsImplUnitTests::Test_StringToLower()

    280. 

  280. {

    281. 

  281. 	AppSecInc::Msi::MsiShim hInstall;

    282. 

  282.     MsiInstall msiInstall(hInstall);

    283. 

  283.     

    284. 

  284. 	typedef struct  

    285. 

  285. 	{

    286. 

  286. 		LPCTSTR input;

    287. 

  287. 		LPCTSTR expected;

    288. 

  288. 	} TestData;

    289. 

  289. 

    290. 

  290. 	TestData testData[] = 

    291. 

  291. 	{

    292. 

  292. 		{ L"ABCdeF123", L"abcdef123" },

    293. 

  293. 		//{ L"??????×??", L"??????÷??" }, // not supported by AppSecInc::StringUtils::lowercase

    294. 

  294. 		{ L"", L"" }

    295. 

  295. 	};

    296. 

  296. 

    297. 

  297. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    298. 

  298. 	{		

    299. 

  299. 		msiInstall.SetProperty(L"STRING_TOLOWER_INPUT", testData[i].input);

    300. 

  300. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_ToLower"));

    301. 

  301. 

    302. 

  302. 		std::wstring result = msiInstall.GetProperty(L"STRING_TOLOWER_RESULT");

    303. 

  303. 		std::wstring expected( testData[i].expected );

    304. 

  304. //		std::wstring expected( testData[i].input );

    305. 

  305. //	    AppSecInc::StringUtils::lowercase( expected );

    306. 

  306. 		

    307. 

  307. 		CPPUNIT_ASSERT(expected == result);

    308. 

  308. 	}

    309. 

  309. }

    310. 

  310. 

    311. 

  311. void StringToolsImplUnitTests::Test_StringToUpper()

    312. 

  312. {

    313. 

  313. 	AppSecInc::Msi::MsiShim hInstall;

    314. 

  314.     MsiInstall msiInstall(hInstall);

    315. 

  315.     

    316. 

  316. 	typedef struct  

    317. 

  317. 	{

    318. 

  318. 		LPCTSTR input;

    319. 

  319. 		LPCTSTR expected;

    320. 

  320. 	} TestData;

    321. 

  321. 

    322. 

  322. 	TestData testData[] = 

    323. 

  323. 	{

    324. 

  324. 		{ L"ABCdeF123", L"ABCDEF123" },

    325. 

  325. //		{ L"??????÷??", L"??????×??" }, // not supported by AppSecInc::StringUtils::uppercase

    326. 

  326. 		{ L"", L"" }

    327. 

  327. 	};

    328. 

  328. 

    329. 

  329. 	for( unsigned int i = 0; i < ARRAYSIZE(testData); i++ )

    330. 

  330. 	{		

    331. 

  331. 		msiInstall.SetProperty(L"STRING_TOUPPER_INPUT", testData[i].input);

    332. 

  332. 		CPPUNIT_ASSERT(ERROR_SUCCESS == hInstall.ExecuteCA(L"StringTools.dll", L"String_ToUpper"));

    333. 

  333. 

    334. 

  334. 		std::wstring result = msiInstall.GetProperty(L"STRING_TOUPPER_RESULT");

    335. 

  335. 		std::wstring expected( testData[i].expected );

    336. 

  336. //		std::wstring expected( testData[i].input );

    337. 

  337. //	    AppSecInc::StringUtils::uppercase( expected );

    338. 

  338. 

    339. 

  339. 		CPPUNIT_ASSERT(expected == result);

    340. 

  340. 	}

    341. 

  341. }

    342. 

  342.