1 files changed, 143 insertions, 0 deletions
diff --git a/src/pal/tests/palsuite/miscellaneous/SetEnvironmentVariableW/test3/test3.cpp b/src/pal/tests/palsuite/miscellaneous/SetEnvironmentVariableW/test3/test3.cpp
new file mode 100644
index 0000000000..5c4d4eba42
--- /dev/null
+++ b/src/pal/tests/palsuite/miscellaneous/SetEnvironmentVariableW/test3/test3.cpp
@@ -0,0 +1,143 @@
+// 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 : test3.c
+**
+** Purpose: Test for SetEnvironmentVariableW() function
+**          Create environment variables that differ only
+**          in case and verify that they return the appropriate
+**          value on the BSD environment.
+**
+**
+===========================================================*/
+
+#include <palsuite.h>
+
+int __cdecl main(int argc, char *argv[]) 
+{
+
+#if WIN32
+
+    return PASS;
+
+#else
+
+    /* Define some buffers needed for the function */
+    WCHAR * pResultBuffer = NULL;
+
+    WCHAR FirstEnvironmentVariable[] = {'P','A','L','T','E','S','T','\0'};
+    WCHAR FirstEnvironmentValue[] = {'F','I','R','S','T','\0'};
+
+    WCHAR SecondEnvironmentVariable[] = {'p','a','l','t','e','s','t','\0'};
+    WCHAR SecondEnvironmentValue[] = {'S','E','C','O','N','D','\0'};
+
+    DWORD size = 0;
+    BOOL bRc = TRUE;
+
+    /*
+     * Initialize the PAL and return FAILURE if this fails
+     */
+
+    if(0 != (PAL_Initialize(argc, argv)))
+    {
+        return FAIL;
+    }
+  
+    /* Set the first environment variable */
+    bRc = SetEnvironmentVariableW(FirstEnvironmentVariable,
+                            FirstEnvironmentValue);
+
+    if(!bRc)
+    {
+        Fail("ERROR: SetEnvironmentVariable failed to set a "
+            "proper environment variable with error %u.\n",
+            GetLastError());
+    }
+
+     /* Set the second environment Variable */
+    bRc = SetEnvironmentVariableW(SecondEnvironmentVariable,
+                            SecondEnvironmentValue);
+
+    if(!bRc)
+    {
+        Fail("ERROR: SetEnvironmentVariable failed to set a "
+            "proper environment variable with error %u.\n",
+            GetLastError());
+    }
+
+
+   /* Normal case, PATH should fit into this buffer */
+    size = GetEnvironmentVariableW(FirstEnvironmentVariable,        
+                                  pResultBuffer,    
+                                  0);  
+
+    /* Increase size to account for the null char at the end */
+    size = size + 1;
+    
+    pResultBuffer = (WCHAR*)malloc(sizeof(WCHAR)*size);
+    if ( pResultBuffer == NULL )
+    {
+        Fail("ERROR: Failed to allocate memory for pResultBuffer pointer.\n");
+    }
+
+    /* Try to retrieve the value of the first environment variable */
+    GetEnvironmentVariableW(FirstEnvironmentVariable,
+                           pResultBuffer,
+                           size);
+
+    if ( pResultBuffer == NULL )
+    {
+        Fail("ERROR: GetEnvironmentVariable failed to return a value "
+            "from a proper environment variable with error %u.\n",
+            GetLastError());
+    }
+
+    /* Compare the strings to see that the correct variable was returned */
+    if(wcscmp(pResultBuffer,FirstEnvironmentValue) != 0)
+    {
+        Trace("ERROR: The value in the buffer should have been '%S' but "
+             "was really '%S'.\n",FirstEnvironmentValue, pResultBuffer);          
+        free(pResultBuffer);    
+        Fail("");
+    }
+
+    free(pResultBuffer);
+
+    /* Reallocate the memory for the string */
+    pResultBuffer = (WCHAR*)malloc(sizeof(WCHAR)*size);
+    if ( pResultBuffer == NULL )
+    {
+        Fail("ERROR: Failed to allocate memory for pResultBuffer pointer.\n");
+    }
+
+    /* Try retrieving the value of the first variable, even though the
+    second variable has the same spelling and only differs in case */
+    GetEnvironmentVariableW(SecondEnvironmentVariable,
+                           pResultBuffer,
+                           size);
+
+    if ( pResultBuffer == NULL )
+    {
+        Fail("ERROR: GetEnvironmentVariable failed to return a value "
+            "from a proper environment variable with error %u.\n",
+            GetLastError());
+    }
+
+    /* Compare the two strings to confirm that the right value is returned */
+    if(wcscmp(pResultBuffer,SecondEnvironmentValue) != 0) 
+    {
+        Trace("ERROR: The value in the buffer should have been '%S' but "
+             "was really '%S'.\n",SecondEnvironmentValue,pResultBuffer);          
+        free(pResultBuffer);    
+        Fail("");
+    }
+  
+    free(pResultBuffer);
+    
+    PAL_Terminate();
+    return PASS;
+
+#endif
+}