add tests for commands.logout() and commands.session()

Tatamo · Tatamo · commit b8e334a232a2 · 2021-08-15T23:02:12.000+09:00
diff --git a/tests/__tests__/__snapshots__/commands.ts.snap b/tests/__tests__/__snapshots__/commands.ts.snap
@@ -9,3 +9,20 @@ exports[`login() login() login success 1`] = `
 "OK
 "
 `;
+
+exports[`login() logout() logout 1`] = `
+"login session aborted.
+"
+`;
+
+exports[`login() session() logged-in 1`] = `
+"check login status...
+OK
+"
+`;
+
+exports[`login() session() not logged-in 1`] = `
+"check login status...
+not login
+"
+`;
diff --git a/tests/__tests__/commands.ts b/tests/__tests__/commands.ts
@@ -1,4 +1,4 @@
-import { login } from "../../src/commands";
+import { login, logout, session } from "../../src/commands";
 import { AtCoder } from "../../src/atcoder";
 import { getAtCoder } from "../../src/di";
 jest.mock("../../src/di");
@@ -34,4 +34,35 @@ describe("login()", () => {
             spy_console_log.mockRestore();
         })
     });
+    describe("logout()", () => {
+        test("logout", async () => {
+            let log = "";
+            const spy_console_log = jest.spyOn(console, "log").mockImplementation(s => log += `${s}\n`);
+            atcoder.logout = jest.fn();
+
+            await logout();
+            expect(log).toMatchSnapshot();
+            spy_console_log.mockRestore();
+        })
+    });
+    describe("session()", () => {
+        test("logged-in", async () => {
+            let log = "";
+            const spy_console_log = jest.spyOn(console, "log").mockImplementation(s => log += `${s}\n`);
+            atcoder.checkSession = jest.fn(async () => true);
+
+            await session();
+            expect(log).toMatchSnapshot();
+            spy_console_log.mockRestore();
+        })
+        test("not logged-in", async () => {
+            let log = "";
+            const spy_console_log = jest.spyOn(console, "log").mockImplementation(s => log += `${s}\n`);
+            atcoder.checkSession = jest.fn(async () => false);
+
+            await session();
+            expect(log).toMatchSnapshot();
+            spy_console_log.mockRestore();
+        })
+    });
 });
