private static async Task RunTest(string code, string expectedChangedCode)
{
var(diagnostics, document, workspace) = await Utilities.GetDiagnosticsAdvanced(code);
Assert.AreEqual(1, diagnostics.Length);
var diagnostic = diagnostics[0];
var codeFixProvider = new CreationCodeFixProvider();
CodeAction registeredCodeAction = null;
var context = new CodeFixContext(document, diagnostic, (codeAction, _) =>
{
if (registeredCodeAction != null)
{
throw new Exception("Code action was registered more than once");
}
registeredCodeAction = codeAction;
}, CancellationToken.None);
await codeFixProvider.RegisterCodeFixesAsync(context);
if (registeredCodeAction == null)
{
throw new Exception("Code action was not registered");
}
var operations = await registeredCodeAction.GetOperationsAsync(CancellationToken.None);
foreach (var operation in operations)
{
operation.Apply(workspace, CancellationToken.None);
}
var updatedDocument = workspace.CurrentSolution.GetDocument(document.Id);
var newCode = (await updatedDocument.GetTextAsync()).ToString();
Assert.AreEqual(expectedChangedCode, newCode);
}
public async Task CreatingAnImmutableArrayViaTheEmptyPropertyWithoutOpeningTheImmutableNamespace_CodeFixMakesTheCodeUseTheCreateMethod()
{
var code = @"
public static class Program
{
public static void Main()
{
var array = System.Collections.Immutable.ImmutableArray<int>.Empty.Add(1);
}
}";
var expectedChangedCode = @"
public static class Program
{
public static void Main()
{
var array = System.Collections.Immutable.ImmutableArray.Create(1);
}
}";
var(diagnostics, document, workspace) = await Utilities.GetDiagnosticsAdvanced(code);
Assert.AreEqual(1, diagnostics.Length);
var diagnostic = diagnostics[0];
var codeFixProvider = new CreationCodeFixProvider();
CodeAction registeredCodeAction = null;
var context = new CodeFixContext(document, diagnostic, (codeAction, _) =>
{
if (registeredCodeAction != null)
{
throw new Exception("Code action was registered more than once");
}
registeredCodeAction = codeAction;
}, CancellationToken.None);
await codeFixProvider.RegisterCodeFixesAsync(context);
if (registeredCodeAction == null)
{
throw new Exception("Code action was not registered");
}
var operations = await registeredCodeAction.GetOperationsAsync(CancellationToken.None);
foreach (var operation in operations)
{
operation.Apply(workspace, CancellationToken.None);
}
var updatedDocument = workspace.CurrentSolution.GetDocument(document.Id);
var newCode = (await updatedDocument.GetTextAsync()).ToString();
Assert.AreEqual(expectedChangedCode, newCode);
}
