expose method map to override timeout at test level

Author: saikrishna321

__tests__/e2e.spec.js

@@ -1,4 +1,4 @@
-import { remote } from 'webdriverio';
+import { remote, command } from 'webdriverio';
 
 const APPIUM_HOST = '127.0.0.1';
 const APPIUM_PORT = 31337;
@@ -22,6 +22,7 @@ describe('Plugin Test', () => {
   });
 
   it('Vertical swipe test', async () => {
+    await driver.setWaitPluginTimeout('data', { timeout: 1111, intervalBetweenAttempts: 11 });
     await driver.$('~username').click();
     await driver.$('~username').clearValue();
     await driver.$('~username').setValue('admin');
@@ -30,6 +31,21 @@ describe('Plugin Test', () => {
   });
 
   afterEach(async () => {
-    // await driver.deleteSession();
+    await driver.deleteSession();
   });
 });
+
+driver.addCommand(
+  'setWaitPluginTimeout',
+  command('POST', '/session/:sessionId/waitplugin/timeout', {
+    command: 'setWaitPluginTimeout',
+    parameters: [
+      {
+        name: 'data',
+        type: 'object',
+        description: 'a valid parameter',
+        required: true,
+      },
+    ],
+  })
+);

src/element.js

@@ -43,6 +43,10 @@ export async function find(driver, args) {
   }
 }
 
+export async function setWait(driver, args) {
+  _setTimeout(args[0], driver.sessionId, true);
+}
+
 export async function elementEnabled(driver, el) {
   let timeoutProp = _getTimeout(driver.sessionId);
   const predicate = async () => {
@@ -114,12 +118,16 @@ function sessionInfo(driver, strategy, selector) {
   }
 }
 
-function _setTimeout(elementWaitProps, session) {
+function _setTimeout(elementWaitProps, session, overrideTimeout = false) {
+  if (overrideTimeout && map.has(session)) {
+    map.delete(session);
+  }
   if (!map.has(session)) {
     log.info(`Timeout properties not set for session ${session}, trying to set one`);
     let defaultTimeoutProp = {
       timeout: 10000,
       intervalBetweenAttempts: 500,
+      overrideTimeout,
     };
     if (
       elementWaitProps.timeout != undefined &&

src/plugin.js

@@ -1,7 +1,22 @@
 import BasePlugin from '@appium/base-plugin';
-import { find, elementEnabled } from './element';
+import { find, elementEnabled, setWait } from './element';
 import log from './logger';
 export default class WaitCommandPlugin extends BasePlugin {
+  // this plugin supports a non-standard 'compare images' command
+  static newMethodMap = {
+    '/session/:sessionId/waitplugin/timeout': {
+      POST: {
+        command: 'setWait',
+        payloadParams: { required: ['data'] },
+        neverProxy: true,
+      },
+    },
+  };
+
+  async setWait(next, driver, ...args) {
+    await setWait(driver, args);
+  }
+
   async findElement(next, driver, ...args) {
     let commandLineArgs = this.cliArgs;
     await find(driver, args, commandLineArgs);