2 * Unit test suite for Background Copy Job Interface
4 * Copyright 2007 Google (Roy Shea)
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
25 #include "wine/test.h"
28 /* Globals used by many tests */
29 static const WCHAR test_displayName
[] = {'T', 'e', 's', 't', 0};
30 static const WCHAR test_remoteNameA
[] = {'r','e','m','o','t','e','A', 0};
31 static const WCHAR test_remoteNameB
[] = {'r','e','m','o','t','e','B', 0};
32 static const WCHAR test_localNameA
[] = {'l','o','c','a','l','A', 0};
33 static const WCHAR test_localNameB
[] = {'l','o','c','a','l','B', 0};
34 static WCHAR
*test_currentDir
;
35 static WCHAR
*test_remotePathA
;
36 static WCHAR
*test_remotePathB
;
37 static WCHAR
*test_localPathA
;
38 static WCHAR
*test_localPathB
;
39 static IBackgroundCopyManager
*test_manager
;
40 static IBackgroundCopyJob
*test_job
;
41 static GUID test_jobId
;
42 static BG_JOB_TYPE test_type
;
44 static BOOL
init_paths(void)
46 static const WCHAR format
[] = {'%','s','\\','%','s', 0};
49 n
= GetCurrentDirectoryW(0, NULL
);
52 skip("Couldn't get current directory size\n");
56 test_currentDir
= HeapAlloc(GetProcessHeap(), 0, n
* sizeof(WCHAR
));
58 = HeapAlloc(GetProcessHeap(), 0,
59 (n
+ 1 + lstrlenW(test_localNameA
)) * sizeof(WCHAR
));
61 = HeapAlloc(GetProcessHeap(), 0,
62 (n
+ 1 + lstrlenW(test_localNameB
)) * sizeof(WCHAR
));
64 = HeapAlloc(GetProcessHeap(), 0,
65 (n
+ 1 + lstrlenW(test_remoteNameA
)) * sizeof(WCHAR
));
67 = HeapAlloc(GetProcessHeap(), 0,
68 (n
+ 1 + lstrlenW(test_remoteNameB
)) * sizeof(WCHAR
));
70 if (!test_currentDir
|| !test_localPathA
|| !test_localPathB
71 || !test_remotePathA
|| !test_remotePathB
)
73 skip("Couldn't allocate memory for full paths\n");
77 if (GetCurrentDirectoryW(n
, test_currentDir
) != n
- 1)
79 skip("Couldn't get current directory\n");
83 wsprintfW(test_localPathA
, format
, test_currentDir
, test_localNameA
);
84 wsprintfW(test_localPathB
, format
, test_currentDir
, test_localNameB
);
85 wsprintfW(test_remotePathA
, format
, test_currentDir
, test_remoteNameA
);
86 wsprintfW(test_remotePathB
, format
, test_currentDir
, test_remoteNameB
);
91 /* Generic test setup */
92 static BOOL
setup(void)
98 memset(&test_jobId
, 0, sizeof test_jobId
);
99 test_type
= BG_JOB_TYPE_DOWNLOAD
;
101 hres
= CoCreateInstance(&CLSID_BackgroundCopyManager
, NULL
,
103 &IID_IBackgroundCopyManager
,
104 (void **) &test_manager
);
108 hres
= IBackgroundCopyManager_CreateJob(test_manager
, test_displayName
,
109 test_type
, &test_jobId
, &test_job
);
112 IBackgroundCopyManager_Release(test_manager
);
119 /* Generic test cleanup */
120 static void teardown(void)
122 IBackgroundCopyJob_Release(test_job
);
123 IBackgroundCopyManager_Release(test_manager
);
126 /* Test that the jobId is properly set */
127 static void test_GetId(void)
132 hres
= IBackgroundCopyJob_GetId(test_job
, &tmpId
);
133 ok(hres
== S_OK
, "GetId failed: %08x\n", hres
);
136 skip("Unable to get ID of test_job.\n");
139 ok(memcmp(&tmpId
, &test_jobId
, sizeof tmpId
) == 0, "Got incorrect GUID\n");
142 /* Test that the type is properly set */
143 static void test_GetType(void)
148 hres
= IBackgroundCopyJob_GetType(test_job
, &type
);
149 ok(hres
== S_OK
, "GetType failed: %08x\n", hres
);
152 skip("Unable to get type of test_job.\n");
155 ok(type
== test_type
, "Got incorrect type\n");
158 /* Test that the display name is properly set */
159 static void test_GetName(void)
164 hres
= IBackgroundCopyJob_GetDisplayName(test_job
, &displayName
);
165 ok(hres
== S_OK
, "GetName failed: %08x\n", hres
);
168 skip("Unable to get display name of test_job.\n");
171 ok(lstrcmpW(displayName
, test_displayName
) == 0, "Got incorrect type\n");
172 CoTaskMemFree(displayName
);
175 /* Test adding a file */
176 static void test_AddFile(void)
180 hres
= IBackgroundCopyJob_AddFile(test_job
, test_remotePathA
,
182 ok(hres
== S_OK
, "First call to AddFile failed: 0x%08x\n", hres
);
185 skip("Unable to add first file to job\n");
189 hres
= IBackgroundCopyJob_AddFile(test_job
, test_remotePathB
,
191 ok(hres
== S_OK
, "Second call to AddFile failed: 0x%08x\n", hres
);
194 /* Test adding a set of files */
195 static void test_AddFileSet(void)
198 BG_FILE_INFO files
[2] =
200 {test_remotePathA
, test_localPathA
},
201 {test_remotePathB
, test_localPathB
}
203 hres
= IBackgroundCopyJob_AddFileSet(test_job
, 2, files
);
204 ok(hres
== S_OK
, "AddFileSet failed: 0x%08x\n", hres
);
207 /* Test creation of a job enumerator */
208 static void test_EnumFiles(void)
211 IEnumBackgroundCopyFiles
*enumFiles
;
214 hres
= IBackgroundCopyJob_AddFile(test_job
, test_remotePathA
,
218 skip("Unable to add file to job\n");
222 hres
= IBackgroundCopyJob_EnumFiles(test_job
, &enumFiles
);
223 ok(hres
== S_OK
, "EnumFiles failed: 0x%08x\n", hres
);
226 skip("Unable to create file enumerator.\n");
230 res
= IEnumBackgroundCopyFiles_Release(enumFiles
);
231 ok(res
== 0, "Bad ref count on release: %u\n", res
);
234 /* Test getting job progress */
235 static void test_GetProgress_preTransfer(void)
238 BG_JOB_PROGRESS progress
;
240 hres
= IBackgroundCopyJob_GetProgress(test_job
, &progress
);
241 ok(hres
== S_OK
, "GetProgress failed: 0x%08x\n", hres
);
244 skip("Unable to get job progress\n");
249 ok(progress
.BytesTotal
== 0, "Incorrect BytesTotal: %llu\n", progress
.BytesTotal
);
250 ok(progress
.BytesTransferred
== 0, "Incorrect BytesTransferred: %llu\n", progress
.BytesTransferred
);
251 ok(progress
.FilesTotal
== 0, "Incorrect FilesTotal: %u\n", progress
.FilesTotal
);
252 ok(progress
.FilesTransferred
== 0, "Incorrect FilesTransferred %u\n", progress
.FilesTransferred
);
255 /* Test getting job state */
256 static void test_GetState(void)
261 state
= BG_JOB_STATE_ERROR
;
262 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
263 ok(hres
== S_OK
, "GetState failed: 0x%08x\n", hres
);
266 skip("Unable to get job state\n");
269 ok(state
== BG_JOB_STATE_SUSPENDED
, "Incorrect job state: %d\n", state
);
272 /* Test resuming a job */
273 static void test_ResumeEmpty(void)
278 hres
= IBackgroundCopyJob_Resume(test_job
);
279 ok(hres
== BG_E_EMPTY
, "Resume failed to return BG_E_EMPTY error: 0x%08x\n", hres
);
280 if (hres
!= BG_E_EMPTY
)
282 skip("Failed calling resume job\n");
286 state
= BG_JOB_STATE_ERROR
;
287 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
290 skip("Unable to get job state\n");
293 ok(state
== BG_JOB_STATE_SUSPENDED
, "Incorrect job state: %d\n", state
);
296 static void makeFile(WCHAR
*name
, const char *contents
)
299 DWORD w
, len
= strlen(contents
);
302 file
= CreateFileW(name
, GENERIC_WRITE
, 0, NULL
, CREATE_ALWAYS
,
303 FILE_ATTRIBUTE_NORMAL
, NULL
);
304 ok(file
!= INVALID_HANDLE_VALUE
, "CreateFile\n");
305 ok(WriteFile(file
, contents
, len
, &w
, NULL
), "WriteFile\n");
309 static void compareFiles(WCHAR
*n1
, WCHAR
*n2
)
316 f1
= CreateFileW(n1
, GENERIC_READ
, FILE_SHARE_READ
, NULL
, OPEN_EXISTING
,
317 FILE_ATTRIBUTE_NORMAL
, NULL
);
318 ok(f1
!= INVALID_HANDLE_VALUE
, "CreateFile\n");
320 f2
= CreateFileW(n2
, GENERIC_READ
, FILE_SHARE_READ
, NULL
, OPEN_EXISTING
,
321 FILE_ATTRIBUTE_NORMAL
, NULL
);
322 ok(f1
!= INVALID_HANDLE_VALUE
, "CreateFile\n");
324 /* Neither of these files is very big */
325 ok(ReadFile(f1
, b1
, sizeof b1
, &s1
, NULL
), "ReadFile\n");
326 ok(ReadFile(f2
, b2
, sizeof b2
, &s2
, NULL
), "ReadFile\n");
331 ok(s1
== s2
, "Files differ in length\n");
332 ok(memcmp(b1
, b2
, s1
) == 0, "Files differ in contents\n");
335 /* Test a complete transfer for local files */
336 static void test_CompleteLocal(void)
338 static const int timeout_sec
= 30;
343 DeleteFileW(test_localPathA
);
344 DeleteFileW(test_localPathB
);
345 makeFile(test_remotePathA
, "This is a WINE test file for BITS\n");
346 makeFile(test_remotePathB
, "This is another WINE test file for BITS\n");
348 hres
= IBackgroundCopyJob_AddFile(test_job
, test_remotePathA
,
352 skip("Unable to add file to job\n");
356 hres
= IBackgroundCopyJob_AddFile(test_job
, test_remotePathB
,
360 skip("Unable to add file to job\n");
364 hres
= IBackgroundCopyJob_Resume(test_job
);
365 ok(hres
== S_OK
, "IBackgroundCopyJob_Resume\n");
367 for (i
= 0; i
< timeout_sec
; ++i
)
369 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
370 ok(hres
== S_OK
, "IBackgroundCopyJob_GetState\n");
371 ok(state
== BG_JOB_STATE_QUEUED
|| state
== BG_JOB_STATE_CONNECTING
372 || state
== BG_JOB_STATE_TRANSFERRING
|| state
== BG_JOB_STATE_TRANSFERRED
,
373 "Bad state: %d\n", state
);
374 if (state
== BG_JOB_STATE_TRANSFERRED
)
379 ok(i
< timeout_sec
, "BITS jobs timed out\n");
380 hres
= IBackgroundCopyJob_Complete(test_job
);
381 ok(hres
== S_OK
, "IBackgroundCopyJob_Complete\n");
382 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
383 ok(hres
== S_OK
, "IBackgroundCopyJob_GetState\n");
384 ok(state
== BG_JOB_STATE_ACKNOWLEDGED
, "Bad state: %d\n", state
);
386 compareFiles(test_remotePathA
, test_localPathA
);
387 compareFiles(test_remotePathB
, test_localPathB
);
389 ok(DeleteFileW(test_remotePathA
), "DeleteFile\n");
390 ok(DeleteFileW(test_remotePathB
), "DeleteFile\n");
391 DeleteFileW(test_localPathA
);
392 DeleteFileW(test_localPathB
);
395 /* Test a complete transfer for local files */
396 static void test_CompleteLocalURL(void)
398 static const WCHAR prot
[] = {'f','i','l','e',':','/','/', 0};
399 static const int timeout_sec
= 30;
405 DeleteFileW(test_localPathA
);
406 DeleteFileW(test_localPathB
);
407 makeFile(test_remotePathA
, "This is a WINE test file for BITS\n");
408 makeFile(test_remotePathB
, "This is another WINE test file for BITS\n");
410 urlA
= HeapAlloc(GetProcessHeap(), 0,
411 (7 + lstrlenW(test_remotePathA
) + 1) * sizeof urlA
[0]);
412 urlB
= HeapAlloc(GetProcessHeap(), 0,
413 (7 + lstrlenW(test_remotePathB
) + 1) * sizeof urlB
[0]);
416 skip("Unable to allocate memory for URLs\n");
420 lstrcpyW(urlA
, prot
);
421 lstrcatW(urlA
, test_remotePathA
);
422 lstrcpyW(urlB
, prot
);
423 lstrcatW(urlB
, test_remotePathB
);
425 hres
= IBackgroundCopyJob_AddFile(test_job
, urlA
, test_localPathA
);
428 skip("Unable to add file to job\n");
432 hres
= IBackgroundCopyJob_AddFile(test_job
, urlB
, test_localPathB
);
435 skip("Unable to add file to job\n");
439 hres
= IBackgroundCopyJob_Resume(test_job
);
440 ok(hres
== S_OK
, "IBackgroundCopyJob_Resume\n");
442 for (i
= 0; i
< timeout_sec
; ++i
)
444 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
445 ok(hres
== S_OK
, "IBackgroundCopyJob_GetState\n");
446 ok(state
== BG_JOB_STATE_QUEUED
|| state
== BG_JOB_STATE_CONNECTING
447 || state
== BG_JOB_STATE_TRANSFERRING
|| state
== BG_JOB_STATE_TRANSFERRED
,
448 "Bad state: %d\n", state
);
449 if (state
== BG_JOB_STATE_TRANSFERRED
)
454 ok(i
< timeout_sec
, "BITS jobs timed out\n");
455 hres
= IBackgroundCopyJob_Complete(test_job
);
456 ok(hres
== S_OK
, "IBackgroundCopyJob_Complete\n");
457 hres
= IBackgroundCopyJob_GetState(test_job
, &state
);
458 ok(hres
== S_OK
, "IBackgroundCopyJob_GetState\n");
459 ok(state
== BG_JOB_STATE_ACKNOWLEDGED
, "Bad state: %d\n", state
);
461 compareFiles(test_remotePathA
, test_localPathA
);
462 compareFiles(test_remotePathB
, test_localPathB
);
464 ok(DeleteFileW(test_remotePathA
), "DeleteFile\n");
465 ok(DeleteFileW(test_remotePathB
), "DeleteFile\n");
466 DeleteFileW(test_localPathA
);
467 DeleteFileW(test_localPathB
);
469 HeapFree(GetProcessHeap(), 0, urlA
);
470 HeapFree(GetProcessHeap(), 0, urlB
);
473 typedef void (*test_t
)(void);
477 static const test_t tests
[] = {
484 test_GetProgress_preTransfer
,
488 test_CompleteLocalURL
,
497 for (test
= tests
; *test
; ++test
)
499 /* Keep state seperate between tests. */
502 skip("Unable to setup test\n");