2 files changed, 66 insertions, 0 deletions
diff --git a/e2e/colorscheme.test.ts b/e2e/colorscheme.test.ts
new file mode 100644
index 0000000..3927a72
--- /dev/null
+++ b/e2e/colorscheme.test.ts
@@ -0,0 +1,60 @@
+import * as path from "path";
+import * as assert from "assert";
+
+import TestServer from "./lib/TestServer";
+import { Builder, Lanthan } from "lanthan";
+import { WebDriver } from "selenium-webdriver";
+import { Options as FirefoxOptions } from "selenium-webdriver/firefox";
+import Page from "./lib/Page";
+
+describe("colorscheme test", () => {
+  const server = new TestServer();
+  let lanthan: Lanthan;
+  let webdriver: WebDriver;
+  let page: Page;
+
+  before(async () => {
+    const opts = (new FirefoxOptions() as any).setPreference(
+      "ui.systemUsesDarkTheme",
+      1
+    );
+
+    lanthan = await Builder.forBrowser("firefox")
+      .setOptions(opts)
+      .spyAddon(path.join(__dirname, ".."))
+      .build();
+    webdriver = lanthan.getWebDriver();
+
+    await server.start();
+  });
+
+  after(async () => {
+    await server.stop();
+    if (lanthan) {
+      await lanthan.quit();
+    }
+  });
+
+  beforeEach(async () => {
+    page = await Page.navigateTo(webdriver, server.url());
+  });
+
+  it("changes color scheme by set command", async () => {
+    let console = await page.showConsole();
+
+    await console.execCommand("set colorscheme=dark");
+    await (webdriver.switchTo() as any).parentFrame();
+    console = await page.showConsole();
+    assert.strictEqual(await console.getTheme(), "dark");
+
+    await console.execCommand("set colorscheme=light");
+    await (webdriver.switchTo() as any).parentFrame();
+    console = await page.showConsole();
+    assert.strictEqual(await console.getTheme(), "light");
+
+    await console.execCommand("set colorscheme=system");
+    await (webdriver.switchTo() as any).parentFrame();
+    console = await page.showConsole();
+    assert.strictEqual(await console.getTheme(), "dark");
+  });
+});
diff --git a/e2e/lib/Console.ts b/e2e/lib/Console.ts
index 68c0e10..0bae2f7 100644
--- a/e2e/lib/Console.ts
+++ b/e2e/lib/Console.ts
@@ -76,6 +76,12 @@ export class Console {
     });
   }
 
+  async getTheme(): Promise<string> {
+    const wrapper = await this.webdriver.findElement(By.css("div[data-theme]"));
+    const theme = await wrapper.getAttribute("data-theme");
+    return theme;
+  }
+
   async close(): Promise<void> {
     const input = await this.webdriver.findElement(By.css("input"));
     await input.sendKeys(Key.ESCAPE);