Jacek Caban : wscript: Added WScript.Quit tests.

[Alexandre Julliard]

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

Author: Jacek Caban 
Date:   Fri Jul  8 12:43:58 2016 +0200

wscript: Added WScript.Quit tests.

Signed-off-by: Jacek Caban 
Signed-off-by: Alexandre Julliard

---

programs/wscript/tests/run.c | 73 +++++++++++++++++++++++++++++---------------
 1 file changed, 49 insertions(+), 24 deletions(-)

diff --git a/programs/wscript/tests/run.c b/programs/wscript/tests/run.c
index e0a6932..eb840cc 100644
--- a/programs/wscript/tests/run.c
+++ b/programs/wscript/tests/run.c
@@ -339,11 +339,12 @@ static const IClassFactoryVtbl ClassFactoryVtbl = {

static IClassFactory testobj_cf = { &ClassFactoryVtbl };

-static void run_test(const char *file_name)
+static void run_script_file(const char *file_name, DWORD expected_exit_code)
 {
     char command[MAX_PATH];
     STARTUPINFOA si = {sizeof(si)};
     PROCESS_INFORMATION pi;
+    DWORD exit_code;
     BOOL bres;

script_name = file_name;
@@ -360,37 +361,29 @@ static void run_test(const char *file_name)

wscript_process = pi.hProcess;
     WaitForSingleObject(pi.hProcess, INFINITE);
+
+    bres = GetExitCodeProcess(pi.hProcess, &exit_code);
+    ok(bres, "GetExitCodeProcess failed: %u\n", GetLastError());
+    ok(exit_code == expected_exit_code, "exit_code = %u, expected %u\n", exit_code, expected_exit_code);
+
     CloseHandle(pi.hThread);
     CloseHandle(pi.hProcess);

CHECK_CALLED(reportSuccess);
 }

-static BOOL WINAPI test_enum_proc(HMODULE module, LPCSTR type, LPSTR name, LONG_PTR param)
+static void run_script(const char *name, const char *script_data, size_t script_size, DWORD expected_exit_code)
 {
-    const char *script_data, *ext;
-    DWORD script_size, size;
     char file_name[MAX_PATH];
+    const char *ext;
     HANDLE file;
-    HRSRC src;
+    DWORD size;
     BOOL res;

-    trace("running %s test...\n", name);
-
-    src = FindResourceA(NULL, name, type);
-    ok(src != NULL, "Could not find resource %s: %u\n", name, GetLastError());
-    if(!src)
-        return TRUE;
-
-    script_data = LoadResource(NULL, src);
-    script_size = SizeofResource(NULL, src);
-    while(script_size && !script_data[script_size-1])
-        script_size--;
-
     ext = strrchr(name, '.');
     ok(ext != NULL, "no script extension\n");
     if(!ext)
-      return TRUE;
+      return;

sprintf(file_name, "test%s", ext);

@@ -398,17 +391,43 @@ static BOOL WINAPI test_enum_proc(HMODULE module, LPCSTR type, LPSTR name, LONG_
             FILE_ATTRIBUTE_NORMAL, NULL);
     ok(file != INVALID_HANDLE_VALUE, "CreateFile failed: %u\n", GetLastError());
     if(file == INVALID_HANDLE_VALUE)
-        return TRUE;
+        return;

res = WriteFile(file, script_data, script_size, &size, NULL);
     CloseHandle(file);
     ok(res, "Could not write to file: %u\n", GetLastError());
     if(!res)
-        return TRUE;
+        return;

-    run_test(file_name);
+    run_script_file(file_name, expected_exit_code);

DeleteFileA(file_name);
+}
+
+static void run_simple_script(const char *script, DWORD expected_exit_code)
+{
+    run_script("simple.js", script, strlen(script), expected_exit_code);
+}
+
+static BOOL WINAPI test_enum_proc(HMODULE module, LPCSTR type, LPSTR name, LONG_PTR param)
+{
+    const char *script_data;
+    DWORD script_size;
+    HRSRC src;
+
+    trace("running %s test...\n", name);
+
+    src = FindResourceA(NULL, name, type);
+    ok(src != NULL, "Could not find resource %s: %u\n", name, GetLastError());
+    if(!src)
+        return TRUE;
+
+    script_data = LoadResource(NULL, src);
+    script_size = SizeofResource(NULL, src);
+    while(script_size && !script_data[script_size-1])
+        script_size--;
+
+    run_script(name, script_data, script_size, 0);
     return TRUE;
 }

@@ -468,11 +487,17 @@ START_TEST(run)
     }

argc = winetest_get_mainargs(&argv);
-    if(argc > 2)
-        run_test(argv[2]);
-    else
+    if(argc > 2) {
+        run_script_file(argv[2], 0);
+    }else {
         EnumResourceNamesA(NULL, "TESTSCRIPT", test_enum_proc, 0);

+        run_simple_script("var winetest = new ActiveXObject('Wine.Test');\n"
+                           "winetest.reportSuccess();\n"
+                           "WScript.Quit(3);\n"
+                           "winetest.ok(false, 'not quit?');\n", 3);
+}
+
     init_registry(FALSE);
     CoUninitialize();
 }