File size: 2,748 Bytes
52efc7b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { describe, it, expect } from 'vitest';
import { renderHook } from '../../test-utils/render.js';
import { useBatchedScroll } from './useBatchedScroll.js';

describe('useBatchedScroll', () => {
  it('returns initial scrollTop', async () => {
    const { result } = await renderHook(() => useBatchedScroll(10));
    expect(result.current.getScrollTop()).toBe(10);
  });

  it('returns updated scrollTop from props', async () => {
    let currentScrollTop = 10;
    const { result, rerender } = await renderHook(() =>
      useBatchedScroll(currentScrollTop),
    );

    expect(result.current.getScrollTop()).toBe(10);

    currentScrollTop = 100;
    rerender();

    expect(result.current.getScrollTop()).toBe(100);
  });

  it('returns pending scrollTop when set', async () => {
    const { result } = await renderHook(() => useBatchedScroll(10));

    result.current.setPendingScrollTop(50);
    expect(result.current.getScrollTop()).toBe(50);
  });

  it('overwrites pending scrollTop with subsequent sets before render', async () => {
    const { result } = await renderHook(() => useBatchedScroll(10));

    result.current.setPendingScrollTop(50);
    result.current.setPendingScrollTop(75);
    expect(result.current.getScrollTop()).toBe(75);
  });

  it('resets pending scrollTop after rerender', async () => {
    let currentScrollTop = 10;
    const { result, rerender } = await renderHook(() =>
      useBatchedScroll(currentScrollTop),
    );

    result.current.setPendingScrollTop(50);
    expect(result.current.getScrollTop()).toBe(50);

    // Rerender with new prop
    currentScrollTop = 100;
    rerender();

    // Should now be the new prop value, pending should be cleared
    expect(result.current.getScrollTop()).toBe(100);
  });

  it('resets pending scrollTop after rerender even if prop is same', async () => {
    const { result, rerender } = await renderHook(() => useBatchedScroll(10));

    result.current.setPendingScrollTop(50);
    expect(result.current.getScrollTop()).toBe(50);

    // Rerender with same prop
    rerender();

    // Pending should still be cleared because useEffect runs after every render
    expect(result.current.getScrollTop()).toBe(10);
  });

  it('maintains stable function references', async () => {
    const { result, rerender } = await renderHook(() => useBatchedScroll(10));
    const initialGetScrollTop = result.current.getScrollTop;
    const initialSetPendingScrollTop = result.current.setPendingScrollTop;

    rerender();

    expect(result.current.getScrollTop).toBe(initialGetScrollTop);
    expect(result.current.setPendingScrollTop).toBe(initialSetPendingScrollTop);
  });
});