fix(webdriverio): add expect-webdriverio to tsconfig (#6309)

vuejs · Feb 23, 2021 · a92bc30 · a92bc30
1 parent 8670997
commit a92bc30
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/packages/@vue/cli-plugin-e2e-webdriverio/migrator/index.js b/packages/@vue/cli-plugin-e2e-webdriverio/migrator/index.js
@@ -1,11 +1,20 @@
 module.exports = (api) => {
   if (api.fromVersion('<= 5.0.0-alpha.4')) {
-    api.render(files => {
+    api.render((files) => {
       if (!files['tsconfig.json']) {
         return
       }
 
-      files['tsconfig.json'] = files['tsconfig.json'].replace('"@wdio/sync"', '"webdriverio/sync"')
+      files['tsconfig.json'] = files['tsconfig.json'].replace(
+        '"@wdio/sync"',
+        '"webdriverio/sync"'
+      )
+      if (!/"expect-webdriverio"/.test(files['tsconfig.json'])) {
+        files['tsconfig.json'] = files['tsconfig.json'].replace(
+          '"@wdio/mocha-framework",',
+          '"@wdio/mocha-framework",\n      "expect-webdriverio",'
+        )
+      }
     })
   }
 }
diff --git a/packages/@vue/cli-plugin-typescript/generator/template/tsconfig.json b/packages/@vue/cli-plugin-typescript/generator/template/tsconfig.json
@@ -25,6 +25,7 @@
       <%_ if (hasWebDriverIO) { _%>
       <% if (!hasMocha && !hasJest) { %>"mocha",<% } %>
       "@wdio/mocha-framework",
+      "expect-webdriverio",
       "webdriverio/sync"<% if (hasMocha || hasJest) { %>,<% } %>
       <%_ } _%>
       <%_ if (hasMocha) { _%>