1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
#include "precomp.h"
using namespace System;
using namespace System::Collections;
using namespace Xunit;
using namespace WixInternal::TestSupport;
using namespace WixInternal::TestSupport::XunitExtensions;
namespace DutilTests
{
public ref class EnvUtil
{
public:
[Fact]
void EnvExpandEnvironmentStringsTest()
{
HRESULT hr = S_OK;
LPWSTR sczExpanded = NULL;
SIZE_T cchExpanded = 0;
LPCWSTR wzSimpleString = L"%USERPROFILE%";
LPCWSTR wzMultipleString = L"%TEMP%;%PATH%";
LPCWSTR wzLongMultipleString = L"%TEMP%;%PATH%;C:\\abcdefghijklomnopqrstuvwxyz0123456789\\abcdefghijklomnopqrstuvwxyz0123456789\\abcdefghijklomnopqrstuvwxyz0123456789\\abcdefghijklomnopqrstuvwxyz0123456789\\abcdefghijklomnopqrstuvwxyz0123456789";
String^ expandedSimpleString = Environment::ExpandEnvironmentVariables(gcnew String(wzSimpleString));
String^ expandedMultipleString = Environment::ExpandEnvironmentVariables(gcnew String(wzMultipleString));
String^ expandedLongMultipleString = Environment::ExpandEnvironmentVariables(gcnew String(wzLongMultipleString));
try
{
hr = EnvExpandEnvironmentStrings(wzSimpleString, &sczExpanded, &cchExpanded);
NativeAssert::Succeeded(hr, "Failed to expand simple string.");
WixAssert::StringEqual(expandedSimpleString, gcnew String(sczExpanded), false);
NativeAssert::Equal<SIZE_T>(expandedSimpleString->Length + 1, cchExpanded);
hr = EnvExpandEnvironmentStrings(wzMultipleString, &sczExpanded, &cchExpanded);
NativeAssert::Succeeded(hr, "Failed to expand multiple string.");
WixAssert::StringEqual(expandedMultipleString, gcnew String(sczExpanded), false);
NativeAssert::Equal<SIZE_T>(expandedMultipleString->Length + 1, cchExpanded);
hr = EnvExpandEnvironmentStrings(wzLongMultipleString, &sczExpanded, &cchExpanded);
NativeAssert::Succeeded(hr, "Failed to expand long multiple string.");
WixAssert::StringEqual(expandedLongMultipleString, gcnew String(sczExpanded), false);
NativeAssert::Equal<SIZE_T>(expandedLongMultipleString->Length + 1, cchExpanded);
}
finally
{
ReleaseStr(sczExpanded);
}
}
[SkippableFact]
void EnvExpandEnvironmentStringsForUserTest()
{
HRESULT hr = S_OK;
LPWSTR sczExpanded = NULL;
SIZE_T cchExpanded = 0;
String^ variableName = nullptr;
String^ variableValue = nullptr;
// Find a system environment variable that doesn't have variables in its value;
for each (DictionaryEntry^ entry in Environment::GetEnvironmentVariables(EnvironmentVariableTarget::Machine))
{
variableValue = (String^)entry->Value;
if (variableValue->Contains("%"))
{
continue;
}
variableName = (String^)entry->Key;
break;
}
if (nullptr == variableName)
{
WixAssert::Skip("No suitable system environment variables");
}
pin_ptr<const wchar_t> wzUnexpanded = PtrToStringChars("%" + variableName + "%_%USERNAME%");
String^ expandedValue = variableValue + "_SYSTEM";
try
{
hr = EnvExpandEnvironmentStringsForUser(NULL, wzUnexpanded, &sczExpanded, &cchExpanded);
NativeAssert::Succeeded(hr, "Failed to expand %ls.", wzUnexpanded);
WixAssert::StringEqual(expandedValue, gcnew String(sczExpanded), false);
NativeAssert::Equal<SIZE_T>(expandedValue->Length + 1, cchExpanded);
}
finally
{
ReleaseStr(sczExpanded);
}
}
};
}
|
