Dan Ditomaso
1 year ago
17 changed files with 471 additions and 368 deletions
@ -1,102 +1,97 @@ |
|||||
import { render, screen, fireEvent } from "@testing-library/react"; |
import { render, screen } from "@testing-library/react"; |
||||
import { beforeEach, describe, expect, it, vi, Mock } from "vitest"; |
import { DeviceContext, useDeviceStore } from "@core/stores/deviceStore.ts"; |
||||
import { RefreshKeysDialog } from "./RefreshKeysDialog.tsx"; |
import { RefreshKeysDialog } from "./RefreshKeysDialog.tsx"; |
||||
|
import { useMessageStore } from "@core/stores/messageStore.ts"; |
||||
import { useRefreshKeysDialog } from "./useRefreshKeysDialog.ts"; |
import { useRefreshKeysDialog } from "./useRefreshKeysDialog.ts"; |
||||
import { useMessageStore } from "@core/stores/messageStore.ts"; // Import for mocking
|
import { expect, test, vi, beforeEach, afterEach } from 'vitest'; |
||||
import { useDevice } from "@core/stores/deviceStore.ts"; // Import for mocking
|
|
||||
import { Protobuf } from "@meshtastic/core"; |
import { Protobuf } from "@meshtastic/core"; |
||||
|
|
||||
|
vi.mock("@core/stores/messageStore"); |
||||
|
vi.mock("./useRefreshKeysDialog"); |
||||
|
|
||||
vi.mock("@core/stores/messageStore.ts", () => ({ |
const mockUseMessageStore = vi.mocked(useMessageStore); |
||||
useMessageStore: vi.fn(), |
const mockUseRefreshKeysDialog = vi.mocked(useRefreshKeysDialog); |
||||
})); |
|
||||
|
|
||||
const mockNodeWithError: Partial<Protobuf.Mesh.NodeInfo> = { |
const getInitialState = () => useDeviceStore.getInitialState?.() ?? { devices: new Map(), remoteDevices: new Map() }; |
||||
user: { longName: "Test Node Long", shortName: "TNL", id: 456 }, |
|
||||
}; |
|
||||
const mockNodes = new Map([[456, mockNodeWithError]]); |
|
||||
const mockNodeErrors = new Map([[123, { node: 456 }]]); |
|
||||
|
|
||||
vi.mock("@core/stores/deviceStore.ts", () => ({ |
beforeEach(() => { |
||||
useDevice: vi.fn(), |
useDeviceStore.setState(getInitialState(), true); |
||||
})); |
|
||||
|
|
||||
const mockHandleCloseDialog = vi.fn(); |
|
||||
const mockHandleNodeRemove = vi.fn(); |
|
||||
vi.mock("./useRefreshKeysDialog.ts", () => ({ |
|
||||
useRefreshKeysDialog: vi.fn(() => ({ |
|
||||
handleCloseDialog: mockHandleCloseDialog, |
|
||||
handleNodeRemove: mockHandleNodeRemove, |
|
||||
})), |
|
||||
})); |
|
||||
|
|
||||
describe("RefreshKeysDialog Component", () => { |
|
||||
let onOpenChangeMock: Mock; |
|
||||
|
|
||||
beforeEach(() => { |
|
||||
vi.clearAllMocks(); |
vi.clearAllMocks(); |
||||
onOpenChangeMock = vi.fn(); |
}); |
||||
|
|
||||
vi.mocked(useMessageStore).mockReturnValue({ activeChat: 123 }); |
|
||||
vi.mocked(useDevice).mockReturnValue({ |
|
||||
nodeErrors: mockNodeErrors, |
|
||||
nodes: mockNodes, |
|
||||
}); |
|
||||
vi.mocked(useRefreshKeysDialog).mockReturnValue({ |
|
||||
handleCloseDialog: mockHandleCloseDialog, |
|
||||
handleNodeRemove: mockHandleNodeRemove, |
|
||||
}); |
|
||||
}); |
|
||||
|
|
||||
it("should render the dialog with dynamic content when open and data is available", () => { |
afterEach(() => { |
||||
render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
vi.restoreAllMocks(); |
||||
|
}); |
||||
|
|
||||
expect(screen.getByText(`Keys Mismatch - ${mockNodeWithError?.user?.longName}`)).toBeInTheDocument(); |
test("renders dialog when there is a node error for the active chat", () => { |
||||
expect(screen.getByText(new RegExp(`${mockNodeWithError?.user?.longName}.*${mockNodeWithError?.user?.shortName}`))).toBeInTheDocument(); |
const deviceId = 1; |
||||
expect(screen.getByRole('button', { name: /request new keys/i })).toBeInTheDocument(); |
const nodeWithErrorNum = 12345; |
||||
expect(screen.getByRole('button', { name: /dismiss/i })).toBeInTheDocument(); |
const activeChatNum = nodeWithErrorNum; |
||||
expect(screen.getByRole('button', { name: /Close/i })).toBeInTheDocument(); |
|
||||
|
const deviceStore = useDeviceStore.getState().addDevice(deviceId); |
||||
|
|
||||
|
deviceStore.addNodeInfo({ |
||||
|
num: nodeWithErrorNum, |
||||
|
user: { |
||||
|
id: nodeWithErrorNum.toString(), |
||||
|
publicKey: new Uint8Array(0), |
||||
|
hwModel: Protobuf.Mesh.HardwareModel.HELTEC_V3, |
||||
|
longName: "Problem Node Long", |
||||
|
shortName: "ProbNode", |
||||
|
isLicensed: false, |
||||
|
macaddr: new Uint8Array(0) |
||||
|
}, |
||||
|
lastHeard: Date.now() / 1000, |
||||
|
snr: 10 |
||||
|
} as Protobuf.Mesh.NodeInfo); |
||||
|
|
||||
|
deviceStore.setNodeError(activeChatNum, "PKI_MISMATCH"); |
||||
|
|
||||
|
const updatedDeviceState = useDeviceStore.getState().getDevice(deviceId); |
||||
|
if (!updatedDeviceState) { |
||||
|
throw new Error("Failed to get updated device state from store for provider"); |
||||
|
} |
||||
|
|
||||
|
mockUseMessageStore.mockReturnValue({ activeChat: activeChatNum }); |
||||
|
const mockHandleClose = vi.fn(); |
||||
|
const mockHandleRemove = vi.fn(); |
||||
|
mockUseRefreshKeysDialog.mockReturnValue({ |
||||
|
handleCloseDialog: mockHandleClose, |
||||
|
handleNodeRemove: mockHandleRemove, |
||||
}); |
}); |
||||
|
|
||||
it("should call handleNodeRemove when 'Request New Keys' button is clicked", () => { |
render( |
||||
render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
<DeviceContext.Provider value={updatedDeviceState}> |
||||
fireEvent.click(screen.getByRole('button', { name: /request new keys/i })); |
<RefreshKeysDialog open onOpenChange={vi.fn()} /> |
||||
expect(mockHandleNodeRemove).toHaveBeenCalledTimes(1); |
</DeviceContext.Provider> |
||||
}); |
); |
||||
|
|
||||
it("should call handleCloseDialog when 'Dismiss' button is clicked", () => { |
expect(screen.getByText(/Keys Mismatch - Problem Node Long/)).toBeInTheDocument(); |
||||
render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
expect(screen.getByText(/Your node is unable to send a direct message to node: Problem Node Long \(ProbNode\)/)).toBeInTheDocument(); |
||||
fireEvent.click(screen.getByRole('button', { name: /dismiss/i })); |
expect(screen.getByRole("button", { name: "Request New Keys" })).toBeInTheDocument(); |
||||
expect(mockHandleCloseDialog).toHaveBeenCalledTimes(1); |
expect(screen.getByRole("button", { name: "Dismiss" })).toBeInTheDocument(); |
||||
}); |
}); |
||||
|
|
||||
it("should call handleCloseDialog when the explicit DialogClose button is clicked", () => { |
test("does not render dialog if no error exists for active chat", () => { |
||||
render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
const deviceId = 1; |
||||
fireEvent.click(screen.getByRole('button', { name: /close/i })); // Use the aria-label
|
const activeChatNum = 54321; |
||||
expect(mockHandleCloseDialog).toHaveBeenCalledTimes(1); |
|
||||
}); |
|
||||
|
|
||||
|
useDeviceStore.getState().addDevice(deviceId); |
||||
|
|
||||
it("should not render the dialog when open is false", () => { |
const currentDeviceState = useDeviceStore.getState().getDevice(deviceId); |
||||
render(<RefreshKeysDialog open={false} onOpenChange={onOpenChangeMock} />); |
if (!currentDeviceState) throw new Error("Device not found"); |
||||
expect(screen.queryByRole("dialog")).not.toBeInTheDocument(); |
|
||||
}); |
|
||||
|
|
||||
it("should render null if nodeErrorNum is not found for activeChat", () => { |
mockUseMessageStore.mockReturnValue({ activeChat: activeChatNum }); |
||||
vi.mocked(useDevice).mockReturnValue({ |
mockUseRefreshKeysDialog.mockReturnValue({ |
||||
nodeErrors: new Map(), |
handleCloseDialog: vi.fn(), |
||||
nodes: mockNodes, |
handleNodeRemove: vi.fn(), |
||||
}); |
|
||||
const { container } = render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
|
||||
expect(container.firstChild).toBeNull(); |
|
||||
}); |
}); |
||||
|
|
||||
it("should render null if nodeWithError is not found for nodeErrorNum.node", () => { |
const { container } = render( |
||||
vi.mocked(useDevice).mockReturnValue({ |
<DeviceContext.Provider value={currentDeviceState}> |
||||
nodeErrors: mockNodeErrors, |
<RefreshKeysDialog open onOpenChange={vi.fn()} /> |
||||
nodes: new Map(), |
</DeviceContext.Provider> |
||||
}); |
); |
||||
const { container } = render(<RefreshKeysDialog open onOpenChange={onOpenChangeMock} />); |
|
||||
expect(container.firstChild).toBeNull(); |
expect(container.firstChild).toBeNull(); |
||||
}); |
|
||||
}); |
}); |
||||
Loading…
Reference in new issue