private static WmiQueryTestArgs AddWmiPropertiesToWrangler_TestArgs3()
{
string expectedresult = "Test Model,Test_VM";
//protected void AddWmiPropertiesToWrangler(ResultWrangler Wrangler, IEnumerable<ManagementObject> WmiObjectList, List<KeyValuePair<string, XElement>> PropertyTemplates)
List <KeyValuePair <string, XElement> > proptemplates = new List <KeyValuePair <string, XElement> >();
XElement propx = new XElement("Property");
propx.Add(new XAttribute("Name", "Model"));
proptemplates.Add(new KeyValuePair <string, XElement>("Model", propx));
propx = new XElement("Property");
propx.Add(new XAttribute("Name", "Serial"));
proptemplates.Add(new KeyValuePair <string, XElement>("Serial", propx));
ResultWrangler wrangler = new ResultWrangler();
wrangler.Separator = ",";
ManagementClass wmi1 = new ManagementClass();
wmi1.Properties.Add("Model", "Test Model", CimType.String);
wmi1.Properties.Add("Serial", "Test_VM", CimType.String);
List <ManagementObject> objcollection = new List <ManagementObject>();
objcollection.Add(wmi1);
return(new WmiQueryTestArgs(expectedresult, wrangler, objcollection, proptemplates));
}
private static WmiQueryTestArgs AddWmiPropertiesToWrangler_TestArgs2(bool IncludeFalseValues)
{
string expectedresult;
if (IncludeFalseValues == true)
{
expectedresult = "1,";
}
else
{
expectedresult = "1";
}
//protected void AddWmiPropertiesToWrangler(ResultWrangler Wrangler, IEnumerable<ManagementObject> WmiObjectList, List<KeyValuePair<string, XElement>> PropertyTemplates)
List <KeyValuePair <string, XElement> > proptemplates = new List <KeyValuePair <string, XElement> >();
ResultWrangler wrangler = new ResultWrangler();
wrangler.IncludeNullValues = IncludeFalseValues;
wrangler.Separator = ",";
ManagementClass batt1 = new ManagementClass();
ManagementClass batt2 = new ManagementClass();
batt1.Properties.Add("BatteryStatus", 1, CimType.UInt16);
batt2.Properties.Add("BatteryStatus", null, CimType.UInt16);
List <ManagementObject> objcollection = new List <ManagementObject>();
objcollection.Add(batt1);
objcollection.Add(batt2);
return(new WmiQueryTestArgs(expectedresult, wrangler, objcollection, proptemplates));
}
public WmiQueryTestArgs(string ExpectedResult, ResultWrangler Wrangler, List <ManagementObject> ManagementObjectList, List <KeyValuePair <string, XElement> > PropertyTemplates)
{
this.ExpectedResult = ExpectedResult;
this.Wrangler = Wrangler;
this.ManagementObjectList = ManagementObjectList;
this.PropertyTemplates = PropertyTemplates;
}
public List <ListItem> Rebuild()
{
LoggerFacade.Debug("ListBuilder rebuild initialised");
int i = 0;
List <ListItem> newlist = new List <ListItem>();
while (i <= this._lastindex)
{
ListItem staticitem;
if (this._staticitems.TryGetValue(i, out staticitem) == true)
{
newlist.Add(staticitem);
i++;
continue;
}
QueryPriorityList qlist;
if (this._querylists.TryGetValue(i, out qlist) == true)
{
qlist.ProcessAllQueries();
ResultWrangler wrangler = qlist.GetResultWrangler();
if (wrangler != null)
{
List <KeyValueTreeNode> kvlist = wrangler.GetKeyValueTree();
foreach (KeyValueTreeNode node in kvlist)
{
//ListItem newoption = new ListItem(node.Value.Key, node.Value.Value, this._parent.ControlFormatting, this._parent, this._director);
newlist.Add(this.CreateItem(node));
}
}
i++;
continue;
}
i++;
}
if (this._parent.Sort)
{
newlist.Sort();
foreach (ListItem item in newlist)
{
item.Sort();
}
}
this.Items = newlist;
LoggerFacade.Debug("ListBuilder rebuild finished");
return(newlist);
}
private void AddPropertiesToWrangler(ResultWrangler wrangler, Principal group, List <KeyValuePair <string, XElement> > PropertyTemplates)
{
wrangler.NewResult();
//if properties have been specified in the xml, query them directly in order
if (PropertyTemplates.Count != 0)
{
foreach (KeyValuePair <string, XElement> template in PropertyTemplates)
{
FormattedProperty prop = new FormattedProperty(template.Value);
prop.Input = PropertyInterogation.GetStringFromPropertyValue(group, template.Key);
wrangler.AddFormattedProperty(prop);
}
}
}
public void AddWmiPropertiesToWrangler_Test(WmiQueryTestArgs TestArgs)
{
//protected void AddWmiPropertiesToWrangler(ResultWrangler Wrangler, IEnumerable<ManagementObject> WmiObjectList, List<KeyValuePair<string, XElement>> PropertyTemplates)
WmiQuery query = new WmiQuery(new DummyLinkTarget());
PrivateObject obj = new PrivateObject(query);
object[] args = new object[3] {
TestArgs.Wrangler, TestArgs.ManagementObjectList, TestArgs.PropertyTemplates
};
obj.Invoke("AddWmiPropertiesToWrangler", args);
ResultWrangler wrangler = TestArgs.Wrangler;
string s = wrangler.GetString();
NUnit.Framework.Assert.AreEqual(s, TestArgs.ExpectedResult);
}
private static WmiQueryTestArgs AddWmiPropertiesToWrangler_TestArgsNull()
{
string expectedresult = null;
//protected void AddWmiPropertiesToWrangler(ResultWrangler Wrangler, IEnumerable<ManagementObject> WmiObjectList, List<KeyValuePair<string, XElement>> PropertyTemplates)
List <KeyValuePair <string, XElement> > proptemplates = new List <KeyValuePair <string, XElement> >();
XElement propx = new XElement("Property");
propx.Add(new XAttribute("Name", "BatteryStatus"));
proptemplates.Add(new KeyValuePair <string, XElement>("BatteryStatus", propx));
ResultWrangler wrangler = new ResultWrangler();
wrangler.Separator = null;
List <ManagementObject> objcollection = new List <ManagementObject>();
return(new WmiQueryTestArgs(expectedresult, wrangler, objcollection, proptemplates));
}
