public void PopulateFromSolution()
{
TestSuiteCollector collector = new TestSuiteCollector();
foreach (HierarchyItem projectHier in
VsShellUtils.GetLoadedProjects(solution))
{
Project project = projectHier.GetExtObjectAs <Project>();
collector.Process(project);
}
suites = collector.Suites;
possibleTestFiles = collector.PossibleTestFiles;
mainExists = collector.MainFunctionExists;
}
public void AddElement(CodeElement element)
{
if (testsTreeView.Nodes.Count == 0)
return;
if (element is VCCodeClass || element is VCCodeStruct)
{
TestSuiteCollector collector = new TestSuiteCollector();
collector.Process(element);
foreach (TestSuite suite in collector.Suites)
AddTestSuiteToNode(suite, testsTreeView.Nodes[0], null);
testsTreeView.Sort();
}
else if (element is VCCodeFunction)
{
VCCodeFunction function = (VCCodeFunction)element;
TreeNode suiteNode = FindNodeForTestSuite(function.Parent);
TestCaseCollector collector = new TestCaseCollector();
collector.Process((CodeElement)function);
foreach (TestCase testCase in collector.TestCases)
AddTestCaseToNode((TestSuite)suiteNode.Tag, testCase, suiteNode, null);
testsTreeView.Sort();
}
}
public void ChangeElement(CodeElement element, vsCMChangeKind changeKind)
{
if (testsTreeView.Nodes.Count == 0)
return;
if (element is VCCodeClass || element is VCCodeStruct)
{
testsTreeView.BeginUpdate();
VCCodeModel model = ((VCCodeElement)element).CodeModel;
TestSuiteCollector collector = new TestSuiteCollector();
collector.Process(model);
List<TestSuite> suitesInModel = new List<TestSuite>(collector.Suites);
// Iterate through the current suites in the view and pull out any
// that are no longer in the model.
TreeNode solutionNode = testsTreeView.Nodes[0];
for (int i = 0; i < solutionNode.Nodes.Count; )
{
TreeNode viewNode = solutionNode.Nodes[i];
int index = suitesInModel.FindIndex(
new Predicate<TestSuite>(delegate(TestSuite suite)
{
return suite.Name == viewNode.Text;
}));
if (index == -1)
viewNode.Remove();
else
{
// The suites that are left over will be those in the
// model that aren't yet in the tree, so they need to
// be added.
suitesInModel.RemoveAt(index);
i++;
}
}
// Make sure a suite with the same name isn't already in the
// view (this could happen if the name changes and then quickly
// changes back before the event fires. If it's not there, add
// it.
foreach(TestSuite suite in suitesInModel)
AddTestSuiteToNode(suite, solutionNode, null).Expand();
testsTreeView.Sort();
testsTreeView.EndUpdate();
}
else if (element is VCCodeFunction)
{
testsTreeView.BeginUpdate();
VCCodeElement parent = (VCCodeElement)((VCCodeElement)element).Parent;
TestCaseCollector collector = new TestCaseCollector();
collector.Process(parent.Children);
List<TestCase> casesInModel = new List<TestCase>(collector.TestCases);
// Iterate through the current test cases for this suite in the
// view and pull out any that are no longer in the model.
TreeNode suiteNode = FindNodeForTestSuite(parent);
if (suiteNode == null)
return;
for (int i = 0; i < suiteNode.Nodes.Count; )
{
TreeNode viewNode = suiteNode.Nodes[i];
int index = casesInModel.FindIndex(
new Predicate<TestCase>(delegate(TestCase testCase)
{
return testCase.Name == viewNode.Text;
}));
if (index == -1)
viewNode.Remove();
else
{
// The test cases that are left over will be those in
// the model that aren't yet in the tree, so they need
// to be added.
casesInModel.RemoveAt(index);
i++;
}
}
foreach (TestCase testCase in casesInModel)
AddTestCaseToNode((TestSuite)suiteNode.Tag, testCase, suiteNode, null);
testsTreeView.Sort();
testsTreeView.EndUpdate();
}
}
public void PopulateFromSolution()
{
TestSuiteCollector collector = new TestSuiteCollector();
foreach (HierarchyItem projectHier in
VsShellUtils.GetLoadedProjects(solution))
{
Project project = projectHier.GetExtObjectAs<Project>();
collector.Process(project);
}
suites = collector.Suites;
possibleTestFiles = collector.PossibleTestFiles;
mainExists = collector.MainFunctionExists;
}
