2 * Unit tests for SetupPromptForDisk
4 * Copyright 2014 Michael Müller
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.
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.
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
31 #include "wine/test.h"
33 static void test_SetupPromptForDiskA(void)
35 #ifndef ROSTESTS_234_IS_FIXED
36 // char path[] = "C:\\windows\\system32";
37 char path
[] = "C:\\ReactOS\\system32";
39 #error "path should use GetSystemDirectoryA()"
41 char file
[] = "kernel32.dll";
42 char buffer
[MAX_PATH
];
46 memset(buffer
, 0, sizeof(buffer
));
47 ret
= SetupPromptForDiskA(0, "Test", "Testdisk", path
, file
, 0, IDF_CHECKFIRST
, buffer
, sizeof(buffer
) - 1, &length
);
48 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %u\n", ret
);
49 ok(length
== strlen(path
)+1, "Expect length %u, got %u\n", (DWORD
)strlen(path
) + 1, length
);
50 ok(strcmp(path
, buffer
) == 0, "Expected path %s, got %s\n", path
, buffer
);
52 memset(buffer
, 0, sizeof(buffer
));
53 ret
= SetupPromptForDiskA(0, "Test", "Testdisk", path
, file
, 0, IDF_CHECKFIRST
, 0, 0, &length
);
54 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %d\n", ret
);
55 ok(length
== strlen(path
)+1, "Expect length %u, got %u\n", (DWORD
)strlen(path
) + 1, length
);
57 memset(buffer
, 0, sizeof(buffer
));
58 ret
= SetupPromptForDiskA(0, "Test", "Testdisk", path
, file
, 0, IDF_CHECKFIRST
, buffer
, 1, &length
);
59 ok(ret
== DPROMPT_BUFFERTOOSMALL
, "Expected DPROMPT_BUFFERTOOSMALL, got %u\n", ret
);
61 memset(buffer
, 0, sizeof(buffer
));
62 ret
= SetupPromptForDiskA(0, "Test", "Testdisk", path
, file
, 0, IDF_CHECKFIRST
, buffer
, strlen(path
), &length
);
63 ok(ret
== DPROMPT_BUFFERTOOSMALL
, "Expected DPROMPT_BUFFERTOOSMALL, got %u\n", ret
);
65 memset(buffer
, 0, sizeof(buffer
));
66 ret
= SetupPromptForDiskA(0, "Test", "Testdisk", path
, file
, 0, IDF_CHECKFIRST
, buffer
, strlen(path
)+1, &length
);
67 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %u\n", ret
);
68 ok(length
== strlen(path
)+1, "Expect length %u, got %u\n", (DWORD
)strlen(path
) + 1, length
);
69 ok(strcmp(path
, buffer
) == 0, "Expected path %s, got %s\n", path
, buffer
);
72 static void test_SetupPromptForDiskW(void)
74 #ifndef ROSTESTS_234_IS_FIXED
75 // WCHAR path[] = {'C',':','\\','w','i','n','d','o','w','s','\\','s','y','s','t','e','m','3','2','\0'};
76 WCHAR path
[] = {'C',':','\\','R','e','a','c','t','O','S','\\','s','y','s','t','e','m','3','2','\0'};
78 #error "path should use GetSystemDirectoryW()"
80 WCHAR file
[] = {'k','e','r','n','e','l','3','2','.','d','l','l','\0'};
81 WCHAR title
[] = {'T','e','s','t','\0'};
82 WCHAR disk
[] = {'T','e','s','t','d','i','s','k','\0'};
83 WCHAR buffer
[MAX_PATH
];
87 memset(buffer
, 0, sizeof(buffer
));
88 ret
= SetupPromptForDiskW(0, title
, disk
, path
, file
, 0, IDF_CHECKFIRST
, buffer
, MAX_PATH
-1, &length
);
89 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %u\n", ret
);
90 ok(length
== lstrlenW(path
)+1, "Expect length %u, got %u\n", lstrlenW(path
)+1, length
);
91 ok(lstrcmpW(path
, buffer
) == 0, "Expected path %s, got %s\n", wine_dbgstr_w(path
), wine_dbgstr_w(buffer
));
93 memset(buffer
, 0, sizeof(buffer
));
94 ret
= SetupPromptForDiskW(0, title
, disk
, path
, file
, 0, IDF_CHECKFIRST
, 0, 0, &length
);
95 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %d\n", ret
);
96 ok(length
== lstrlenW(path
)+1, "Expect length %u, got %u\n", lstrlenW(path
)+1, length
);
98 memset(buffer
, 0, sizeof(buffer
));
99 ret
= SetupPromptForDiskW(0, title
, disk
, path
, file
, 0, IDF_CHECKFIRST
, buffer
, 1, &length
);
100 ok(ret
== DPROMPT_BUFFERTOOSMALL
, "Expected DPROMPT_BUFFERTOOSMALL, got %u\n", ret
);
102 memset(buffer
, 0, sizeof(buffer
));
103 ret
= SetupPromptForDiskW(0, title
, disk
, path
, file
, 0, IDF_CHECKFIRST
, buffer
, lstrlenW(path
), &length
);
104 ok(ret
== DPROMPT_BUFFERTOOSMALL
, "Expected DPROMPT_BUFFERTOOSMALL, got %u\n", ret
);
106 memset(buffer
, 0, sizeof(buffer
));
107 ret
= SetupPromptForDiskW(0, title
, disk
, path
, file
, 0, IDF_CHECKFIRST
, buffer
, lstrlenW(path
)+1, &length
);
108 ok(ret
== DPROMPT_SUCCESS
, "Expected DPROMPT_SUCCESS, got %u\n", ret
);
109 ok(length
== lstrlenW(path
)+1, "Expect length %u, got %u\n", lstrlenW(path
)+1, length
);
110 ok(lstrcmpW(path
, buffer
) == 0, "Expected path %s, got %s\n", wine_dbgstr_w(path
), wine_dbgstr_w(buffer
));
115 test_SetupPromptForDiskA();
116 test_SetupPromptForDiskW();