dlls/prntvpt/tests/prntvpt.c

   1 /*
   2  * Copyright 2019 Dmitry Timoshkov
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #define COBJMACROS
  20
  21 #include <stdarg.h>
  22
  23 #include <windef.h>
  24 #include <winbase.h>
  25 #include <wingdi.h>
  26 #include <winspool.h>
  27 #include <objbase.h>
  28 #include <prntvpt.h>
  29
  30 #include "wine/test.h"
  31
  32 static WCHAR default_name[256];
  33
  34 struct hprov_data
  35 {
  36     HPTPROVIDER hprov;
  37     HRESULT hr;
  38 };
  39
  40 static DWORD WINAPI CloseProvider_proc(void *param)
  41 {
  42     struct hprov_data *data = param;
  43
  44     data->hr = PTCloseProvider(data->hprov);
  45
  46     return 0;
  47 }
  48
  49 static void test_PTOpenProvider(void)
  50 {
  51     DWORD tid, i;
  52     HPTPROVIDER hprov;
  53     HRESULT hr;
  54     HANDLE hthread;
  55     struct hprov_data data;
  56
  57     hr = PTOpenProvider(default_name, 1, &hprov);
  58     ok(hr == S_OK, "got %#x\n", hr);
  59
  60     data.hprov = hprov;
  61     hthread = CreateThread(NULL, 0, CloseProvider_proc, &data, 0, &tid);
  62     WaitForSingleObject(hthread, INFINITE);
  63     CloseHandle(hthread);
  64     ok(data.hr == E_HANDLE || data.hr == E_INVALIDARG /* XP */ || broken(data.hr == S_OK) /* Win8 */, "got %#x\n", data.hr);
  65
  66     if (data.hr != S_OK)
  67     {
  68         hr = PTCloseProvider(hprov);
  69         ok(hr == S_OK, "got %#x\n", hr);
  70     }
  71
  72     hr = PTOpenProvider(default_name, 0, &hprov);
  73     ok(hr == E_INVALIDARG, "got %#x\n", hr);
  74
  75     for (i = 2; i < 20; i++)
  76     {
  77         hr = PTOpenProvider(default_name, i, &hprov);
  78         ok(hr == 0x80040001 || hr == E_INVALIDARG /* Wine */, "%u: got %#x\n", i, hr);
  79     }
  80 }
  81
  82 static void test_PTOpenProviderEx(void)
  83 {
  84     DWORD tid, ver, i;
  85     HPTPROVIDER hprov;
  86     HRESULT hr;
  87     HANDLE hthread;
  88     struct hprov_data data;
  89
  90     hr = PTOpenProviderEx(default_name, 1, 1, &hprov, NULL);
  91     ok(hr == E_INVALIDARG, "got %#x\n", hr);
  92
  93     ver = 0xdeadbeef;
  94     hr = PTOpenProviderEx(default_name, 1, 1, &hprov, &ver);
  95     ok(hr == S_OK, "got %#x\n", hr);
  96     ok(ver == 1, "got %#x\n", ver);
  97
  98     data.hprov = hprov;
  99     hthread = CreateThread(NULL, 0, CloseProvider_proc, &data, 0, &tid);
 100     WaitForSingleObject(hthread, INFINITE);
 101     CloseHandle(hthread);
 102     ok(data.hr == E_HANDLE || data.hr == E_INVALIDARG /* XP */ || broken(data.hr == S_OK) /* Win8 */, "got %#x\n", data.hr);
 103
 104     if (data.hr != S_OK)
 105     {
 106         hr = PTCloseProvider(hprov);
 107         ok(hr == S_OK, "got %#x\n", hr);
 108     }
 109
 110     for (i = 1; i < 20; i++)
 111     {
 112         hr = PTOpenProviderEx(default_name, 0, i, &hprov, &ver);
 113         ok(hr == E_INVALIDARG, "%u: got %#x\n", i, hr);
 114
 115         ver = 0xdeadbeef;
 116         hr = PTOpenProviderEx(default_name, 1, i, &hprov, &ver);
 117         ok(hr == S_OK, "%u: got %#x\n", i, hr);
 118         ok(ver == 1, "%u: got %#x\n", i, ver);
 119         PTCloseProvider(hprov);
 120
 121         ver = 0xdeadbeef;
 122         hr = PTOpenProviderEx(default_name, i, i, &hprov, &ver);
 123         ok(hr == S_OK, "%u: got %#x\n", i, hr);
 124         ok(ver == 1, "%u: got %#x\n", i, ver);
 125         PTCloseProvider(hprov);
 126     }
 127 }
 128
 129 START_TEST(prntvpt)
 130 {
 131     DWORD size;
 132
 133     CoInitializeEx(NULL, COINIT_MULTITHREADED);
 134
 135     size = ARRAY_SIZE(default_name);
 136     if (!GetDefaultPrinterW(default_name, &size))
 137     {
 138         skip("no default printer set\n");
 139         return;
 140     }
 141
 142     trace("default printer: %s\n", wine_dbgstr_w(default_name));
 143
 144     test_PTOpenProvider();
 145     test_PTOpenProviderEx();
 146
 147     CoUninitialize();
 148 }