private void SetupTestClassCleanupMethod(TestClassGenerationContext generationContext)
{
var testClassCleanupMethod = generationContext.TestClassCleanupMethod;
testClassCleanupMethod.ReturnType = new CodeTypeReference(typeof(Task));
testClassCleanupMethod.Attributes = MemberAttributes.Public;
testClassCleanupMethod.Name = GeneratorConstants.TESTCLASS_CLEANUP_NAME;
_testGeneratorProvider.MarkCodeMemberMethodAsAsync(testClassCleanupMethod);
_testGeneratorProvider.SetTestClassCleanupMethod(generationContext);
var testRunnerField = _scenarioPartHelper.GetTestRunnerExpression();
// await testRunner.OnFeatureEndAsync();
var expression = new CodeMethodInvokeExpression(
testRunnerField,
nameof(ITestRunner.OnFeatureEndAsync));
_codeDomHelper.MarkCodeMethodInvokeExpressionAsAwait(expression);
testClassCleanupMethod.Statements.Add(expression);
// testRunner = null;
testClassCleanupMethod.Statements.Add(
new CodeAssignStatement(
testRunnerField,
new CodePrimitiveExpression(null)));
}
public void SetupTestClassCleanupMethod()
{
var classCleanupMethod = GenerationContext.TestClassCleanupMethod;
classCleanupMethod.Attributes = MemberAttributes.Public;
classCleanupMethod.Name = "FeatureTearDown";
_testGeneratorProvider.SetTestClassCleanupMethod(GenerationContext);
var runnerExpression = GetTestRunnerExpression();
classCleanupMethod.Statements.Add(new CodeMethodInvokeExpression(runnerExpression, "OnFeatureEnd", new CodeExpression[0]));
classCleanupMethod.Statements.Add(new CodeAssignStatement(runnerExpression, new CodePrimitiveExpression(null)));
}
private void SetupTestClassCleanupMethod(TestClassGenerationContext generationContext)
{
CodeMemberMethod testClassCleanupMethod = generationContext.TestClassCleanupMethod;
testClassCleanupMethod.Attributes = MemberAttributes.Public;
testClassCleanupMethod.Name = TESTCLASS_CLEANUP_NAME;
testGeneratorProvider.SetTestClassCleanupMethod(generationContext);
var testRunnerField = GetTestRunnerExpression();
// testRunner.OnFeatureEnd();
testClassCleanupMethod.Statements.Add(
new CodeMethodInvokeExpression(
testRunnerField,
"OnFeatureEnd"));
// testRunner = null;
testClassCleanupMethod.Statements.Add(
new CodeAssignStatement(
testRunnerField,
new CodePrimitiveExpression(null)));
}
public void SetTestClassCleanupMethod(TestClassGenerationContext generationContext)
{
_unitTestGeneratorProvider.SetTestClassCleanupMethod(generationContext);
UpdateAttributes(generationContext.TestClassCleanupMethod);
}
public void SetTestClassCleanupMethod(TestClassGenerationContext generationContext)
{
baseGeneratorProvider.SetTestClassCleanupMethod(generationContext);
}
public void SetTestClassCleanupMethod(TestClassGenerationContext generationContext)
{
_innerGenerator.SetTestClassCleanupMethod(generationContext);
}
