fix(type): Fixes the getType method of CtVariableAccess

INRIA · Mar 3, 2016 · bf98b41 · bf98b41
1 parent 8428851
commit bf98b41
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/main/java/spoon/support/reflect/code/CtFieldAccessImpl.java b/src/main/java/spoon/support/reflect/code/CtFieldAccessImpl.java
@@ -21,6 +21,7 @@
 import spoon.reflect.code.CtTargetedExpression;
 import spoon.reflect.code.CtVariableAccess;
 import spoon.reflect.reference.CtFieldReference;
+import spoon.reflect.reference.CtTypeReference;
 import spoon.reflect.reference.CtVariableReference;
 
 public abstract class CtFieldAccessImpl<T> extends CtVariableReadImpl<T> implements CtFieldAccess<T> {
@@ -52,4 +53,9 @@ public CtFieldReference<T> getVariable() {
 	public <C extends CtVariableAccess<T>> C setVariable(CtVariableReference<T> variable) {
 		return super.setVariable(variable);
 	}
+
+	@Override
+	public CtTypeReference<T> getType() {
+		return getVariable() == null ? null : getVariable().getType();
+	}
 }
diff --git a/src/main/java/spoon/support/reflect/code/CtVariableAccessImpl.java b/src/main/java/spoon/support/reflect/code/CtVariableAccessImpl.java
@@ -17,6 +17,7 @@
 package spoon.support.reflect.code;
 
 import spoon.reflect.code.CtVariableAccess;
+import spoon.reflect.reference.CtTypeReference;
 import spoon.reflect.reference.CtVariableReference;
 
 public abstract class CtVariableAccessImpl<T> extends CtExpressionImpl<T> implements CtVariableAccess<T> {