2008-02-23 00:07:12 +01:00
|
|
|
/*
|
|
|
|
* Unit test suite for Background Copy Job Interface
|
|
|
|
*
|
|
|
|
* Copyright 2007 Google (Roy Shea)
|
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
#define COBJMACROS
|
|
|
|
|
|
|
|
#include "wine/test.h"
|
|
|
|
#include "bits.h"
|
|
|
|
|
|
|
|
/* Globals used by many tests */
|
|
|
|
static const WCHAR test_displayName[] = {'T', 'e', 's', 't', 0};
|
2008-02-27 02:51:20 +01:00
|
|
|
static const WCHAR test_remoteNameA[] = {'r','e','m','o','t','e','A', 0};
|
|
|
|
static const WCHAR test_remoteNameB[] = {'r','e','m','o','t','e','B', 0};
|
|
|
|
static const WCHAR test_localNameA[] = {'l','o','c','a','l','A', 0};
|
|
|
|
static const WCHAR test_localNameB[] = {'l','o','c','a','l','B', 0};
|
|
|
|
static WCHAR *test_currentDir;
|
|
|
|
static WCHAR *test_remotePathA;
|
|
|
|
static WCHAR *test_remotePathB;
|
|
|
|
static WCHAR *test_localPathA;
|
|
|
|
static WCHAR *test_localPathB;
|
2008-02-23 00:07:12 +01:00
|
|
|
static IBackgroundCopyManager *test_manager;
|
|
|
|
static IBackgroundCopyJob *test_job;
|
|
|
|
static GUID test_jobId;
|
|
|
|
static BG_JOB_TYPE test_type;
|
|
|
|
|
2008-02-27 02:51:20 +01:00
|
|
|
static BOOL init_paths(void)
|
|
|
|
{
|
|
|
|
static const WCHAR format[] = {'%','s','\\','%','s', 0};
|
|
|
|
DWORD n;
|
|
|
|
|
|
|
|
n = GetCurrentDirectoryW(0, NULL);
|
|
|
|
if (n == 0)
|
|
|
|
{
|
|
|
|
skip("Couldn't get current directory size\n");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
test_currentDir = HeapAlloc(GetProcessHeap(), 0, n * sizeof(WCHAR));
|
|
|
|
test_localPathA
|
|
|
|
= HeapAlloc(GetProcessHeap(), 0,
|
|
|
|
(n + 1 + lstrlenW(test_localNameA)) * sizeof(WCHAR));
|
|
|
|
test_localPathB
|
|
|
|
= HeapAlloc(GetProcessHeap(), 0,
|
|
|
|
(n + 1 + lstrlenW(test_localNameB)) * sizeof(WCHAR));
|
|
|
|
test_remotePathA
|
|
|
|
= HeapAlloc(GetProcessHeap(), 0,
|
|
|
|
(n + 1 + lstrlenW(test_remoteNameA)) * sizeof(WCHAR));
|
|
|
|
test_remotePathB
|
|
|
|
= HeapAlloc(GetProcessHeap(), 0,
|
|
|
|
(n + 1 + lstrlenW(test_remoteNameB)) * sizeof(WCHAR));
|
|
|
|
|
|
|
|
if (!test_currentDir || !test_localPathA || !test_localPathB
|
|
|
|
|| !test_remotePathA || !test_remotePathB)
|
|
|
|
{
|
|
|
|
skip("Couldn't allocate memory for full paths\n");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (GetCurrentDirectoryW(n, test_currentDir) != n - 1)
|
|
|
|
{
|
|
|
|
skip("Couldn't get current directory\n");
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
wsprintfW(test_localPathA, format, test_currentDir, test_localNameA);
|
|
|
|
wsprintfW(test_localPathB, format, test_currentDir, test_localNameB);
|
|
|
|
wsprintfW(test_remotePathA, format, test_currentDir, test_remoteNameA);
|
|
|
|
wsprintfW(test_remotePathB, format, test_currentDir, test_remoteNameB);
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2008-02-23 00:07:12 +01:00
|
|
|
/* Generic test setup */
|
|
|
|
static BOOL setup(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
|
|
|
|
test_manager = NULL;
|
|
|
|
test_job = NULL;
|
|
|
|
memset(&test_jobId, 0, sizeof test_jobId);
|
|
|
|
test_type = BG_JOB_TYPE_DOWNLOAD;
|
|
|
|
|
|
|
|
hres = CoCreateInstance(&CLSID_BackgroundCopyManager, NULL,
|
|
|
|
CLSCTX_LOCAL_SERVER,
|
|
|
|
&IID_IBackgroundCopyManager,
|
|
|
|
(void **) &test_manager);
|
|
|
|
if(hres != S_OK)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyManager_CreateJob(test_manager, test_displayName,
|
|
|
|
test_type, &test_jobId, &test_job);
|
|
|
|
if(hres != S_OK)
|
|
|
|
{
|
|
|
|
IBackgroundCopyManager_Release(test_manager);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Generic test cleanup */
|
|
|
|
static void teardown(void)
|
|
|
|
{
|
|
|
|
IBackgroundCopyJob_Release(test_job);
|
|
|
|
IBackgroundCopyManager_Release(test_manager);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Test that the jobId is properly set */
|
|
|
|
static void test_GetId(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
GUID tmpId;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_GetId(test_job, &tmpId);
|
|
|
|
ok(hres == S_OK, "GetId failed: %08x\n", hres);
|
|
|
|
if(hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to get ID of test_job.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
ok(memcmp(&tmpId, &test_jobId, sizeof tmpId) == 0, "Got incorrect GUID\n");
|
|
|
|
}
|
|
|
|
|
2008-02-26 02:40:57 +01:00
|
|
|
/* Test that the type is properly set */
|
|
|
|
static void test_GetType(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
BG_JOB_TYPE type;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_GetType(test_job, &type);
|
|
|
|
ok(hres == S_OK, "GetType failed: %08x\n", hres);
|
|
|
|
if(hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to get type of test_job.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
ok(type == test_type, "Got incorrect type\n");
|
|
|
|
}
|
|
|
|
|
2008-02-26 02:41:35 +01:00
|
|
|
/* Test that the display name is properly set */
|
|
|
|
static void test_GetName(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
LPWSTR displayName;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_GetDisplayName(test_job, &displayName);
|
|
|
|
ok(hres == S_OK, "GetName failed: %08x\n", hres);
|
|
|
|
if(hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to get display name of test_job.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
ok(lstrcmpW(displayName, test_displayName) == 0, "Got incorrect type\n");
|
|
|
|
CoTaskMemFree(displayName);
|
|
|
|
}
|
|
|
|
|
2008-02-27 02:51:20 +01:00
|
|
|
/* Test adding a file */
|
|
|
|
static void test_AddFile(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_AddFile(test_job, test_remotePathA,
|
|
|
|
test_localPathA);
|
|
|
|
ok(hres == S_OK, "First call to AddFile failed: 0x%08x\n", hres);
|
|
|
|
if (hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to add first file to job\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_AddFile(test_job, test_remotePathB,
|
|
|
|
test_localPathB);
|
|
|
|
ok(hres == S_OK, "Second call to AddFile failed: 0x%08x\n", hres);
|
|
|
|
}
|
|
|
|
|
2008-02-28 06:04:07 +01:00
|
|
|
/* Test creation of a job enumerator */
|
|
|
|
static void test_EnumFiles(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
IEnumBackgroundCopyFiles *enumFiles;
|
|
|
|
ULONG res;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_AddFile(test_job, test_remotePathA,
|
|
|
|
test_localPathA);
|
|
|
|
if (hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to add file to job\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_EnumFiles(test_job, &enumFiles);
|
|
|
|
ok(hres == S_OK, "EnumFiles failed: 0x%08x\n", hres);
|
|
|
|
if(hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to create file enumerator.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
res = IEnumBackgroundCopyFiles_Release(enumFiles);
|
|
|
|
ok(res == 0, "Bad ref count on release: %u\n", res);
|
|
|
|
}
|
|
|
|
|
2008-03-04 01:48:08 +01:00
|
|
|
/* Test getting job progress */
|
|
|
|
static void test_GetProgress_preTransfer(void)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
|
|
|
BG_JOB_PROGRESS progress;
|
|
|
|
|
|
|
|
hres = IBackgroundCopyJob_GetProgress(test_job, &progress);
|
|
|
|
ok(hres == S_OK, "GetProgress failed: 0x%08x\n", hres);
|
|
|
|
if (hres != S_OK)
|
|
|
|
{
|
|
|
|
skip("Unable to get job progress\n");
|
|
|
|
teardown();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ok(progress.BytesTotal == 0, "Incorrect BytesTotal: %llu\n", progress.BytesTotal);
|
|
|
|
ok(progress.BytesTransferred == 0, "Incorrect BytesTransferred: %llu\n", progress.BytesTransferred);
|
|
|
|
ok(progress.FilesTotal == 0, "Incorrect FilesTotal: %u\n", progress.FilesTotal);
|
|
|
|
ok(progress.FilesTransferred == 0, "Incorrect FilesTransferred %u\n", progress.FilesTransferred);
|
|
|
|
}
|
|
|
|
|
2008-02-23 00:07:12 +01:00
|
|
|
typedef void (*test_t)(void);
|
|
|
|
|
|
|
|
START_TEST(job)
|
|
|
|
{
|
|
|
|
static const test_t tests[] = {
|
|
|
|
test_GetId,
|
2008-02-26 02:40:57 +01:00
|
|
|
test_GetType,
|
2008-02-26 02:41:35 +01:00
|
|
|
test_GetName,
|
2008-02-27 02:51:20 +01:00
|
|
|
test_AddFile,
|
2008-02-28 06:04:07 +01:00
|
|
|
test_EnumFiles,
|
2008-03-04 01:48:08 +01:00
|
|
|
test_GetProgress_preTransfer,
|
2008-02-23 00:07:12 +01:00
|
|
|
0
|
|
|
|
};
|
|
|
|
const test_t *test;
|
|
|
|
|
2008-02-27 02:51:20 +01:00
|
|
|
if (!init_paths())
|
|
|
|
return;
|
|
|
|
|
2008-02-23 00:07:12 +01:00
|
|
|
CoInitialize(NULL);
|
|
|
|
for (test = tests; *test; ++test)
|
|
|
|
{
|
|
|
|
/* Keep state seperate between tests. */
|
|
|
|
if (!setup())
|
|
|
|
{
|
|
|
|
skip("Unable to setup test\n");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
(*test)();
|
|
|
|
teardown();
|
|
|
|
}
|
|
|
|
CoUninitialize();
|
|
|
|
}
|