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)); }