|
| 1 | +import { cleanup, render, waitFor } from "@solidjs/testing-library"; |
| 2 | +import { createSignal } from "solid-js"; |
| 3 | +import uPlot from "uplot"; |
| 4 | +import { afterEach, beforeEach, describe, expect, test, vi } from "vitest"; |
| 5 | + |
| 6 | +import { SolidUplot } from "../SolidUplot"; |
| 7 | + |
| 8 | +const DEFAULT_DATA: uPlot.AlignedData = [ |
| 9 | + [1, 2, 3, 4, 5], // x-values |
| 10 | + [10, 20, 30, 40, 50], // y-values for the first series |
| 11 | +]; |
| 12 | + |
| 13 | +const DEFAULT_OPTIONS: uPlot.Options = { |
| 14 | + width: 400, |
| 15 | + height: 300, |
| 16 | + series: [ |
| 17 | + {}, |
| 18 | + { |
| 19 | + label: "Series 1", |
| 20 | + stroke: "red", |
| 21 | + }, |
| 22 | + ], |
| 23 | +}; |
| 24 | + |
| 25 | +describe("COMPONENT: <SolidUplot />", () => { |
| 26 | + beforeEach(() => { |
| 27 | + vi.clearAllMocks(); |
| 28 | + }); |
| 29 | + |
| 30 | + afterEach(() => { |
| 31 | + cleanup(); |
| 32 | + }); |
| 33 | + |
| 34 | + test("renders with default props", async () => { |
| 35 | + const onCreateSpy = vi.fn(); |
| 36 | + |
| 37 | + const { container } = render(() => ( |
| 38 | + <SolidUplot {...DEFAULT_OPTIONS} data={DEFAULT_DATA} onCreate={onCreateSpy} /> |
| 39 | + )); |
| 40 | + |
| 41 | + await waitFor(() => expect(onCreateSpy).toHaveBeenCalled()); |
| 42 | + |
| 43 | + expect(container.querySelector("div")).not.toBeNull(); |
| 44 | + }); |
| 45 | + |
| 46 | + test("renders children inside the container", () => { |
| 47 | + const { getByText } = render(() => ( |
| 48 | + <SolidUplot id="test-id" {...DEFAULT_OPTIONS} data={DEFAULT_DATA}> |
| 49 | + <div data-testid="child-element">Child content</div> |
| 50 | + </SolidUplot> |
| 51 | + )); |
| 52 | + |
| 53 | + expect(getByText("Child content")).toBeInTheDocument(); |
| 54 | + }); |
| 55 | + |
| 56 | + test("passes props to the container", () => { |
| 57 | + const { container } = render(() => ( |
| 58 | + <SolidUplot |
| 59 | + {...DEFAULT_OPTIONS} |
| 60 | + id="test-id" |
| 61 | + class="test-class" |
| 62 | + style={{ width: "500px", height: "300px" }} |
| 63 | + data={DEFAULT_DATA} |
| 64 | + /> |
| 65 | + )); |
| 66 | + |
| 67 | + const div = container.querySelector("div"); |
| 68 | + |
| 69 | + expect(div).toHaveAttribute("id", "test-id"); |
| 70 | + expect(div).toHaveAttribute("class", "test-class"); |
| 71 | + expect(div).toHaveStyle({ width: "500px", height: "300px" }); |
| 72 | + }); |
| 73 | + |
| 74 | + test("calls onCreate callback when plot is created", () => { |
| 75 | + const onCreateMock = vi.fn(); |
| 76 | + |
| 77 | + render(() => <SolidUplot {...DEFAULT_OPTIONS} data={DEFAULT_DATA} onCreate={onCreateMock} />); |
| 78 | + |
| 79 | + expect(onCreateMock).toHaveBeenCalledTimes(1); |
| 80 | + expect(onCreateMock).toHaveBeenCalledWith(expect.any(Object), expect.any(HTMLDivElement)); |
| 81 | + }); |
| 82 | + |
| 83 | + test("reactively updates the chart size when width or height changes", async () => { |
| 84 | + let chartInstance: uPlot; |
| 85 | + const onCreateSpy = vi.fn((chart: uPlot) => { |
| 86 | + chartInstance = chart; |
| 87 | + }); |
| 88 | + const [dimensions, setDimensions] = createSignal({ width: 400, height: 300 }); |
| 89 | + |
| 90 | + render(() => ( |
| 91 | + <SolidUplot |
| 92 | + {...DEFAULT_OPTIONS} |
| 93 | + data={DEFAULT_DATA} |
| 94 | + width={dimensions().width} |
| 95 | + height={dimensions().height} |
| 96 | + onCreate={onCreateSpy} |
| 97 | + /> |
| 98 | + )); |
| 99 | + |
| 100 | + await waitFor(() => expect(chartInstance).toBeDefined()); |
| 101 | + |
| 102 | + expect(chartInstance!.width).toBe(400); |
| 103 | + expect(chartInstance!.height).toBe(300); |
| 104 | + |
| 105 | + setDimensions({ width: 500, height: 400 }); // Trigger reactive update |
| 106 | + |
| 107 | + expect(chartInstance!.width).toBe(500); |
| 108 | + expect(chartInstance!.height).toBe(400); |
| 109 | + |
| 110 | + expect(onCreateSpy).toHaveBeenCalledTimes(1); |
| 111 | + }); |
| 112 | + |
| 113 | + test("reactively updates the chart data when data changes", async () => { |
| 114 | + let chartInstance: uPlot; |
| 115 | + |
| 116 | + const onCreateSpy = vi.fn((chart: uPlot) => { |
| 117 | + chartInstance = chart; |
| 118 | + }); |
| 119 | + |
| 120 | + const [chartData, setChartData] = createSignal<uPlot.AlignedData>(DEFAULT_DATA); |
| 121 | + |
| 122 | + render(() => <SolidUplot {...DEFAULT_OPTIONS} data={chartData()} onCreate={onCreateSpy} />); |
| 123 | + |
| 124 | + await waitFor(() => expect(chartInstance).toBeDefined()); |
| 125 | + |
| 126 | + expect(chartInstance!.data).toEqual(DEFAULT_DATA); |
| 127 | + |
| 128 | + const newData: uPlot.AlignedData = [ |
| 129 | + [1, 2, 3, 4, 5], |
| 130 | + [15, 25, 35, 45, 55], |
| 131 | + ]; |
| 132 | + |
| 133 | + setChartData(newData); // Trigger reactive update |
| 134 | + |
| 135 | + expect(chartInstance!.data).toEqual(newData); |
| 136 | + }); |
| 137 | + |
| 138 | + test("reactively applies resetScales option when updating data", async () => { |
| 139 | + let chartInstance: uPlot; |
| 140 | + |
| 141 | + const onCreateSpy = vi.fn((chart: uPlot) => { |
| 142 | + chartInstance = chart; |
| 143 | + }); |
| 144 | + |
| 145 | + const [chartData, setChartData] = createSignal<uPlot.AlignedData>(DEFAULT_DATA); |
| 146 | + const [resetScales, setResetScales] = createSignal(false); |
| 147 | + |
| 148 | + render(() => ( |
| 149 | + <SolidUplot |
| 150 | + {...DEFAULT_OPTIONS} |
| 151 | + data={chartData()} |
| 152 | + resetScales={resetScales()} |
| 153 | + onCreate={onCreateSpy} |
| 154 | + /> |
| 155 | + )); |
| 156 | + |
| 157 | + await waitFor(() => expect(chartInstance).toBeDefined()); |
| 158 | + |
| 159 | + vi.spyOn(chartInstance!, "setData"); |
| 160 | + |
| 161 | + // New data for the update |
| 162 | + const newData: uPlot.AlignedData = [ |
| 163 | + [1, 2, 3, 4, 5], |
| 164 | + [15, 25, 35, 45, 55], |
| 165 | + ]; |
| 166 | + |
| 167 | + setChartData(newData); |
| 168 | + |
| 169 | + expect(chartInstance!.setData).toHaveBeenCalledWith(newData, false); |
| 170 | + |
| 171 | + setResetScales(true); |
| 172 | + |
| 173 | + expect(chartInstance!.setData).toHaveBeenCalledWith(newData, true); |
| 174 | + |
| 175 | + expect(onCreateSpy).toHaveBeenCalledTimes(1); |
| 176 | + }); |
| 177 | + |
| 178 | + test("destroys the chart when component is unmounted", async () => { |
| 179 | + let chartInstance: uPlot; |
| 180 | + |
| 181 | + const onCreateSpy = vi.fn((chart: uPlot) => { |
| 182 | + chartInstance = chart; |
| 183 | + }); |
| 184 | + |
| 185 | + const { unmount } = render(() => ( |
| 186 | + <SolidUplot {...DEFAULT_OPTIONS} data={DEFAULT_DATA} onCreate={onCreateSpy} /> |
| 187 | + )); |
| 188 | + |
| 189 | + await waitFor(() => expect(onCreateSpy).toHaveBeenCalled()); |
| 190 | + |
| 191 | + vi.spyOn(chartInstance!, "destroy"); |
| 192 | + |
| 193 | + unmount(); |
| 194 | + |
| 195 | + expect(chartInstance!.destroy).toHaveBeenCalledTimes(1); |
| 196 | + }); |
| 197 | + |
| 198 | + test("creates a new chart when a non-updateable prop changes (e.g. series)", async () => { |
| 199 | + const onCreateSpy = vi.fn(); |
| 200 | + |
| 201 | + const initialSeries = [ |
| 202 | + { |
| 203 | + label: "Series 1", |
| 204 | + stroke: "red", |
| 205 | + }, |
| 206 | + ] as uPlot.Series[]; |
| 207 | + |
| 208 | + const [data, setData] = createSignal<uPlot.AlignedData>(DEFAULT_DATA); |
| 209 | + const [series, setSeries] = createSignal<uPlot.Series[]>(initialSeries); |
| 210 | + |
| 211 | + render(() => ( |
| 212 | + <SolidUplot {...DEFAULT_OPTIONS} data={data()} series={series()} onCreate={onCreateSpy} /> |
| 213 | + )); |
| 214 | + |
| 215 | + await waitFor(() => expect(onCreateSpy).toHaveBeenCalledTimes(1)); |
| 216 | + |
| 217 | + setData([...DEFAULT_DATA, [5, 15, 25, 35, 45]]); |
| 218 | + |
| 219 | + setSeries([ |
| 220 | + ...initialSeries, |
| 221 | + { |
| 222 | + label: "Series 2", |
| 223 | + stroke: "blue", |
| 224 | + }, |
| 225 | + ]); |
| 226 | + |
| 227 | + await waitFor(() => expect(onCreateSpy).toHaveBeenCalledTimes(2)); |
| 228 | + }); |
| 229 | +}); |
0 commit comments