131 lines
3.3 KiB
TypeScript
131 lines
3.3 KiB
TypeScript
import { describe, expect, test, vi } from "vitest";
|
|
|
|
import { registerBackgroundMessageHandler } from "../src/background/index";
|
|
|
|
describe("background-index", () => {
|
|
test("downloads csv files when the content script sends a download message", async () => {
|
|
const listeners: Array<
|
|
(message: unknown, sender: unknown, sendResponse: (response: unknown) => void) => boolean | void
|
|
> = [];
|
|
const download = vi.fn(async () => undefined);
|
|
const sendResponse = vi.fn();
|
|
|
|
registerBackgroundMessageHandler({
|
|
downloads: {
|
|
download
|
|
},
|
|
runtime: {
|
|
onMessage: {
|
|
addListener(listener) {
|
|
listeners.push(listener);
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
expect(listeners).toHaveLength(1);
|
|
const result = listeners[0](
|
|
{
|
|
csv: "列1,列2\n值1,值2",
|
|
filename: "market.csv",
|
|
type: "download-market-csv"
|
|
},
|
|
{},
|
|
sendResponse
|
|
);
|
|
|
|
expect(result).toBe(true);
|
|
await new Promise((resolve) => setTimeout(resolve, 0));
|
|
|
|
expect(download).toHaveBeenCalledTimes(1);
|
|
expect(download).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
filename: "market.csv",
|
|
saveAs: false,
|
|
url: expect.stringContaining("data:text/csv;charset=utf-8,")
|
|
})
|
|
);
|
|
expect(sendResponse).toHaveBeenCalledWith({ ok: true });
|
|
});
|
|
|
|
test("responds to auth:get-state with auth status", async () => {
|
|
const listeners: Array<
|
|
(message: unknown, sender: unknown, sendResponse: (response: unknown) => void) => boolean | void
|
|
> = [];
|
|
const sendResponse = vi.fn();
|
|
|
|
registerBackgroundMessageHandler(
|
|
{
|
|
runtime: {
|
|
onMessage: {
|
|
addListener(listener) {
|
|
listeners.push(listener);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
authController: {
|
|
getAccessToken: vi.fn(),
|
|
getAuthState: vi.fn(async () => ({ isAuthenticated: false })),
|
|
signIn: vi.fn(),
|
|
signOut: vi.fn()
|
|
}
|
|
}
|
|
);
|
|
|
|
const result = listeners[0]({ type: "auth:get-state" }, {}, sendResponse);
|
|
|
|
expect(result).toBe(true);
|
|
await new Promise((resolve) => setTimeout(resolve, 0));
|
|
|
|
expect(sendResponse).toHaveBeenCalledWith({
|
|
ok: true,
|
|
type: "auth:state",
|
|
value: { isAuthenticated: false }
|
|
});
|
|
});
|
|
|
|
test("responds to auth:get-access-token with the current token", async () => {
|
|
const listeners: Array<
|
|
(message: unknown, sender: unknown, sendResponse: (response: unknown) => void) => boolean | void
|
|
> = [];
|
|
const sendResponse = vi.fn();
|
|
|
|
registerBackgroundMessageHandler(
|
|
{
|
|
runtime: {
|
|
onMessage: {
|
|
addListener(listener) {
|
|
listeners.push(listener);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
authController: {
|
|
getAccessToken: vi.fn(async () => "test-access-token"),
|
|
getAuthState: vi.fn(),
|
|
signIn: vi.fn(),
|
|
signOut: vi.fn()
|
|
}
|
|
}
|
|
);
|
|
|
|
const result = listeners[0](
|
|
{ type: "auth:get-access-token" },
|
|
{},
|
|
sendResponse
|
|
);
|
|
|
|
expect(result).toBe(true);
|
|
await new Promise((resolve) => setTimeout(resolve, 0));
|
|
|
|
expect(sendResponse).toHaveBeenCalledWith({
|
|
ok: true,
|
|
type: "auth:token",
|
|
value: { accessToken: "test-access-token" }
|
|
});
|
|
});
|
|
});
|