public void TestAddMissingAttribute()
{
var project = DiagnosticVerifier.CreateProject(new[] { OutdatedAnnotationProgram });
var newSolution = MissingAttributeAdder.AddMissingAttributes(project.Solution).Result;
var document = newSolution.Projects.SelectMany(newProject => newProject.Documents).First();
var documentText = CodeFixVerifier.GetStringFromDocument(document);
Assert.AreEqual(FixedProgram.Replace("\r\n", "\n"), documentText.Replace("\r\n", "\n"));
}
public void TestUpdate()
{
var project = DiagnosticVerifier.CreateProject(new[] { OutdatedAnnotationProgram, OutdatedAnnotationProgram, ProgramWithUnnecessaryAnnotation });
var newSolution = SolutionAttributeUpdater.UpdateAttributes(project.Solution).Result;
var document = newSolution.Projects.SelectMany(newProject => newProject.Documents).First();
Assert.AreEqual(FixedProgram, CodeFixVerifier.GetStringFromDocument(document));
var document2 = newSolution.Projects.SelectMany(newProject => newProject.Documents).Skip(1).First();
Assert.AreEqual(FixedProgram, CodeFixVerifier.GetStringFromDocument(document2));
var document3 = newSolution.Projects.SelectMany(newProject => newProject.Documents).Skip(2).First();
Assert.AreEqual(ProgramWithoutAnnotationNeeded, CodeFixVerifier.GetStringFromDocument(document3));
}
