4e2f57a35599c659db68bb17302423f0aa1bf90e
[idea/community.git] / plugins / groovy / test / org / jetbrains / plugins / groovy / transformations / BaseScriptTransformationSupportTest.groovy
1 /*
2  * Copyright 2000-2016 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.jetbrains.plugins.groovy.transformations
17
18 import com.intellij.psi.PsiClass
19 import com.intellij.psi.util.InheritanceUtil
20 import com.intellij.testFramework.LightProjectDescriptor
21 import groovy.transform.CompileStatic
22 import org.jetbrains.plugins.groovy.GroovyLightProjectDescriptor
23 import org.jetbrains.plugins.groovy.LightGroovyTestCase
24 import org.jetbrains.plugins.groovy.lang.psi.impl.GroovyFileImpl
25 import org.jetbrains.plugins.groovy.lang.psi.impl.synthetic.GroovyScriptClass
26
27 @CompileStatic
28 class BaseScriptTransformationSupportTest extends LightGroovyTestCase {
29
30   LightProjectDescriptor projectDescriptor = GroovyLightProjectDescriptor.GROOVY_LATEST
31
32   @Override
33   void setUp() throws Exception {
34     super.setUp()
35     fixture.addFileToProject 'base.groovy', 'abstract class MyBaseScript extends Script {}'
36   }
37
38   private void doTest(String text) {
39     def file = fixture.addFileToProject('Zzz.groovy', """\
40 import groovy.transform.BaseScript
41
42 $text
43 """) as GroovyFileImpl
44     def clazz = fixture.findClass('Zzz')
45     assert clazz instanceof GroovyScriptClass
46     assert InheritanceUtil.isInheritor(clazz as PsiClass, 'MyBaseScript')
47     assert !file.contentsLoaded
48   }
49
50   void 'test top level'() {
51     doTest '@BaseScript MyBaseScript hello'
52   }
53
54   void 'test script block level'() {
55     doTest 'if (true) @BaseScript MyBaseScript hello'
56   }
57
58   void 'test within method'() {
59     doTest '''\
60 def foo() {
61   @BaseScript MyBaseScript hello  
62 }
63 '''
64   }
65 }