[PATCH] qmgr/tests: Test registering an IBackgroundCopyCallback2
aaron
aa1ronham at gmail.com
Fri Oct 16 11:23:48 CDT 2020
Several tests failed, but all of the failures appear to be spurious
network issues unrelated to this patch (the existing test tries to
download a file over the network using BITS).
On Thu, Oct 15, 2020 at 08:38:00PM -0400, Aaron Hill wrote:
> Test that we can register an IBackgroundCopyCallback2 with a job
> successfully.
>
> This does not test that any of the callback functions in
> the IBackgroundCopyCallback2 are actually called. Accordings to the BITS
> docs, these functions are called on separate threads on Windows, so a
> any test will need to take this into account to avoid spurious failures.
>
> Signed-off-by: Aaron Hill <aa1ronham at gmail.com>
> ---
> dlls/qmgr/tests/Makefile.in | 2 +-
> dlls/qmgr/tests/job.c | 137 ++++++++++++++++++++++++++++++++++++
> 2 files changed, 138 insertions(+), 1 deletion(-)
>
> diff --git a/dlls/qmgr/tests/Makefile.in b/dlls/qmgr/tests/Makefile.in
> index 6482d4d9b7a..2bf92a94019 100644
> --- a/dlls/qmgr/tests/Makefile.in
> +++ b/dlls/qmgr/tests/Makefile.in
> @@ -1,5 +1,5 @@
> TESTDLL = qmgr.dll
> -IMPORTS = ole32 shlwapi user32
> +IMPORTS = uuid ole32 shlwapi user32
>
> C_SRCS = \
> enum_files.c \
> diff --git a/dlls/qmgr/tests/job.c b/dlls/qmgr/tests/job.c
> index 53b724eadbf..d4e256b4469 100644
> --- a/dlls/qmgr/tests/job.c
> +++ b/dlls/qmgr/tests/job.c
> @@ -38,6 +38,123 @@ static IBackgroundCopyJob *test_job;
> static GUID test_jobId;
> static BG_JOB_TYPE test_type;
>
> +typedef struct IBackgroundCopyCallback2Impl {
> + IBackgroundCopyCallback2 IBackgroundCopyCallback2_iface;
> + LONG ref;
> +} IBackgroundCopyCallback2Impl;
> +
> +static inline IBackgroundCopyCallback2Impl *impl_from_IBackgroundCopyCallback2(IBackgroundCopyCallback2 *iface)
> +{
> + return CONTAINING_RECORD(iface, IBackgroundCopyCallback2Impl, IBackgroundCopyCallback2_iface);
> +}
> +
> +static HRESULT WINAPI IBackgroundCopyCallback2Impl_QueryInterface(IBackgroundCopyCallback2 *iface, REFIID riid, void **ppv)
> +{
> + IBackgroundCopyCallback2Impl *This = impl_from_IBackgroundCopyCallback2(iface);
> +
> + if (!ppv)
> + {
> + return E_INVALIDARG;
> + }
> +
> + *ppv = NULL;
> +
> + if (IsEqualIID(riid, &IID_IUnknown) ||
> + IsEqualIID(riid, &IID_IBackgroundCopyCallback) ||
> + IsEqualIID(riid, &IID_IBackgroundCopyCallback2))
> + {
> + *ppv = &This->IBackgroundCopyCallback2_iface;
> + }
> + else
> + {
> + return E_NOINTERFACE;
> + }
> +
> + IUnknown_AddRef((IUnknown*)*ppv);
> + return S_OK;
> +}
> +
> +static ULONG WINAPI IBackgroundCopyCallback2Impl_AddRef(IBackgroundCopyCallback2 *iface)
> +{
> + IBackgroundCopyCallback2Impl *This = impl_from_IBackgroundCopyCallback2(iface);
> + ULONG ref = InterlockedIncrement(&This->ref);
> +
> + trace("IBackgroundCopyCallback2Impl_AddRef called (%p, ref = %d)\n", This, ref);
> + return ref;
> +}
> +
> +static ULONG WINAPI IBackgroundCopyCallback2Impl_Release(IBackgroundCopyCallback2 *iface)
> +{
> + IBackgroundCopyCallback2Impl *This = impl_from_IBackgroundCopyCallback2(iface);
> + ULONG ref = InterlockedDecrement(&This->ref);
> +
> + trace("IBackgroundCopyCallback2Impl_Release called (%p, ref = %d)\n", This, ref);
> +
> + if (ref == 0)
> + {
> + HeapFree(GetProcessHeap(), 0, This);
> + }
> +
> + return ref;
> +}
> +
> +static HRESULT WINAPI IBackgroundCopyCallback2Impl_JobError(IBackgroundCopyCallback2 *iface, IBackgroundCopyJob *pJob, IBackgroundCopyError *pError)
> +{
> + trace("IBackgroundCopyCallback2Impl_JobError called (%p, %p, %p)\n", iface, pJob, pError);
> + return S_OK;
> +}
> +
> +static HRESULT WINAPI IBackgroundCopyCallback2Impl_JobModification(IBackgroundCopyCallback2 *iface, IBackgroundCopyJob *pJob, DWORD dwReserved)
> +{
> + trace("IBackgroundCopyCallback2Impl_JobModification called (%p, %p)\n", iface, pJob);
> + return S_OK;
> +}
> +
> +static HRESULT WINAPI IBackgroundCopyCallback2Impl_JobTransferred(IBackgroundCopyCallback2 *iface, IBackgroundCopyJob *pJob)
> +{
> + trace("IBackgroundCopyCallback2Impl_JobTransferred called (%p, %p)\n", iface, pJob);
> + return S_OK;
> +}
> +
> +static HRESULT WINAPI IBackgroundCopyCallback2Impl_FileTransferred(IBackgroundCopyCallback2 *iface, IBackgroundCopyJob *pJob, IBackgroundCopyFile *pFile)
> +{
> + trace("IBackgroundCopyCallback2Impl_FileTransferred called (%p, %p, %p)\n", iface, pJob, pFile);
> + return S_OK;
> +}
> +
> +
> +static const IBackgroundCopyCallback2Vtbl copyCallback_vtbl =
> +{
> + IBackgroundCopyCallback2Impl_QueryInterface,
> + IBackgroundCopyCallback2Impl_AddRef,
> + IBackgroundCopyCallback2Impl_Release,
> + IBackgroundCopyCallback2Impl_JobTransferred,
> + IBackgroundCopyCallback2Impl_JobError,
> + IBackgroundCopyCallback2Impl_JobModification,
> + IBackgroundCopyCallback2Impl_FileTransferred
> +};
> +
> +static BOOL create_background_copy_callback2(IBackgroundCopyCallback2 **copyCallback)
> +{
> + IBackgroundCopyCallback2Impl *obj;
> + *copyCallback = NULL;
> +
> + obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*obj));
> +
> + if (!obj)
> + {
> + trace("Out of memory creating IBackgroundCopyCallback2\n");
> + return FALSE;
> + }
> +
> + obj->IBackgroundCopyCallback2_iface.lpVtbl = ©Callback_vtbl;
> + obj->ref = 1;
> +
> + *copyCallback = &obj->IBackgroundCopyCallback2_iface;
> +
> + return TRUE;
> +}
> +
> static HRESULT test_create_manager(void)
> {
> HRESULT hres;
> @@ -512,6 +629,20 @@ static void test_HttpOptions(void)
> unsigned int i;
> WCHAR *headers;
> ULONG flags, orig_flags;
> + IBackgroundCopyCallback2 *copyCallback;
> + IUnknown *copyCallbackUnknown;
> +
> + ok(create_background_copy_callback2(©Callback) == TRUE, "create_background_copy_callback2 failed\n");
> +
> + hr = IBackgroundCopyCallback2_QueryInterface(copyCallback, &IID_IUnknown, (LPVOID*)©CallbackUnknown);
> + ok(hr == S_OK,"IBackgroundCopyCallback_QueryInterface(IID_IUnknown) failed: %08x\n", hr);
> +
> + hr = IBackgroundCopyJob_SetNotifyInterface(test_job, copyCallbackUnknown);
> + ok(hr == S_OK,"IBackgroundCopyCallback_SetNotifyInterface failed: %08x\n", hr);
> +
> + hr = IBackgroundCopyJob_SetNotifyFlags(test_job, BG_NOTIFY_JOB_TRANSFERRED | BG_NOTIFY_JOB_ERROR | BG_NOTIFY_DISABLE | BG_NOTIFY_JOB_MODIFICATION | BG_NOTIFY_FILE_TRANSFERRED);
> + ok(hr == S_OK,"IBackgroundCopyCallback_SetNotifyFlags failed: %08x\n", hr);
> +
>
> DeleteFileW(test_localPathA);
> hr = IBackgroundCopyJob_AddFile(test_job, L"http://test.winehq.org/", test_localPathA);
> @@ -619,6 +750,12 @@ static void test_HttpOptions(void)
> ok(hr == BG_E_INVALID_STATE, "got 0x%08x\n", hr);
>
> DeleteFileW(test_localPathA);
> +
> + hr = IBackgroundCopyJob_SetNotifyInterface(test_job, NULL);
> + ok(hr == BG_E_INVALID_STATE, "got 0x%08x\n", hr);
> +
> + IUnknown_Release(copyCallbackUnknown);
> + IBackgroundCopyCallback2_Release(copyCallback);
> }
>
> typedef void (*test_t)(void);
> --
> 2.28.0
>
More information about the wine-devel
mailing list