transform expected conditions where it was not put into a EC var

webdriverio · Apr 21, 2021 · b839c6a · b839c6a
1 parent a8c0eba
commit b839c6a
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 6 deletions.
diff --git a/protractor/index.js b/protractor/index.js
@@ -670,6 +670,12 @@ module.exports = function transformer(file, api) {
      * into
      * - const EC = require('wdio-wait-for')
      */
+    const wdioWaitForImport = j.callExpression(
+        j.identifier('require'),
+        [
+            j.literal('wdio-wait-for')
+        ]
+    )
     root.find(j.VariableDeclaration)
         .filter((path) => (
             path.value.declarations[0] &&
@@ -683,16 +689,25 @@ module.exports = function transformer(file, api) {
             [
                 j.variableDeclarator(
                     path.value.declarations[0].id,
-                    j.callExpression(
-                        j.identifier('require'),
-                        [
-                            j.literal('wdio-wait-for')
-                        ]
-                    )
+                    wdioWaitForImport
                 )
             ]
         ))
 
+    /**
+     * transform expected conditions not put into a var
+     */
+    root.find(j.MemberExpression)
+        .filter((path) => (
+            path.value.object.object &&
+            path.value.object.object.name === 'protractor' &&
+            path.value.object.property.name === 'ExpectedConditions'
+        ))
+        .replaceWith((path) => j.memberExpression(
+            wdioWaitForImport,
+            path.value.property
+        ))
+
     /**
      * no support
      */

diff --git a/test/__fixtures__/protractor/source/spec.js b/test/__fixtures__/protractor/source/spec.js
@@ -88,3 +88,7 @@ var width = 800;
 var height = 600;
 browser.driver.manage().window().setSize(width, height);
 browser.manage().deleteAllCookies();
+
+var titleIsNotFoo = protractor.ExpectedConditions.not(
+    protractor.ExpectedConditions.titleIs('Foo')
+);
diff --git a/test/__fixtures__/protractor/transformed/spec.js b/test/__fixtures__/protractor/transformed/spec.js
@@ -87,3 +87,7 @@ var width = 800;
 var height = 600;
 browser.setWindowRect(0, 0, width, height);
 browser.deleteCookies();
+
+var titleIsNotFoo = require("wdio-wait-for").not(
+    require("wdio-wait-for").titleIs('Foo')
+);