import { createStore } from "jotai";
import { describe, expect, test, vi } from "vite-plus/test";

// The store reads the engine and the tree's selection through these; the toggle needs neither.
vi.mock("./rpc", () => ({ rpc: vi.fn(), listen: vi.fn() }));
vi.mock("@yaakapp-internal/ui", async () => {
  const { atom } = await vi.importActual<typeof import("jotai")>("jotai");
  return { selectedIdsFamily: () => atom<string[]>([]) };
});

const { activeQueryAtom, allTrafficTotalAtom, flowTotalAtom, hideSystemAtom, nonSystemTotalAtom } =
  await import("./store");

describe("Show in All Traffic, in System's menu", () => {
  test("shows System's traffic in All traffic by default", () => {
    const store = createStore();
    expect(store.get(hideSystemAtom)).toBe(false);
    expect(store.get(activeQueryAtom)).toBeNull();
  });

  test("leaves System out of All traffic and brings it back at once", () => {
    const store = createStore();
    store.set(nonSystemTotalAtom, 3);
    expect(store.get(allTrafficTotalAtom)).toBe(store.get(flowTotalAtom));
    store.set(hideSystemAtom, true);
    expect(store.get(activeQueryAtom)).toEqual({ query: "", hideSystem: true });
    expect(store.get(allTrafficTotalAtom)).toBe(3);
    store.set(hideSystemAtom, false);
    expect(store.get(activeQueryAtom)).toBeNull();
    expect(store.get(allTrafficTotalAtom)).toBe(store.get(flowTotalAtom));
  });
});