add tab tests

2 files changed, 15 insertions, 2 deletions

diff --git a/e2e/ambassador/src/background/index.js b/e2e/ambassador/src/background/index.js
index c3e9dc1..6425b54 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_CREATE, TABS_SELECT_AT,
   EVENT_KEYPRESS, EVENT_KEYDOWN, EVENT_KEYUP,
   SCROLL_GET, SCROLL_SET,
 } from '../shared/messages';
@@ -20,6 +20,11 @@ receiveContentMessage((message) => {
       url: message.url,
       windowId: message.windowId,
     });
+  case TABS_SELECT_AT:
+    return tabs.selectAt({
+      windowId: message.windowId,
+      index: message.index,
+    });
   case EVENT_KEYPRESS:
   case EVENT_KEYDOWN:
   case EVENT_KEYUP:
diff --git a/e2e/ambassador/src/background/tabs.js b/e2e/ambassador/src/background/tabs.js
index 93d47a3..d049500 100644
--- a/e2e/ambassador/src/background/tabs.js
+++ b/e2e/ambassador/src/background/tabs.js
@@ -13,6 +13,14 @@ const create = (props = {}) => {
   });
 };
 
+const selectAt = (props = {}) => {
+  return browser.tabs.query({ windowId: props.windowId }).then((tabs) => {
+    let target = tabs[props.index];
+    return browser.tabs.update(target.id, { active: true });
+  });
+};
+
+
 export {
-  create,
+  create, selectAt
 };