fix(js): fix test framework

unit-mesh · Aug 1, 2023 · 58f4ef4 · 58f4ef4
1 parent 33c1026
commit 58f4ef4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/webstorm/src/main/kotlin/cc/unitmesh/ide/webstorm/provider/JavaScriptContextProvider.kt b/webstorm/src/main/kotlin/cc/unitmesh/ide/webstorm/provider/JavaScriptContextProvider.kt
@@ -112,14 +112,16 @@ class JavaScriptContextProvider : ChatContextProvider {
 
                         devDependencies[name] = entry.versionRange
 
-                        JsWebFrameworks.values().forEach { frameworkName ->
-                            if (name.startsWith(frameworkName.name) || name == frameworkName.name) {
-                                frameworks[frameworkName.name] = true
+
+                        JsWebFrameworks.values().forEach { framework ->
+                            if (name.startsWith(framework.packageName) || name == framework.packageName) {
+                                frameworks[framework.packageName] = true
                             }
                         }
-                        JsTestFrameworks.values().forEach { testFramework ->
-                            if (name.startsWith(testFramework.name) || name == testFramework.name) {
-                                testFrameworks[testFramework.name] = true
+
+                        JsTestFrameworks.values().forEach { framework ->
+                            if (name.startsWith(framework.packageName) || name == framework.packageName) {
+                                testFrameworks[framework.packageName] = true
                             }
                         }
                     }

diff --git a/webstorm/src/main/kotlin/cc/unitmesh/ide/webstorm/provider/JavaScriptFrameworks.kt b/webstorm/src/main/kotlin/cc/unitmesh/ide/webstorm/provider/JavaScriptFrameworks.kt
@@ -25,6 +25,5 @@ enum class JsTestFrameworks(val presentation: String, val packageName: String) {
     Cypress("Cypress", "cypress"),
     Protractor("Protractor", "protractor"),
     Nightwatch("Nightwatch", "nightwatch"),
-    Selenium("Selenium", "selenium"),
-    Webdriverio("Webdriverio", "webdriverio")
+    Vitest("Vitest", "vitest")
 }