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

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