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

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