import React from 'react';
import { render } from '@testing-library/react';
import TableCellLoader from '../table-cell-loader';

describe('<TableCellLoader>', () => {
    test('renders correctly when no variant specified', () => {
        render(<TableCellLoader />);

        expect(document.querySelector('.TableCellLoader-item')).toBeInTheDocument();
    });

    test('renders with variant when specified', () => {
        render(<TableCellLoader variant="sm" />);

        expect(document.querySelector('.TableCellLoader-item-sm')).toBeInTheDocument();
    });
});