Paul Vriens : advapi32/tests: Add some service related tests.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 3418ed23e199012cef6c15bbe6fe01a1a20c85e6
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=3418ed23e199012cef6c15bbe6...

Author: Paul Vriens <paul.vriens.wine(a)gmail.com>
Date:   Tue Jun 26 12:56:53 2007 +0200

advapi32/tests: Add some service related tests.

---

 dlls/advapi32/tests/Makefile.in |    3 +-
 dlls/advapi32/tests/service.c   |  150 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 152 insertions(+), 1 deletions(-)

diff --git a/dlls/advapi32/tests/Makefile.in b/dlls/advapi32/tests/Makefile.in
index e5d9a9f..2036759 100644
--- a/dlls/advapi32/tests/Makefile.in
+++ b/dlls/advapi32/tests/Makefile.in
@@ -14,7 +14,8 @@ CTESTS = \
 	crypt_sha.c \
 	lsa.c \
 	registry.c \
-	security.c
+	security.c \
+	service.c
 
 @MAKE_TEST_RULES@
 
diff --git a/dlls/advapi32/tests/service.c b/dlls/advapi32/tests/service.c
new file mode 100644
index 0000000..7dde5f5
--- /dev/null
+++ b/dlls/advapi32/tests/service.c
@@ -0,0 +1,150 @@
+/*
+ * Unit tests for service functions
+ *
+ * Copyright (c) 2007 Paul Vriens
+ *
+ * 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 <stdarg.h>
+
+#include "windef.h"
+#include "winbase.h"
+#include "winerror.h"
+#include "winsvc.h"
+
+#include "wine/test.h"
+
+static void test_sequence(void)
+{
+    SC_HANDLE scm_handle, svc_handle;
+    BOOL ret;
+    QUERY_SERVICE_CONFIGA *config;
+    DWORD given, needed;
+    static const CHAR servicename [] = "Winetest";
+    static const CHAR displayname [] = "Winetest dummy service";
+    static const CHAR pathname    [] = "we_dont_care.exe";
+    static const CHAR dependencies[] = "Master1\0Master2\0+MasterGroup1\0\0";
+    static const CHAR password    [] = "";
+    static const CHAR empty       [] = "";
+    static const CHAR localsystem [] = "LocalSystem";
+
+    SetLastError(0xdeadbeef);
+    scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
+
+    if (!scm_handle && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
+    {
+        skip("OpenSCManagerA is not implemented\n");
+        return;
+    }
+    else if (!scm_handle && (GetLastError() == ERROR_ACCESS_DENIED))
+    {
+        skip("Not enough rights to get a handle to the manager\n");
+        return;
+    }
+    else
+        ok(scm_handle != NULL, "Could not get a handle to the manager: %d\n", GetLastError());
+
+    if (!scm_handle) return;
+
+    /* Create a dummy service */
+    SetLastError(0xdeadbeef);
+    svc_handle = CreateServiceA(scm_handle, servicename, displayname, GENERIC_ALL,
+        SERVICE_INTERACTIVE_PROCESS | SERVICE_WIN32_OWN_PROCESS, SERVICE_DISABLED, SERVICE_ERROR_IGNORE,
+        pathname, NULL, NULL, dependencies, NULL, password);
+
+    if (!svc_handle && (GetLastError() == ERROR_SERVICE_EXISTS))
+    {
+        /* We try and open the service and do the rest of the tests. Some could
+         * fail if the tests were changed between these runs.
+         */
+        trace("Deletion probably didn't work last time\n");
+        SetLastError(0xdeadbeef);
+        svc_handle = OpenServiceA(scm_handle, servicename, GENERIC_ALL);
+        if (!svc_handle && (GetLastError() == ERROR_ACCESS_DENIED))
+        {
+            skip("Not enough rights to open the service\n");
+            CloseServiceHandle(scm_handle);        
+            return;
+        }
+        ok(svc_handle != NULL, "Could not open the service : %d\n", GetLastError());
+    }
+    else if (!svc_handle && (GetLastError() == ERROR_ACCESS_DENIED))
+    {
+        skip("Not enough rights to create the service\n");
+        CloseServiceHandle(scm_handle);        
+        return;
+    }
+    else
+        ok(svc_handle != NULL, "Could not create the service : %d\n", GetLastError());
+
+    if (!svc_handle) return;
+
+    /* TODO:
+     * Before we do a QueryServiceConfig we should check the registry. This will make sure
+     * that the correct keys are used.
+     */
+
+    /* Request the size for the buffer */
+    SetLastError(0xdeadbeef);
+    ret = QueryServiceConfigA(svc_handle, NULL, 0, &needed);
+    ok(!ret, "Expected failure\n");
+    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
+
+    config = HeapAlloc(GetProcessHeap(), 0, needed);
+    given = needed;
+    SetLastError(0xdeadbeef);
+    ret = QueryServiceConfigA(svc_handle, config, given, &needed);
+    ok(ret, "Expected success\n");
+    todo_wine
+    {
+    ok(GetLastError() == ERROR_SUCCESS    /* W2K3 */||
+       GetLastError() == 0xdeadbeef       /* NT4, XP, Vista */ ||
+       GetLastError() == ERROR_IO_PENDING /* W2K */,
+        "Expected ERROR_SUCCESS, ERROR_IO_PENDING or 0xdeadbeef, got %d\n", GetLastError());
+    ok(given == needed, "Expected the given (%d) and needed (%d) buffersizes to be equal\n", given, needed);
+    }
+    ok(config->dwServiceType == (SERVICE_INTERACTIVE_PROCESS | SERVICE_WIN32_OWN_PROCESS),
+        "Expected SERVICE_INTERACTIVE_PROCESS | SERVICE_WIN32_OWN_PROCESS, got %d\n", config->dwServiceType);
+    ok(config->dwStartType == SERVICE_DISABLED, "Expected SERVICE_DISABLED, got %d\n", config->dwStartType);
+    ok(config->dwErrorControl == SERVICE_ERROR_IGNORE, "Expected SERVICE_ERROR_IGNORE, got %d\n", config->dwErrorControl);
+    ok(!strcmp(config->lpBinaryPathName, pathname), "Expected '%s', got '%s'\n", pathname, config->lpBinaryPathName);
+    ok(!strcmp(config->lpLoadOrderGroup, empty), "Expected an empty string, got '%s'\n", config->lpLoadOrderGroup);
+    ok(config->dwTagId == 0, "Expected 0, got %d\n", config->dwTagId);
+    /* TODO: Show the double 0 terminated string */
+    todo_wine
+    ok(!memcmp(config->lpDependencies, dependencies, sizeof(dependencies)), "Wrong string\n");
+    if(config->lpServiceStartName) /* FIXME: Wine workaround, remove when fixed */
+        ok(!strcmp(config->lpServiceStartName, localsystem), "Expected 'LocalSystem', got '%s'\n", config->lpServiceStartName);
+    if(config->lpDisplayName) /* FIXME: Wine workaround, remove when fixed */
+        ok(!strcmp(config->lpDisplayName, displayname), "Expected '%s', got '%s'\n", displayname, config->lpDisplayName);
+    
+    SetLastError(0xdeadbeef);
+    ret = DeleteService(svc_handle);
+    ok(ret, "Expected success\n");
+    ok(GetLastError() == ERROR_SUCCESS    /* W2K3 */||
+       GetLastError() == 0xdeadbeef       /* NT4, XP, Vista */ ||
+       GetLastError() == ERROR_IO_PENDING /* W2K */,
+        "Expected ERROR_SUCCESS, ERROR_IO_PENDING or 0xdeadbeef, got %d\n", GetLastError());
+    
+    CloseServiceHandle(svc_handle);
+    CloseServiceHandle(scm_handle);
+}
+
+START_TEST(service)
+{
+    /* Test the creation, querying and deletion of a service */
+    test_sequence();
+}