1 files changed, 72 insertions, 0 deletions
diff --git a/e2e/lib/Console.ts b/e2e/lib/Console.ts
new file mode 100644
index 0000000..233bf48
--- /dev/null
+++ b/e2e/lib/Console.ts
@@ -0,0 +1,72 @@
+import { WebDriver, By, Key } from 'selenium-webdriver';
+
+export type CompletionItem = {
+  type: string;
+  text: string;
+  highlight: boolean;
+}
+
+export class Console {
+  constructor(private webdriver: WebDriver) {
+  }
+
+  async sendKeys(...keys: string[]) {
+    let input = await this.webdriver.findElement(By.css('input'));
+    input.sendKeys(...keys);
+  }
+
+  async currentValue() {
+    return await this.webdriver.executeScript(() => {
+      let input = document.querySelector('input');
+      if (input === null) {
+        throw new Error('could not find input element');
+      }
+      return input.value;
+    });
+  }
+
+  async execCommand(command: string): Promise<void> {
+    let input = await this.webdriver.findElement(By.css('input.vimvixen-console-command-input'));
+    await input.sendKeys(command, Key.ENTER);
+  }
+
+  async getErrorMessage(): Promise<string> {
+    let p = await this.webdriver.findElement(By.css('.vimvixen-console-error'));
+    return p.getText();
+  }
+
+  async inputKeys(...keys: string[]) {
+    let input = await this.webdriver.findElement(By.css('input'));
+    await input.sendKeys(...keys);
+  }
+
+  getCompletions(): Promise<CompletionItem[]> {
+    return this.webdriver.executeScript(() => {
+      let items = document.querySelectorAll('.vimvixen-console-completion > li');
+      if (items.length === 0) {
+        throw new Error('completion items not found');
+      }
+
+      let objs = [];
+      for (let li of Array.from(items)) {
+        if (li.classList.contains('vimvixen-console-completion-title')) {
+          objs.push({ type: 'title', text: li.textContent!!.trim() });
+        } else if ('vimvixen-console-completion-item') {
+          let highlight = li.classList.contains('vimvixen-completion-selected');
+          objs.push({ type: 'item', text: li.textContent!!.trim(), highlight });
+        } else {
+          throw new Error(`unexpected class: ${li.className}`);
+        }
+      }
+      return objs;
+    });
+  }
+
+  async close(): Promise<void> {
+    let input = await this.webdriver.findElement(By.css('input'));
+    await input.sendKeys(Key.ESCAPE);
+    // TODO remove sleep
+    await new Promise(resolve => setTimeout(resolve, 100));
+    await (this.webdriver.switchTo() as any).parentFrame();
+  }
+}