Fix tests for highlighting in scripts with custom script definition

idea/testData/script/definition/highlighting/asyncResolver/template/template.kt

@@ -4,6 +4,7 @@ import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.*
 import java.io.File
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : AsyncDependenciesResolver {
     suspend override fun resolveAsync(scriptContents: ScriptContents, environment: Environment): DependenciesResolver.ResolveResult {
@@ -13,6 +14,7 @@ class TestDependenciesResolver : AsyncDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 

idea/testData/script/definition/highlighting/customBaseClass/template/template.kt

@@ -5,6 +5,7 @@ import kotlin.script.templates.*
 import java.io.File
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : ScriptDependenciesResolver {
     override fun resolve(
@@ -20,6 +21,7 @@ class TestDependenciesResolver : ScriptDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template: Base()
 

idea/testData/script/definition/highlighting/doNotSpeakAboutJava/template/template.kt

@@ -4,6 +4,7 @@ import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.*
 import java.io.File
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(scriptContents: ScriptContents, environment: Environment): DependenciesResolver.ResolveResult {
@@ -41,6 +42,7 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 

idea/testData/script/definition/highlighting/doNotSpeakAboutJavaLegacy/template/template.kt

@@ -5,6 +5,7 @@ import kotlin.script.templates.*
 import java.io.File
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : ScriptDependenciesResolver {
     override fun resolve(
@@ -35,6 +36,7 @@ class TestDependenciesResolver : ScriptDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 class Template : Base()
 

idea/testData/script/definition/highlighting/emptyAsyncResolver/template/template.kt

@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : AsyncDependenciesResolver {
     override suspend fun resolveAsync(
@@ -14,6 +15,7 @@ class TestDependenciesResolver : AsyncDependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template: Base()
 

idea/testData/script/definition/highlighting/errorResolver/template/template.kt

@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(
@@ -20,5 +21,6 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template
\ No newline at end of file

idea/testData/script/definition/highlighting/noResolver/template/template.kt

@@ -3,7 +3,9 @@ package custom.scriptDefinition
 import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(scriptFilePattern = "script.kts")
 class Template: Base()
 

idea/testData/script/definition/highlighting/throwingResolver/template/template.kt

@@ -4,6 +4,7 @@ import java.io.File
 import kotlin.script.dependencies.*
 import kotlin.script.experimental.dependencies.*
 import kotlin.script.templates.ScriptTemplateDefinition
+import kotlin.script.experimental.location.*
 
 class TestDependenciesResolver : DependenciesResolver {
     override fun resolve(
@@ -14,5 +15,6 @@ class TestDependenciesResolver : DependenciesResolver {
     }
 }
 
+@ScriptExpectedLocations([ScriptExpectedLocation.Everywhere])
 @ScriptTemplateDefinition(TestDependenciesResolver::class, scriptFilePattern = "script.kts")
 open class Template
\ No newline at end of file