1 files changed, 125 insertions, 0 deletions
diff --git a/e2e/command_quit.test.js b/e2e/command_quit.test.js
new file mode 100644
index 0000000..ee4c2d8
--- /dev/null
+++ b/e2e/command_quit.test.js
@@ -0,0 +1,125 @@
+const express = require('express');
+const lanthan = require('lanthan');
+const path = require('path');
+const assert = require('assert');
+const eventually = require('./eventually');
+
+const Key = lanthan.Key;
+
+const newApp = () => {
+  let app = express();
+  app.get('/*', (req, res) => {
+    res.send(`<!DOCTYPEhtml>
+<html lang="en">
+  <head>
+    <title>my_${req.path.slice(1)}</title>
+  </head>
+  <body><h1>${req.path}</h1></body>
+</html">`);
+  });
+  return app;
+};
+
+describe('quit/quitall command test', () => {
+  const port = 12321;
+  let http;
+  let firefox;
+  let session;
+  let browser;
+
+  before(async() => {
+    http = newApp().listen(port);
+
+    firefox = await lanthan.firefox({
+      spy: path.join(__dirname, '..'),
+      builderf: (builder) => {
+        builder.addFile('build/settings.js');
+      },
+    });
+    session = firefox.session;
+    browser = firefox.browser;
+  });
+
+  after(async() => {
+    http.close();
+    if (firefox) {
+      await firefox.close();
+    }
+  });
+
+  beforeEach(async() => {
+    let tabs = await browser.tabs.query({});
+    for (let tab of tabs.slice(1)) {
+      await browser.tabs.remove(tab.id);
+    }
+    await browser.tabs.update(tabs[0].id, { url: `http://127.0.0.1:${port}/site1` });
+    for (let i = 2; i <= 5; ++i) {
+      await browser.tabs.create({ url: `http://127.0.0.1:${port}/site${i}`})
+    }
+
+    await eventually(async() => {
+      let handles = await session.getWindowHandles();
+      assert.equal(handles.length, 5);
+      await session.switchToWindow(handles[2]);
+      await session.findElementByCSS('iframe');
+    });
+
+    await new Promise((resolve) => setTimeout(resolve, 100));
+  });
+
+  it('should current tab by q command', async() => {
+    let body = await session.findElementByCSS('body');
+    await body.sendKeys(':');
+
+    await session.switchToFrame(0);
+    let input = await session.findElementByCSS('input');
+    await input.sendKeys('q', Key.Enter);
+
+    await eventually(async() => {
+      let tabs = await browser.tabs.query({});
+      assert.equal(tabs.length, 4)
+    });
+  });
+
+  it('should current tab by quit command', async() => {
+    let body = await session.findElementByCSS('body');
+    await body.sendKeys(':');
+
+    await session.switchToFrame(0);
+    let input = await session.findElementByCSS('input');
+    await input.sendKeys('quit', Key.Enter);
+
+    await eventually(async() => {
+      let tabs = await browser.tabs.query({});
+      assert.equal(tabs.length, 4)
+    });
+  });
+
+  it('should current tab by qa command', async() => {
+    let body = await session.findElementByCSS('body');
+    await body.sendKeys(':');
+
+    await session.switchToFrame(0);
+    let input = await session.findElementByCSS('input');
+    await input.sendKeys('qa', Key.Enter);
+
+    await eventually(async() => {
+      let tabs = await browser.tabs.query({});
+      assert.equal(tabs.length, 1)
+    });
+  });
+
+  it('should current tab by quitall command', async() => {
+    let body = await session.findElementByCSS('body');
+    await body.sendKeys(':');
+
+    await session.switchToFrame(0);
+    let input = await session.findElementByCSS('input');
+    await input.sendKeys('quitall', Key.Enter);
+
+    await eventually(async() => {
+      let tabs = await browser.tabs.query({});
+      assert.equal(tabs.length, 1)
+    });
+  });
+});