[Analysis API] Explicitly specify the context module in a directive

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/annotationUsage.kt

@@ -22,6 +22,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/contextReceiver.kt

@@ -21,6 +21,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/contextReceiverExplicit.kt

@@ -21,6 +21,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiver.kt

@@ -15,6 +15,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverExplicit.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverLabeled.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverProperty.kt

@@ -13,6 +13,7 @@ val Int.ext: String
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverPropertyExplicit.kt

@@ -13,6 +13,7 @@ val Int.ext: String
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverPropertyLabeled.kt

@@ -13,6 +13,7 @@ val Int.ext: String
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/extensionReceiverSmartCasted.kt

@@ -18,6 +18,7 @@ fun main() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/initializer.kt

@@ -10,6 +10,7 @@ class Foo(a: Int, b: String) {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/local.kt

@@ -9,6 +9,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localDelegatedProperty.kt

@@ -21,6 +21,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunction.kt

@@ -14,6 +14,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionContainingClassClosure.kt

@@ -16,6 +16,7 @@ fun consume(obj: Foo) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionExtensionReceiverClosure.kt

@@ -20,6 +20,7 @@ fun consume(text: String) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionLambdaParameterClosure.kt

@@ -20,6 +20,7 @@ fun consume(text: String) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionLocalClosure.kt

@@ -16,6 +16,7 @@ fun consume(n: Int) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionLocalClosureMutating.kt

@@ -14,6 +14,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionMultipleCapturing.kt

@@ -21,6 +21,7 @@ fun consume(obj: Any) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localFunctionParameterClosure.kt

@@ -14,6 +14,7 @@ fun consume(n: Int) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localMutated.kt

@@ -9,6 +9,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/localVariable.kt

@@ -9,6 +9,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/multipleClassAndFunctionContextReceivers.kt

@@ -22,6 +22,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/multipleClassContextReceivers.kt

@@ -19,6 +19,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/multipleFunctionContextReceivers.kt

@@ -17,6 +17,7 @@ fun foo() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/nestedOuterClass.kt

@@ -24,6 +24,7 @@ fun callString(string: String): Int {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/objectFunction.kt

@@ -12,6 +12,7 @@ object Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/outerClass.kt

@@ -12,6 +12,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/outerClassMutated.kt

@@ -12,6 +12,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/outerClassMutatedPrivate.kt

@@ -12,6 +12,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/preferCorrectResolve.kt

@@ -12,6 +12,7 @@ fun test(a: A) {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/privateMemberCall.kt

@@ -12,6 +12,7 @@ fun test(a: A) {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/secondClassContextReceiver.kt

@@ -19,6 +19,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/singleClassContextReceiver.kt

@@ -18,6 +18,7 @@ class Test {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/smartCastExtensionReceiver.kt

@@ -24,6 +24,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/smartCastValueParameter.kt

@@ -14,6 +14,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/capturing/valueParameter.kt

@@ -12,6 +12,7 @@ class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/commonContext.kt

@@ -17,6 +17,7 @@ expect class Foo {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: common
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/customName.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/duplicateLibrary.kt

@@ -12,6 +12,7 @@ fun consume(text: String) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/fragmentFunction.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/imports.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/inlineFunctionUsage.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/inlineFunctionUsageSource.kt

@@ -12,6 +12,7 @@ inline fun call(block: (Int) -> Int) {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/local.kt

@@ -8,6 +8,7 @@ fun foo() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/simple.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/syntaxError.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/syntaxError.txt

@@ -1 +1 @@
-(134,135) SYNTAX Syntax error: Unexpected symbol.
+(161,162) SYNTAX Syntax error: Unexpected symbol.

analysis/analysis-api/testData/components/compilerFacility/compilationMultiModule/codeFragments/compilerPluginUsage.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/components/containingDeclarationProvider/containingDeclarationByPsi/codeFragments.kt

@@ -10,6 +10,7 @@ fun foo() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: main.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/BinaryExpression.kt

@@ -13,6 +13,7 @@ val b: Int = 5
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/CompanionCall.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/ContextFunctionCall.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/FunctionCall.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/LocalFunctionCall.kt

@@ -11,6 +11,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/NestedCodeFragment.kt

@@ -13,6 +13,7 @@ fun bar() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION
@@ -20,6 +21,7 @@ fun bar() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/NestedCodeFragmentClass.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: BLOCK
@@ -20,6 +21,7 @@ class Local {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/NestedCodeFragmentClassMember.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: BLOCK
@@ -24,6 +25,7 @@ class Local {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/NestedCodeFragmentFunction.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: BLOCK
@@ -20,6 +21,7 @@ fun bar() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/PropertyCall.kt

@@ -12,6 +12,7 @@ val foo = "foo"
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/blockCodeFragment/SecondStatement.kt

@@ -13,6 +13,7 @@ fun bar() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/BinaryExpression.kt

@@ -13,6 +13,7 @@ val b: Int = 5
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/CompanionCall.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/ContextFunctionCall.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/FunctionCall.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/LocalFunctionCall.kt

@@ -11,6 +11,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/NestedCodeFragment.kt

@@ -13,6 +13,7 @@ fun bar() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION
@@ -20,6 +21,7 @@ fun bar() {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/NestedCodeFragmentUsage.kt

@@ -12,6 +12,7 @@ fun foo() {}
 
 // MODULE: contextFragment
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: contextFragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION
@@ -23,6 +24,7 @@ object {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: contextFragment
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/expressionCodeFragment/PropertyCall.kt

@@ -12,6 +12,7 @@ val foo = "foo"
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/FromContextFile.kt

@@ -12,6 +12,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/FromContextModule.kt

@@ -13,6 +13,7 @@ object Foo
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/GenericType.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/GenericTypeArgument.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/LocalClass.kt

@@ -12,6 +12,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/LocalClass.txt

@@ -0,0 +1,2 @@
+Resolved to:
+0: (in <local>: test) class Local

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/Simple.kt

@@ -10,6 +10,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-api/testData/referenceResolve/codeFragment/typeCodeFragment/TypeAnnotation.kt

@@ -13,6 +13,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: TYPE

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/AnalysisApiTestDirectives.kt

@@ -34,4 +34,9 @@ object AnalysisApiTestDirectives : SimpleDirectivesContainer() {
         description = "Mark the module as main",
         applicability = DirectiveApplicability.Module,
     )
+
+    val CONTEXT_MODULE by stringDirective(
+        description = "Specifies the module name which should be treated as a context module for the current one",
+        applicability = DirectiveApplicability.Module
+    )
 }

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/project/structure/KtCodeFragmentModuleFactory.kt

@@ -31,7 +31,7 @@ object KtCodeFragmentModuleFactory : KtModuleFactory {
         testServices: TestServices,
         project: Project,
     ): KtModuleWithFiles {
-        requireNotNull(contextModule) { "Context module should precede the code fragment module" }
+        requireNotNull(contextModule) { "Code fragment requires a context module" }
 
         val testFile = testModule.files.singleOrNull() ?: error("A single file is expected for a code fragment module")
 

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/project/structure/TestModuleStructureFactory.kt

@@ -16,6 +16,7 @@ import org.jetbrains.kotlin.analysis.project.structure.KtBinaryModule
 import org.jetbrains.kotlin.analysis.project.structure.KtDanglingFileModule
 import org.jetbrains.kotlin.analysis.project.structure.KtModule
 import org.jetbrains.kotlin.analysis.project.structure.KtNotUnderContentRootModule
+import org.jetbrains.kotlin.analysis.test.framework.AnalysisApiTestDirectives
 import org.jetbrains.kotlin.analysis.test.framework.services.environmentManager
 import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
 import org.jetbrains.kotlin.cli.jvm.config.JvmClasspathRoot
@@ -25,6 +26,7 @@ import org.jetbrains.kotlin.platform.TargetPlatform
 import org.jetbrains.kotlin.platform.js.JsPlatforms
 import org.jetbrains.kotlin.platform.jvm.JvmPlatforms
 import org.jetbrains.kotlin.platform.jvm.isJvm
+import org.jetbrains.kotlin.test.directives.model.singleOrZeroValue
 import org.jetbrains.kotlin.test.model.DependencyRelation
 import org.jetbrains.kotlin.test.model.TestModule
 import org.jetbrains.kotlin.test.services.*
@@ -46,9 +48,7 @@ object TestModuleStructureFactory {
         testServices: TestServices,
         project: Project
     ): KtModuleProjectStructure {
-        val modules = moduleStructure.modules.mapWithPrevious<TestModule, KtModuleWithFiles> { contextModule, testModule ->
+        val modules = createModules(moduleStructure, testServices, project)
-            testServices.getKtModuleFactoryForTestModule(testModule).createModule(testModule, contextModule, testServices, project)
-        }
 
         val modulesByName = modules.associateByName()
 
@@ -64,14 +64,27 @@ object TestModuleStructureFactory {
         return KtModuleProjectStructure(modules, libraryCache.values)
     }
 
-    private fun <T : Any, R> List<T>.mapWithPrevious(transform: (R?, T) -> R): List<R> {
+    private fun createModules(
-        val result = ArrayList<R>(this.size)
+        moduleStructure: TestModuleStructure,
-        var previousResult: R? = null
+        testServices: TestServices,
-        for (item in this) {
+        project: Project
-            val newResult = transform(previousResult, item)
+    ): List<KtModuleWithFiles> {
-            result.add(newResult)
+        val moduleCount = moduleStructure.modules.size
-            previousResult = newResult
+        val existingModules = HashMap<String, KtModuleWithFiles>(moduleCount)
+        val result = ArrayList<KtModuleWithFiles>(moduleCount)
+
+        for (testModule in moduleStructure.modules) {
+            val contextModuleName = testModule.directives.singleOrZeroValue(AnalysisApiTestDirectives.CONTEXT_MODULE)
+            val contextModule = contextModuleName?.let(existingModules::getValue)
+
+            val moduleWithFiles = testServices
+                .getKtModuleFactoryForTestModule(testModule)
+                .createModule(testModule, contextModule, testServices, project)
+
+            existingModules[testModule.name] = moduleWithFiles
+            result.add(moduleWithFiles)
         }
+
         return result
     }
 

analysis/low-level-api-fir/testData/inBlockModification/codeFragments/localClass.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/low-level-api-fir/testData/inBlockModification/codeFragments/localFunction.kt

@@ -8,6 +8,7 @@ fun test() {
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: BLOCK

analysis/low-level-api-fir/testData/inBlockModification/codeFragments/simple.kt

@@ -15,6 +15,7 @@ fun consume(foo: Foo) {}
 
 // MODULE: main
 // MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: context
 
 // FILE: fragment.kt
 // CODE_FRAGMENT_KIND: EXPRESSION