box-ui-elements-mlh/src/features/pagination/__tests__/PaginationMenu.test.js

Box UI Elements

import React from 'react';
import { shallow } from 'enzyme';
import MenuItem from '../../../components/menu/MenuItem';
import PaginationMenu from '../PaginationMenu';

describe('elements/Pagination/PaginationMenu', () => {
    test.each([1, 5, 10])('should render a button and menu with %i menu items', pageCount => {
        const wrapper = shallow(<PaginationMenu onPageClick={jest.fn()} pageCount={pageCount} pageNumber={1} />);

        expect(wrapper).toMatchSnapshot();
    });

    describe('page click handler', () => {
        test('should return the page number when an item is clicked', () => {
            const onClick = jest.fn();
            const wrapper = shallow(<PaginationMenu onPageClick={onClick} pageCount={10} pageNumber={1} />);

            wrapper
                .find(MenuItem)
                .at(2)
                .simulate('click');

            expect(onClick).toBeCalledWith(3);
        });
    });
});