add tests and fix bug

Author: baseballyama

docs/rules/valid-context-access.md

@@ -45,11 +45,23 @@ This rule reports where context API is called except during component initializa
     update()
     setContext("answer", 42)
   })
+
+  const update2 = async () => {
+    await Promise.resolve()
+    setContext("answer", 42)
+  }
+
+  ;(async () => {
+    await Promise.resolve()
+    setContext("answer", 42)
+  })()
 </script>
 ```
 
 </ESLintCodeBlock>
 
+- :warning: This rule only inspects Svelte files, not JS / TS files.
+
 ## :wrench: Options
 
 Nothing.

src/rules/valid-context-access.ts

@@ -107,6 +107,7 @@ export default createRule("valid-context-access", {
       belongingFunction:
         | TSESTree.FunctionDeclaration
         | TSESTree.VariableDeclaration
+        | TSESTree.ArrowFunctionExpression
       node: TSESTree.Node
     }[] = []
 
@@ -207,9 +208,11 @@ export default createRule("valid-context-access", {
       AwaitExpression(node) {
         let parent: TSESTree.Node | undefined = node.parent
         while (parent) {
-          if (parent.type === "FunctionDeclaration") {
-            awaitExpressions.push({ belongingFunction: parent, node })
-          } else if (parent.type === "VariableDeclaration") {
+          if (
+            parent.type === "FunctionDeclaration" ||
+            parent.type === "VariableDeclaration" ||
+            parent.type === "ArrowFunctionExpression"
+          ) {
             awaitExpressions.push({ belongingFunction: parent, node })
           }
           parent = parent.parent

tests/fixtures/rules/valid-context-access/invalid/case13-errors.yaml

@@ -0,0 +1,3 @@
+- message: Do not call setContext except during component initialization.
+  line: 5
+  column: 5

tests/fixtures/rules/valid-context-access/invalid/case13-input.svelte

@@ -0,0 +1,7 @@
+<script>
+  import { setContext } from "svelte"
+  ;(async () => {
+    await Promise.resolve()
+    setContext("answer", 42)
+  })()
+</script>

tests/fixtures/rules/valid-context-access/invalid/case14-errors.yaml

@@ -0,0 +1,3 @@
+- message: Do not call setContext except during component initialization.
+  line: 4
+  column: 5

tests/fixtures/rules/valid-context-access/invalid/case14-input.svelte

@@ -0,0 +1,10 @@
+<script>
+  import { setContext, onMount } from "svelte"
+  const doSomething = () => {
+    setContext("answer", 42)
+  }
+
+  onMount(() => {
+    doSomething()
+  })
+</script>

tests/fixtures/rules/valid-context-access/invalid/case15-errors.yaml

@@ -0,0 +1,3 @@
+- message: Do not call setContext except during component initialization.
+  line: 4
+  column: 5

tests/fixtures/rules/valid-context-access/invalid/case15-input.svelte

@@ -0,0 +1,11 @@
+<script>
+  import { setContext, onMount } from "svelte"
+  const doSomething = () => {
+    setContext("answer", 42)
+  }
+
+  ;(async () => {
+    await Promise.resolve()
+    doSomething()
+  })()
+</script>