Package org.apache.groovy.contracts.generation

Class AssertStatementCreationUtility.AddResultReturnStatementVisitor

java.lang.Object

org.codehaus.groovy.ast.CodeVisitorSupport

org.codehaus.groovy.ast.ClassCodeVisitorSupport

org.apache.groovy.contracts.generation.AssertStatementCreationUtility.AddResultReturnStatementVisitor

All Implemented Interfaces:

GroovyClassVisitor, GroovyCodeVisitor, ErrorCollecting

Enclosing class:

AssertStatementCreationUtility

public static class AssertStatementCreationUtility.AddResultReturnStatementVisitor
extends ClassCodeVisitorSupport

Replaces a given ReturnStatement with the appropriate assertion call statement and returns a result variable expression.

Constructor Summary

Constructors

Constructor	Description
AddResultReturnStatementVisitor(ReturnStatement returnStatement, ClassNode returnType, BlockStatement assertionCallBlock)	Creates a visitor that rewrites one return statement to expose the result variable.

Method Summary

Modifier and Type	Method	Description
protected SourceUnit	getSourceUnit()	This visitor is source-independent.
void	visitBlockStatement(BlockStatement block)	Rewrites the block containing the target return statement.

Methods inherited from class org.codehaus.groovy.ast.ClassCodeVisitorSupport

addError, visitAnnotation, visitAnnotations, visitAnnotations, visitAssertStatement, visitBreakStatement, visitCaseStatement, visitCatchStatement, visitClass, visitClassCodeContainer, visitClosureExpression, visitConstructor, visitConstructorOrMethod, visitContinueStatement, visitDeclarationExpression, visitDoWhileLoop, visitExpressionStatement, visitField, visitForLoop, visitIfElse, visitImports, visitMethod, visitObjectInitializerStatements, visitPackage, visitProperty, visitReturnStatement, visitStatement, visitStatementAnnotations, visitSwitch, visitSynchronizedStatement, visitThrowStatement, visitTryCatchFinally, visitWhileLoop

Methods inherited from class org.codehaus.groovy.ast.CodeVisitorSupport

afterSwitchCaseStatementsVisited, afterSwitchConditionExpressionVisited, visitArgumentlistExpression, visitArrayExpression, visitAttributeExpression, visitBinaryExpression, visitBitwiseNegationExpression, visitBooleanExpression, visitBytecodeExpression, visitCastExpression, visitClassExpression, visitClosureListExpression, visitConstantExpression, visitConstructorCallExpression, visitEmptyStatement, visitFieldExpression, visitGStringExpression, visitLambdaExpression, visitListExpression, visitMapEntryExpression, visitMapExpression, visitMethodCallExpression, visitMethodPointerExpression, visitMethodReferenceExpression, visitNotExpression, visitPostfixExpression, visitPrefixExpression, visitPropertyExpression, visitRangeExpression, visitShortTernaryExpression, visitSpreadExpression, visitSpreadMapExpression, visitStaticMethodCallExpression, visitTernaryExpression, visitTupleExpression, visitUnaryMinusExpression, visitUnaryPlusExpression, visitVariableExpression

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.codehaus.groovy.ast.GroovyCodeVisitor

visit, visit, visitEmptyExpression, visitListOfExpressions

Constructor Details

AddResultReturnStatementVisitor

public AddResultReturnStatementVisitor(ReturnStatement returnStatement,
 ClassNode returnType,
 BlockStatement assertionCallBlock)

Creates a visitor that rewrites one return statement to expose the result variable.

Parameters:

returnStatement - the return statement to replace

returnType - the declared return type

assertionCallBlock - the assertion block to inject before returning

Method Details

getSourceUnit

protected SourceUnit getSourceUnit()

This visitor is source-independent.

Specified by:

getSourceUnit in class ClassCodeVisitorSupport

Returns:

null

visitBlockStatement

public void visitBlockStatement(BlockStatement block)

Rewrites the block containing the target return statement.

Specified by:

visitBlockStatement in interface GroovyCodeVisitor

Overrides:

visitBlockStatement in class ClassCodeVisitorSupport

Parameters:

block - the block being visited

See Also:

• CodeVisitorSupport.visitForLoop(ForStatement)
• CodeVisitorSupport.visitWhileLoop(WhileStatement)