Groovy AstTransformationSupport Test

Hi,

I'm writing a plugin to support a custom Groovy AST transformation.

The plugin itself is working as expected but I have problem writing up tests.

I set up a base class for my tests which adds the Groovy library: https://gist.github.com/felixscheinost/7636813a531df226851980e491166cb8

Just to test whether my setup works I copied a spec from the official Groovy plugin (from GrStubAstSwitchTest) where @Delegate is tested, which has similarities with custom annotation:

GroovyFileImpl file = myFixture.addFileToProject('a.groovy', '''\
class A {
  def foo(){}
}
class B {
  @Delegate
  A a = new A()
}
''') as GroovyFileImpl

assert !file.contentsLoaded
PsiClass clazzB = file.classes[1]
assert !file.contentsLoaded

assert clazzB.methods.find { it.name == 'foo' }
assert !file.contentsLoaded

But not even that one passes. It seems like TransformationUtil#transformDefinition isn't called in time. Is there some way to trigger AstTransformationSupport and wait for it to finish?

Is someting with my setup wrong?

Thanks for your help!

Felix