blob: aa02d5f3d90fa02c53e0565b46c632a6fd180d34 (
plain)
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
|
// 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.
namespace WixTestTools
{
using System;
using System.Globalization;
using System.Linq;
using System.Management;
using System.Collections.Generic;
using System.Security.Principal;
using WixInternal.TestSupport.XunitExtensions;
public class RuntimePrereqFeatureFactAttribute : RuntimeFactAttribute
{
public static HashSet<string> OptionalFeatures = new(StringComparer.OrdinalIgnoreCase);
public static HashSet<string> ServerFeatures = new(StringComparer.OrdinalIgnoreCase);
static RuntimePrereqFeatureFactAttribute()
{
AddFeaturesToSet(ServerFeatures, "Win32_ServerFeature");
AddFeaturesToSet(OptionalFeatures, "Win32_OptionalFeature");
}
private static void AddFeaturesToSet(HashSet<string> featureSet, string featureSetName)
{
try
{
var objMC = new ManagementClass(featureSetName);
var objMOC = objMC?.GetInstances();
if (objMOC is not null)
{
foreach (var objMO in objMOC)
{
string featureName = (string)objMO.Properties["Name"].Value;
if ((uint)objMO.Properties["InstallState"].Value == 1)
{
featureSet.Add(featureName);
}
}
}
}
catch
{
}
}
public RuntimePrereqFeatureFactAttribute(params string[] prerequisiteFeatures) : base()
{
var missingRequirements = prerequisiteFeatures.Select(x => x).Where(x => !ServerFeatures.Contains(x) && !OptionalFeatures.Contains(x));
if (missingRequirements.Any())
{
this.Skip = "This test is missing the following Feature pre-requisites: " + String.Join(", ", missingRequirements);
}
}
}
}
|
