From 82aad419a74a3061247d4e82656fa711e63f213c Mon Sep 17 00:00:00 2001
From: Shin'ya Ueoka <ueokande@i-beam.org>
Date: Sat, 12 May 2018 23:21:16 +0900
Subject: Add navigate test

---
 e2e/ambassador/src/background/index.js |  4 +++-
 e2e/ambassador/src/background/tabs.js  |  2 +-
 e2e/ambassador/src/client/tabs.js      | 11 +++++++++--
 e2e/ambassador/src/shared/messages.js  |  2 ++
 4 files changed, 15 insertions(+), 4 deletions(-)

(limited to 'e2e/ambassador')

diff --git a/e2e/ambassador/src/background/index.js b/e2e/ambassador/src/background/index.js
index db2deb3..c40b4d8 100644
--- a/e2e/ambassador/src/background/index.js
+++ b/e2e/ambassador/src/background/index.js
@@ -1,6 +1,6 @@
 import {
   WINDOWS_CREATE, WINDOWS_REMOVE, WINDOWS_GET,
-  TABS_CREATE, TABS_SELECT_AT, TABS_GET_ZOOM, TABS_SET_ZOOM,
+  TABS_CREATE, TABS_SELECT_AT, TABS_GET, TABS_GET_ZOOM, TABS_SET_ZOOM,
   EVENT_KEYPRESS, EVENT_KEYDOWN, EVENT_KEYUP,
   SCROLL_GET, SCROLL_SET,
 } from '../shared/messages';
@@ -25,6 +25,8 @@ receiveContentMessage((message) => {
       windowId: message.windowId,
       index: message.index,
     });
+  case TABS_GET:
+    return browser.tabs.get(message.tabId);
   case TABS_GET_ZOOM:
     return browser.tabs.getZoom(message.tabId);
   case TABS_SET_ZOOM:
diff --git a/e2e/ambassador/src/background/tabs.js b/e2e/ambassador/src/background/tabs.js
index d049500..37156c4 100644
--- a/e2e/ambassador/src/background/tabs.js
+++ b/e2e/ambassador/src/background/tabs.js
@@ -5,7 +5,7 @@ const create = (props = {}) => {
         if (tab.url !== 'about:blank' && tabId === createdTab.id &&
             changeInfo.status === 'complete') {
           browser.tabs.onUpdated.removeListener(callback);
-          resolve(tab);
+          setTimeout(() => resolve(tab), 50) // wait for 50 milliseconds to ensure plugin loaded;
         }
       };
       browser.tabs.onUpdated.addListener(callback);
diff --git a/e2e/ambassador/src/client/tabs.js b/e2e/ambassador/src/client/tabs.js
index c7b1340..14d524d 100644
--- a/e2e/ambassador/src/client/tabs.js
+++ b/e2e/ambassador/src/client/tabs.js
@@ -1,5 +1,5 @@
 import {
-  TABS_CREATE, TABS_SELECT_AT, TABS_GET_ZOOM, TABS_SET_ZOOM,
+  TABS_CREATE, TABS_SELECT_AT, TABS_GET, TABS_GET_ZOOM, TABS_SET_ZOOM,
 } from '../shared/messages';
 import * as ipc from './ipc';
 
@@ -19,6 +19,13 @@ const selectAt = (windowId, index) => {
   });
 };
 
+const get = (tabId) => {
+  return ipc.send({
+    type: TABS_GET,
+    tabId,
+  });
+};
+
 const getZoom = (tabId) => {
   return ipc.send({
     tabId,
@@ -34,4 +41,4 @@ const setZoom = (tabId, factor) => {
   });
 };
 
-export { create, selectAt, getZoom, setZoom };
+export { create, selectAt, get, getZoom, setZoom };
diff --git a/e2e/ambassador/src/shared/messages.js b/e2e/ambassador/src/shared/messages.js
index 1fc47d2..34ec5d3 100644
--- a/e2e/ambassador/src/shared/messages.js
+++ b/e2e/ambassador/src/shared/messages.js
@@ -5,6 +5,7 @@ const WINDOWS_REMOVE = 'windows.remove';
 const WINDOWS_GET = 'windows.get';
 const TABS_CREATE = 'tabs.create';
 const TABS_SELECT_AT = 'tabs.selectAt';
+const TABS_GET = 'tabs.get';
 const TABS_GET_ZOOM = 'tabs.get.zoom';
 const TABS_SET_ZOOM = 'tabs.set.zoom';
 const EVENT_KEYPRESS = 'event.keypress';
@@ -21,6 +22,7 @@ export {
   WINDOWS_REMOVE,
   WINDOWS_GET,
 
+  TABS_GET,
   TABS_CREATE,
   TABS_SELECT_AT,
   TABS_GET_ZOOM,
-- 
cgit v1.2.3