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 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

(limited to 'e2e/ambassador/src/background')

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);
-- 
cgit v1.2.3