private string GenerateError(TestGeneratorResult generationResult, CodeDomHelper codeDomHelper)
{
var errorsArray = generationResult.Errors.ToArray();
return(string.Join(Environment.NewLine, errorsArray.Select(e => codeDomHelper.GetErrorStatementString(e.Message)).ToArray()));
}
public TestFileGeneratorResult(TestGeneratorResult generatorResult, string fileName)
{
if (generatorResult == null)
{
throw new ArgumentNullException(nameof(generatorResult));
}
Filename = fileName ?? throw new ArgumentNullException(nameof(fileName));
Errors = generatorResult.Errors;
IsUpToDate = generatorResult.IsUpToDate;
GeneratedTestCode = generatorResult.GeneratedTestCode;
}
protected override string GenerateCode(string inputFileContent)
{
var generatorServices = new VsGeneratorServices(CurrentProject);
using (var testGenerator = generatorServices.CreateTestGenerator())
{
string projectFolder = VsxHelper.GetProjectFolder(CurrentProject);
var fullPath = Path.GetFullPath(Path.Combine(projectFolder, CodeFilePath));
FeatureFileInput featureFileInput =
new FeatureFileInput(FileSystemHelper.GetRelativePath(fullPath, projectFolder))
{
FeatureFileContent = inputFileContent
};
generationResult = testGenerator.GenerateTestFile(featureFileInput, new GenerationSettings());
if (!generationResult.Success)
{
return(null);
}
return(generationResult.GeneratedTestCode);
}
}
private string Generate(string inputFilePath, string inputFileContent, GeneratorServices generatorServices, CodeDomHelper codeDomHelper)
{
string outputFileContent;
try
{
TestGeneratorResult generationResult = GenerateCode(inputFilePath, inputFileContent, generatorServices);
if (generationResult.Success)
{
outputFileContent = generationResult.GeneratedTestCode;
}
else
{
outputFileContent = GenerateError(generationResult, codeDomHelper);
}
}
catch (Exception ex)
{
outputFileContent = GenerateError(ex, codeDomHelper);
}
return(outputFileContent);
}
protected override string GenerateCode(string inputFileContent)
{
IVisualStudioTracer tracer = VsxHelper.ResolveMefDependency <IVisualStudioTracer>(ServiceProvider.GlobalProvider);
var generatorServices = new VsGeneratorServices(CurrentProject, new VsSpecFlowConfigurationReader(CurrentProject, tracer), tracer);
using (var testGenerator = generatorServices.CreateTestGenerator())
{
string projectFolder = VsxHelper.GetProjectFolder(CurrentProject);
var fullPath = Path.GetFullPath(Path.Combine(projectFolder, CodeFilePath));
FeatureFileInput featureFileInput =
new FeatureFileInput(FileSystemHelper.GetRelativePath(fullPath, projectFolder))
{
FeatureFileContent = inputFileContent
};
generationResult = testGenerator.GenerateTestFile(featureFileInput, new GenerationSettings());
if (!generationResult.Success)
{
return(null);
}
return(generationResult.GeneratedTestCode);
}
}
