1 /* 2 * Copyright (C) 2005 Mike McCormack for CodeWeavers 3 * 4 * A test program for MSI records 5 * 6 * This library is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * This library is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with this library; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA 19 */ 20 21 #include <windows.h> 22 #include <msi.h> 23 #include <msiquery.h> 24 25 #include "wine/test.h" 26 27 static const char *msifile = "winetest-record.msi"; 28 static const WCHAR msifileW[] = 29 {'w','i','n','e','t','e','s','t','-','r','e','c','o','r','d','.','m','s','i',0}; 30 31 static BOOL create_temp_file(char *name) 32 { 33 UINT r; 34 unsigned char buffer[26], i; 35 DWORD sz; 36 HANDLE handle; 37 38 r = GetTempFileNameA(".", "msitest",0,name); 39 if(!r) 40 return r; 41 handle = CreateFileA(name, GENERIC_READ|GENERIC_WRITE, 42 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); 43 if(handle==INVALID_HANDLE_VALUE) 44 return FALSE; 45 for(i=0; i<26; i++) 46 buffer[i]=i+'a'; 47 r = WriteFile(handle,buffer,sizeof buffer,&sz,NULL); 48 CloseHandle(handle); 49 return r; 50 } 51 52 static void test_msirecord(void) 53 { 54 DWORD r, sz; 55 INT i; 56 MSIHANDLE h; 57 char buf[10]; 58 WCHAR bufW[10]; 59 const char str[] = "hello"; 60 const WCHAR strW[] = { 'h','e','l','l','o',0}; 61 char filename[MAX_PATH]; 62 63 /* check behaviour with an invalid record */ 64 r = MsiRecordGetFieldCount(0); 65 ok(r==-1, "field count for invalid record not -1\n"); 66 SetLastError(0); 67 r = MsiRecordIsNull(0, 0); 68 ok(r==0, "invalid handle not considered to be non-null...\n"); 69 ok(GetLastError()==0, "MsiRecordIsNull set LastError\n"); 70 r = MsiRecordGetInteger(0,0); 71 ok(r == MSI_NULL_INTEGER, "got integer from invalid record\n"); 72 r = MsiRecordSetInteger(0,0,0); 73 ok(r == ERROR_INVALID_HANDLE, "MsiRecordSetInteger returned wrong error\n"); 74 r = MsiRecordSetInteger(0,-1,0); 75 ok(r == ERROR_INVALID_HANDLE, "MsiRecordSetInteger returned wrong error\n"); 76 SetLastError(0); 77 h = MsiCreateRecord(-1); 78 ok(h==0, "created record with -1 elements\n"); 79 h = MsiCreateRecord(0x10000); 80 ok(h==0, "created record with 0x10000 elements\n"); 81 /* doesn't set LastError */ 82 ok(GetLastError()==0, "MsiCreateRecord set last error\n"); 83 r = MsiRecordClearData(0); 84 ok(r == ERROR_INVALID_HANDLE, "MsiRecordClearData returned wrong error\n"); 85 r = MsiRecordDataSize(0,0); 86 ok(r == 0, "MsiRecordDataSize returned wrong error\n"); 87 88 89 /* check behaviour of a record with 0 elements */ 90 h = MsiCreateRecord(0); 91 ok(h!=0, "couldn't create record with zero elements\n"); 92 r = MsiRecordGetFieldCount(h); 93 ok(r==0, "field count should be zero\n"); 94 r = MsiRecordIsNull(h,0); 95 ok(r, "new record wasn't null\n"); 96 r = MsiRecordIsNull(h,1); 97 ok(r, "out of range record wasn't null\n"); 98 r = MsiRecordIsNull(h,-1); 99 ok(r, "out of range record wasn't null\n"); 100 r = MsiRecordDataSize(h,0); 101 ok(r==0, "size of null record is 0\n"); 102 sz = sizeof buf; 103 strcpy(buf,"x"); 104 r = MsiRecordGetStringA(h, 0, buf, &sz); 105 ok(r==ERROR_SUCCESS, "failed to get null string\n"); 106 ok(sz==0, "null string too long\n"); 107 ok(buf[0]==0, "null string not set\n"); 108 109 /* same record, but add an integer to it */ 110 r = MsiRecordSetInteger(h, 0, 0); 111 ok(r == ERROR_SUCCESS, "Failed to set integer at 0 to 0\n"); 112 r = MsiRecordIsNull(h,0); 113 ok(r==0, "new record is null after setting an integer\n"); 114 r = MsiRecordDataSize(h,0); 115 ok(r==sizeof(DWORD), "size of integer record is 4\n"); 116 r = MsiRecordSetInteger(h, 0, 1); 117 ok(r == ERROR_SUCCESS, "Failed to set integer at 0 to 1\n"); 118 r = MsiRecordSetInteger(h, 1, 1); 119 ok(r == ERROR_INVALID_PARAMETER, "set integer at 1\n"); 120 r = MsiRecordSetInteger(h, -1, 0); 121 ok(r == ERROR_INVALID_PARAMETER, "set integer at -1\n"); 122 r = MsiRecordIsNull(h,0); 123 ok(r==0, "new record is null after setting an integer\n"); 124 r = MsiRecordGetInteger(h, 0); 125 ok(r == 1, "failed to get integer\n"); 126 127 /* same record, but add a null or empty string to it */ 128 r = MsiRecordSetStringA(h, 0, NULL); 129 ok(r == ERROR_SUCCESS, "Failed to set null string at 0\n"); 130 r = MsiRecordIsNull(h, 0); 131 ok(r == TRUE, "null string not null field\n"); 132 r = MsiRecordDataSize(h, 0); 133 ok(r == 0, "size of string record is strlen\n"); 134 buf[0] = 0; 135 sz = sizeof buf; 136 r = MsiRecordGetStringA(h, 0, buf, &sz); 137 ok(r == ERROR_SUCCESS, "Failed to get string at 0\n"); 138 ok(buf[0] == 0, "MsiRecordGetStringA returned the wrong string\n"); 139 ok(sz == 0, "MsiRecordGetStringA returned the wrong length\n"); 140 bufW[0] = 0; 141 sz = sizeof bufW / sizeof bufW[0]; 142 r = MsiRecordGetStringW(h, 0, bufW, &sz); 143 ok(r == ERROR_SUCCESS, "Failed to get string at 0\n"); 144 ok(bufW[0] == 0, "MsiRecordGetStringW returned the wrong string\n"); 145 ok(sz == 0, "MsiRecordGetStringW returned the wrong length\n"); 146 r = MsiRecordSetStringA(h, 0, ""); 147 ok(r == ERROR_SUCCESS, "Failed to set empty string at 0\n"); 148 r = MsiRecordIsNull(h, 0); 149 ok(r == TRUE, "null string not null field\n"); 150 r = MsiRecordDataSize(h, 0); 151 ok(r == 0, "size of string record is strlen\n"); 152 buf[0] = 0; 153 sz = sizeof buf; 154 r = MsiRecordGetStringA(h, 0, buf, &sz); 155 ok(r == ERROR_SUCCESS, "Failed to get string at 0\n"); 156 ok(buf[0] == 0, "MsiRecordGetStringA returned the wrong string\n"); 157 ok(sz == 0, "MsiRecordGetStringA returned the wrong length\n"); 158 bufW[0] = 0; 159 sz = sizeof bufW / sizeof bufW[0]; 160 r = MsiRecordGetStringW(h, 0, bufW, &sz); 161 ok(r == ERROR_SUCCESS, "Failed to get string at 0\n"); 162 ok(bufW[0] == 0, "MsiRecordGetStringW returned the wrong string\n"); 163 ok(sz == 0, "MsiRecordGetStringW returned the wrong length\n"); 164 165 /* same record, but add a string to it */ 166 r = MsiRecordSetStringA(h,0,str); 167 ok(r == ERROR_SUCCESS, "Failed to set string at 0\n"); 168 r = MsiRecordGetInteger(h, 0); 169 ok(r == MSI_NULL_INTEGER, "should get invalid integer\n"); 170 r = MsiRecordDataSize(h,0); 171 ok(r==sizeof str-1, "size of string record is strlen\n"); 172 buf[0]=0; 173 sz = sizeof buf; 174 r = MsiRecordGetStringA(h,0,buf,&sz); 175 ok(r == ERROR_SUCCESS, "Failed to get string at 0\n"); 176 ok(0==strcmp(buf,str), "MsiRecordGetStringA returned the wrong string\n"); 177 ok(sz == sizeof str-1, "MsiRecordGetStringA returned the wrong length\n"); 178 buf[0]=0; 179 sz = sizeof str - 2; 180 r = MsiRecordGetStringA(h,0,buf,&sz); 181 ok(r == ERROR_MORE_DATA, "small buffer should yield ERROR_MORE_DATA\n"); 182 ok(sz == sizeof str-1, "MsiRecordGetStringA returned the wrong length\n"); 183 ok(0==strncmp(buf,str,sizeof str-3), "MsiRecordGetStringA returned the wrong string\n"); 184 ok(buf[sizeof str - 3]==0, "string wasn't nul terminated\n"); 185 186 buf[0]=0; 187 sz = sizeof str; 188 r = MsiRecordGetStringA(h,0,buf,&sz); 189 ok(r == ERROR_SUCCESS, "wrong error\n"); 190 ok(sz == sizeof str-1, "MsiRecordGetStringA returned the wrong length\n"); 191 ok(0==strcmp(buf,str), "MsiRecordGetStringA returned the wrong string\n"); 192 193 194 memset(bufW, 0, sizeof bufW); 195 sz = 5; 196 r = MsiRecordGetStringW(h,0,bufW,&sz); 197 ok(r == ERROR_MORE_DATA, "wrong error\n"); 198 ok(sz == 5, "MsiRecordGetStringA returned the wrong length\n"); 199 ok(0==memcmp(bufW,strW,8), "MsiRecordGetStringA returned the wrong string\n"); 200 201 sz = 0; 202 bufW[0] = 'x'; 203 r = MsiRecordGetStringW(h,0,bufW,&sz); 204 ok(r == ERROR_MORE_DATA, "wrong error\n"); 205 ok(sz == 5, "MsiRecordGetStringA returned the wrong length\n"); 206 ok('x'==bufW[0], "MsiRecordGetStringA returned the wrong string\n"); 207 208 memset(buf, 0, sizeof buf); 209 sz = 5; 210 r = MsiRecordGetStringA(h,0,buf,&sz); 211 ok(r == ERROR_MORE_DATA, "wrong error\n"); 212 ok(sz == 5, "MsiRecordGetStringA returned the wrong length\n"); 213 ok(0==memcmp(buf,str,4), "MsiRecordGetStringA returned the wrong string\n"); 214 215 sz = 0; 216 buf[0] = 'x'; 217 r = MsiRecordGetStringA(h,0,buf,&sz); 218 ok(r == ERROR_MORE_DATA, "wrong error\n"); 219 ok(sz == 5, "MsiRecordGetStringA returned the wrong length\n"); 220 ok('x'==buf[0], "MsiRecordGetStringA returned the wrong string\n"); 221 222 /* same record, check we can wipe all the data */ 223 r = MsiRecordClearData(h); 224 ok(r == ERROR_SUCCESS, "Failed to clear record\n"); 225 r = MsiRecordClearData(h); 226 ok(r == ERROR_SUCCESS, "Failed to clear record again\n"); 227 r = MsiRecordIsNull(h,0); 228 ok(r, "cleared record wasn't null\n"); 229 230 /* same record, try converting strings to integers */ 231 i = MsiRecordSetStringA(h,0,"42"); 232 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 233 i = MsiRecordGetInteger(h, 0); 234 ok(i == 42, "should get invalid integer\n"); 235 i = MsiRecordSetStringA(h,0,"-42"); 236 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 237 i = MsiRecordGetInteger(h, 0); 238 ok(i == -42, "should get invalid integer\n"); 239 i = MsiRecordSetStringA(h,0," 42"); 240 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 241 i = MsiRecordGetInteger(h, 0); 242 ok(i == MSI_NULL_INTEGER, "should get invalid integer\n"); 243 i = MsiRecordSetStringA(h,0,"42 "); 244 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 245 i = MsiRecordGetInteger(h, 0); 246 ok(i == MSI_NULL_INTEGER, "should get invalid integer\n"); 247 i = MsiRecordSetStringA(h,0,"42.0"); 248 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 249 i = MsiRecordGetInteger(h, 0); 250 ok(i == MSI_NULL_INTEGER, "should get invalid integer\n"); 251 i = MsiRecordSetStringA(h,0,"0x42"); 252 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 253 i = MsiRecordGetInteger(h, 0); 254 ok(i == MSI_NULL_INTEGER, "should get invalid integer\n"); 255 i = MsiRecordSetStringA(h,0,"1000000000000000"); 256 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 257 i = MsiRecordGetInteger(h, 0); 258 ok(i == -1530494976, "should get truncated integer\n"); 259 i = MsiRecordSetStringA(h,0,"2147483647"); 260 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 261 i = MsiRecordGetInteger(h, 0); 262 ok(i == 2147483647, "should get maxint\n"); 263 i = MsiRecordSetStringA(h,0,"-2147483647"); 264 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 265 i = MsiRecordGetInteger(h, 0); 266 ok(i == -2147483647, "should get -maxint-1\n"); 267 i = MsiRecordSetStringA(h,0,"4294967297"); 268 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 269 i = MsiRecordGetInteger(h, 0); 270 ok(i == 1, "should get one\n"); 271 i = MsiRecordSetStringA(h,0,"foo"); 272 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 273 i = MsiRecordGetInteger(h, 0); 274 ok(i == MSI_NULL_INTEGER, "should get zero\n"); 275 i = MsiRecordSetStringA(h,0,""); 276 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 277 i = MsiRecordGetInteger(h, 0); 278 ok(i == MSI_NULL_INTEGER, "should get zero\n"); 279 i = MsiRecordSetStringA(h,0,"+1"); 280 ok(i == ERROR_SUCCESS, "Failed to set string at 0\n"); 281 i = MsiRecordGetInteger(h, 0); 282 ok(i == MSI_NULL_INTEGER, "should get zero\n"); 283 284 /* same record, try converting integers to strings */ 285 r = MsiRecordSetInteger(h, 0, 32); 286 ok(r == ERROR_SUCCESS, "Failed to set integer at 0 to 32\n"); 287 sz = 1; 288 r = MsiRecordGetStringA(h, 0, NULL, &sz); 289 ok(r == ERROR_SUCCESS, "failed to get string from integer\n"); 290 ok(sz == 2, "length wrong\n"); 291 buf[0]=0; 292 sz = sizeof buf; 293 r = MsiRecordGetStringA(h, 0, buf, &sz); 294 ok(r == ERROR_SUCCESS, "failed to get string from integer\n"); 295 ok(0==strcmp(buf,"32"), "failed to get string from integer\n"); 296 r = MsiRecordSetInteger(h, 0, -32); 297 ok(r == ERROR_SUCCESS, "Failed to set integer at 0 to 32\n"); 298 buf[0]=0; 299 sz = 1; 300 r = MsiRecordGetStringA(h, 0, NULL, &sz); 301 ok(r == ERROR_SUCCESS, "failed to get string from integer\n"); 302 ok(sz == 3, "length wrong\n"); 303 sz = sizeof buf; 304 r = MsiRecordGetStringA(h, 0, buf, &sz); 305 ok(r == ERROR_SUCCESS, "failed to get string from integer\n"); 306 ok(0==strcmp(buf,"-32"), "failed to get string from integer\n"); 307 buf[0]=0; 308 309 /* same record, now try streams */ 310 r = MsiRecordSetStreamA(h, 0, NULL); 311 ok(r == ERROR_INVALID_PARAMETER, "set NULL stream\n"); 312 sz = sizeof buf; 313 r = MsiRecordReadStream(h, 0, buf, &sz); 314 ok(r == ERROR_INVALID_DATATYPE, "read non-stream type\n"); 315 ok(sz == sizeof buf, "set sz\n"); 316 r = MsiRecordDataSize( h, -1); 317 ok(r == 0,"MsiRecordDataSize returned wrong size\n"); 318 r = MsiRecordDataSize( h, 0); 319 ok(r == 4,"MsiRecordDataSize returned wrong size\n"); 320 321 /* same record, now close it */ 322 r = MsiCloseHandle(h); 323 ok(r == ERROR_SUCCESS, "Failed to close handle\n"); 324 325 /* now try streams in a new record - need to create a file to play with */ 326 r = create_temp_file(filename); 327 if(!r) 328 return; 329 330 /* streams can't be inserted in field 0 for some reason */ 331 h = MsiCreateRecord(2); 332 ok(h, "couldn't create a two field record\n"); 333 r = MsiRecordSetStreamA(h, 0, filename); 334 ok(r == ERROR_INVALID_PARAMETER, "added stream to field 0\n"); 335 r = MsiRecordSetStreamA(h, 1, filename); 336 ok(r == ERROR_SUCCESS, "failed to add stream to record\n"); 337 r = MsiRecordReadStream(h, 1, buf, NULL); 338 ok(r == ERROR_INVALID_PARAMETER, "should return error\n"); 339 DeleteFileA(filename); /* Windows 98 doesn't like this at all, so don't check return. */ 340 r = MsiRecordReadStream(h, 1, NULL, NULL); 341 ok(r == ERROR_INVALID_PARAMETER, "should return error\n"); 342 sz = sizeof buf; 343 r = MsiRecordReadStream(h, 1, NULL, &sz); 344 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 345 ok(sz==26,"couldn't get size of stream\n"); 346 sz = 0; 347 r = MsiRecordReadStream(h, 1, buf, &sz); 348 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 349 ok(sz==0,"short read\n"); 350 sz = sizeof buf; 351 r = MsiRecordReadStream(h, 1, buf, &sz); 352 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 353 ok(sz==sizeof buf,"short read\n"); 354 ok(!strncmp(buf,"abcdefghij",10), "read the wrong thing\n"); 355 sz = sizeof buf; 356 r = MsiRecordReadStream(h, 1, buf, &sz); 357 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 358 ok(sz==sizeof buf,"short read\n"); 359 ok(!strncmp(buf,"klmnopqrst",10), "read the wrong thing\n"); 360 memset(buf,0,sizeof buf); 361 sz = sizeof buf; 362 r = MsiRecordReadStream(h, 1, buf, &sz); 363 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 364 ok(sz==6,"short read\n"); 365 ok(!strcmp(buf,"uvwxyz"), "read the wrong thing\n"); 366 memset(buf,0,sizeof buf); 367 sz = sizeof buf; 368 r = MsiRecordReadStream(h, 1, buf, &sz); 369 ok(r == ERROR_SUCCESS, "failed to read stream\n"); 370 ok(sz==0,"size non-zero at end of stream\n"); 371 ok(buf[0]==0, "read something at end of the stream\n"); 372 r = MsiRecordSetStreamA(h, 1, NULL); 373 ok(r == ERROR_SUCCESS, "failed to reset stream\n"); 374 sz = 0; 375 r = MsiRecordReadStream(h, 1, NULL, &sz); 376 ok(r == ERROR_SUCCESS, "bytes left wrong after reset\n"); 377 ok(sz==26,"couldn't get size of stream\n"); 378 r = MsiRecordDataSize(h,1); 379 ok(r == 26,"MsiRecordDataSize returned wrong size\n"); 380 381 /* now close the stream record */ 382 r = MsiCloseHandle(h); 383 ok(r == ERROR_SUCCESS, "Failed to close handle\n"); 384 DeleteFileA(filename); /* Delete it for sure, when everything else is closed. */ 385 } 386 387 static void test_MsiRecordGetString(void) 388 { 389 MSIHANDLE rec; 390 CHAR buf[MAX_PATH]; 391 DWORD sz; 392 UINT r; 393 394 rec = MsiCreateRecord(2); 395 ok(rec != 0, "Expected a valid handle\n"); 396 397 sz = MAX_PATH; 398 r = MsiRecordGetStringA(rec, 1, NULL, &sz); 399 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n",r); 400 ok(sz == 0, "Expected 0, got %d\n",sz); 401 402 sz = MAX_PATH; 403 lstrcpyA(buf, "apple"); 404 r = MsiRecordGetStringA(rec, 1, buf, &sz); 405 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 406 ok(!lstrcmpA(buf, ""), "Expected \"\", got \"%s\"\n", buf); 407 ok(sz == 0, "Expected 0, got %d\n", sz); 408 409 sz = MAX_PATH; 410 lstrcpyA(buf, "apple"); 411 r = MsiRecordGetStringA(rec, 10, buf, &sz); 412 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 413 ok(!lstrcmpA(buf, ""), "Expected \"\", got \"%s\"\n", buf); 414 ok(sz == 0, "Expected 0, got %d\n", sz); 415 416 MsiCloseHandle(rec); 417 418 rec = MsiCreateRecord(1); 419 ok(rec != 0, "Expected a valid handle\n"); 420 421 r = MsiRecordSetInteger(rec, 1, 5); 422 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 423 424 sz = MAX_PATH; 425 r = MsiRecordGetStringA(rec, 1, NULL, &sz); 426 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n",r); 427 ok(sz == 1, "Expected 1, got %d\n",sz); 428 429 sz = MAX_PATH; 430 lstrcpyA(buf, "apple"); 431 r = MsiRecordGetStringA(rec, 1, buf, &sz); 432 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 433 ok(!lstrcmpA(buf, "5"), "Expected \"5\", got \"%s\"\n", buf); 434 ok(sz == 1, "Expectd 1, got %d\n", sz); 435 436 r = MsiRecordSetInteger(rec, 1, -5); 437 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 438 439 sz = MAX_PATH; 440 lstrcpyA(buf, "apple"); 441 r = MsiRecordGetStringA(rec, 1, buf, &sz); 442 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 443 ok(!lstrcmpA(buf, "-5"), "Expected \"-5\", got \"%s\"\n", buf); 444 ok(sz == 2, "Expectd 2, got %d\n", sz); 445 446 MsiCloseHandle(rec); 447 } 448 449 static void test_MsiRecordGetInteger(void) 450 { 451 MSIHANDLE rec; 452 INT val; 453 UINT r; 454 455 rec = MsiCreateRecord(1); 456 ok(rec != 0, "Expected a valid handle\n"); 457 458 r = MsiRecordSetStringA(rec, 1, "5"); 459 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 460 461 val = MsiRecordGetInteger(rec, 1); 462 ok(val == 5, "Expected 5, got %d\n", val); 463 464 r = MsiRecordSetStringA(rec, 1, "-5"); 465 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 466 467 val = MsiRecordGetInteger(rec, 1); 468 ok(val == -5, "Expected -5, got %d\n", val); 469 470 r = MsiRecordSetStringA(rec, 1, "5apple"); 471 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 472 473 val = MsiRecordGetInteger(rec, 1); 474 ok(val == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", val); 475 476 MsiCloseHandle(rec); 477 } 478 479 static void test_fieldzero(void) 480 { 481 MSIHANDLE hdb, hview, rec; 482 CHAR buf[MAX_PATH]; 483 LPCSTR query; 484 DWORD sz; 485 UINT r; 486 487 rec = MsiCreateRecord(1); 488 ok(rec != 0, "Expected a valid handle\n"); 489 490 r = MsiRecordGetInteger(rec, 0); 491 ok(r == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", r); 492 493 sz = MAX_PATH; 494 lstrcpyA(buf, "apple"); 495 r = MsiRecordGetStringA(rec, 0, buf, &sz); 496 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 497 ok(!lstrcmpA(buf, ""), "Expected \"\", got \"%s\"\n", buf); 498 ok(sz == 0, "Expectd 0, got %d\n", sz); 499 500 r = MsiRecordIsNull(rec, 0); 501 ok(r == TRUE, "Expected TRUE, got %d\n", r); 502 503 r = MsiRecordGetInteger(rec, 1); 504 ok(r == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", r); 505 506 r = MsiRecordSetInteger(rec, 1, 42); 507 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 508 509 r = MsiRecordGetInteger(rec, 0); 510 ok(r == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", r); 511 512 sz = MAX_PATH; 513 lstrcpyA(buf, "apple"); 514 r = MsiRecordGetStringA(rec, 0, buf, &sz); 515 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 516 ok(!lstrcmpA(buf, ""), "Expected \"\", got \"%s\"\n", buf); 517 ok(sz == 0, "Expectd 0, got %d\n", sz); 518 519 r = MsiRecordIsNull(rec, 0); 520 ok(r == TRUE, "Expected TRUE, got %d\n", r); 521 522 r = MsiRecordGetInteger(rec, 1); 523 ok(r == 42, "Expected 42, got %d\n", r); 524 525 r = MsiRecordSetStringA(rec, 1, "bologna"); 526 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 527 528 r = MsiRecordGetInteger(rec, 0); 529 ok(r == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", r); 530 531 sz = MAX_PATH; 532 lstrcpyA(buf, "apple"); 533 r = MsiRecordGetStringA(rec, 0, buf, &sz); 534 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 535 ok(!lstrcmpA(buf, ""), "Expected \"\", got \"%s\"\n", buf); 536 ok(sz == 0, "Expectd 0, got %d\n", sz); 537 538 r = MsiRecordIsNull(rec, 0); 539 ok(r == TRUE, "Expected TRUE, got %d\n", r); 540 541 sz = MAX_PATH; 542 lstrcpyA(buf, "apple"); 543 r = MsiRecordGetStringA(rec, 1, buf, &sz); 544 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 545 ok(!lstrcmpA(buf, "bologna"), "Expected \"bologna\", got \"%s\"\n", buf); 546 ok(sz == 7, "Expectd 7, got %d\n", sz); 547 548 MsiCloseHandle(rec); 549 550 r = MsiOpenDatabaseW(msifileW, MSIDBOPEN_CREATE, &hdb); 551 ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n"); 552 553 query = "CREATE TABLE `drone` ( " 554 "`id` INT, `name` CHAR(32), `number` CHAR(32) " 555 "PRIMARY KEY `id`)"; 556 r = MsiDatabaseOpenViewA(hdb, query, &hview); 557 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 558 r = MsiViewExecute(hview, 0); 559 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 560 r = MsiViewClose(hview); 561 ok(r == ERROR_SUCCESS, "MsiViewClose failed\n"); 562 r = MsiCloseHandle(hview); 563 ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n"); 564 565 query = "INSERT INTO `drone` ( `id`, `name`, `number` )" 566 "VALUES('1', 'Abe', '8675309')"; 567 r = MsiDatabaseOpenViewA(hdb, query, &hview); 568 ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n"); 569 r = MsiViewExecute(hview, 0); 570 ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n"); 571 r = MsiViewClose(hview); 572 ok(r == ERROR_SUCCESS, "MsiViewClose failed\n"); 573 r = MsiCloseHandle(hview); 574 ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n"); 575 576 r = MsiDatabaseGetPrimaryKeysA(hdb, "drone", &rec); 577 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 578 579 r = MsiRecordGetInteger(rec, 0); 580 ok(r == MSI_NULL_INTEGER, "Expected MSI_NULL_INTEGER, got %d\n", r); 581 582 sz = MAX_PATH; 583 lstrcpyA(buf, "apple"); 584 r = MsiRecordGetStringA(rec, 0, buf, &sz); 585 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 586 ok(!lstrcmpA(buf, "drone"), "Expected \"drone\", got \"%s\"\n", buf); 587 ok(sz == 5, "Expectd 5, got %d\n", sz); 588 589 r = MsiRecordIsNull(rec, 0); 590 ok(r == FALSE, "Expected FALSE, got %d\n", r); 591 592 MsiCloseHandle(rec); 593 594 r = MsiDatabaseGetPrimaryKeysA(hdb, "nosuchtable", &rec); 595 ok(r == ERROR_INVALID_TABLE, "Expected ERROR_INVALID_TABLE, got %d\n", r); 596 597 query = "SELECT * FROM `drone` WHERE `id` = 1"; 598 r = MsiDatabaseOpenViewA(hdb, query, &hview); 599 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 600 r = MsiViewExecute(hview, 0); 601 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 602 r = MsiViewFetch(hview, &rec); 603 ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", r); 604 605 r = MsiRecordGetInteger(rec, 0); 606 ok(r != MSI_NULL_INTEGER && r != 0, "Expected non-NULL value, got %d\n", r); 607 608 r = MsiRecordIsNull(rec, 0); 609 ok(r == FALSE, "Expected FALSE, got %d\n", r); 610 611 r = MsiCloseHandle(hview); 612 ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n"); 613 MsiCloseHandle(rec); 614 MsiCloseHandle(hdb); 615 DeleteFileA(msifile); 616 } 617 618 START_TEST(record) 619 { 620 test_msirecord(); 621 test_MsiRecordGetString(); 622 test_MsiRecordGetInteger(); 623 test_fieldzero(); 624 } 625