Imported Upstream version 1.1.0upstream/1.1.0

1 files changed, 79 insertions, 0 deletions

diff --git a/src/pal/tests/palsuite/miscellaneous/GetEnvironmentVariableW/test1/test.c b/src/pal/tests/palsuite/miscellaneous/GetEnvironmentVariableW/test1/test.c
new file mode 100644
index 0000000000..6c764509d0
--- /dev/null
+++ b/src/pal/tests/palsuite/miscellaneous/GetEnvironmentVariableW/test1/test.c
@@ -0,0 +1,79 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+/*============================================================
+**
+** Source : test.c
+**
+** Purpose: Test for GetEnvironmentVariable() function
+**
+**
+**=========================================================*/
+
+#define UNICODE
+#include <palsuite.h>
+
+int __cdecl main(int argc, char *argv[]) 
+{
+
+    /* Define some buffers needed for the function */
+    WCHAR * pResultBuffer = NULL;
+    int size = 0;
+  
+    /* A place to stash the returned values */
+    int ReturnValueForLargeBuffer = 0;
+  
+    /*
+     * Initialize the PAL and return FAILURE if this fails
+     */
+
+    if(0 != (PAL_Initialize(argc, argv)))
+    {
+        return FAIL;
+    }
+
+    /* Recieve and allocate the correct amount of memory for the buffer */
+    size = ReturnValueForLargeBuffer = 
+        GetEnvironmentVariable(convert("PATH"),    
+                               pResultBuffer,  
+                               0);
+  
+    pResultBuffer = malloc(size*sizeof(WCHAR));
+    if ( pResultBuffer == NULL )
+    {
+	Fail("ERROR: Failed to allocate memory for pResultBuffer pointer. "
+             "Can't properly exec test case without this.\n");
+    }
+  
+  
+    /* Normal case, PATH should fit into this buffer */
+    ReturnValueForLargeBuffer = GetEnvironmentVariable(convert("PATH"),        
+                                                       pResultBuffer,  
+                                                       size); 
+    free(pResultBuffer); 
+  
+    /* Ensure that it returned a positive value */
+    if(ReturnValueForLargeBuffer <= 0)
+    {
+        Fail("The return was %d, which indicates that the function failed.\n",
+             ReturnValueForLargeBuffer);
+    }
+  
+    /* Ensure that it succeeded and copied the correct number of characters.
+       If this is true, then the return value should be one less of the 
+       size of the buffer.  (Doesn't include that NULL byte)
+    */
+    if(ReturnValueForLargeBuffer != size-1)
+    {
+        Fail("The value returned was %d when it should have been %d.  This "
+             "should be the number of characters copied, "
+	     "minus the NULL byte.\n", ReturnValueForLargeBuffer, size-1);
+    }
+  
+    PAL_Terminate();
+    return PASS;
+}
+
+
+