materialuiupgraded/packages/material-ui/src/ListItemAvatar/ListItemAvatar.test.js

Material-UI's workspace package

import React from 'react';
import { assert } from 'chai';
import consoleErrorMock from 'test/utils/consoleErrorMock';
import { createShallow, getClasses } from '../test-utils';
import Avatar from '../Avatar';
import ListItemAvatar from './ListItemAvatar';

describe('<ListItemAvatar />', () => {
  let shallow;
  let classes;

  before(() => {
    shallow = createShallow({ dive: true });
    classes = getClasses(
      <ListItemAvatar className="foo">
        <Avatar className="bar" />
      </ListItemAvatar>,
      { context: { dense: true } },
    );
  });

  it('should render with the user and root classes', () => {
    const wrapper = shallow(
      <ListItemAvatar className="foo">
        <Avatar className="bar" />
      </ListItemAvatar>,
      {
        context: {
          dense: true,
        },
      },
    );
    assert.strictEqual(wrapper.hasClass('foo'), true);
    assert.strictEqual(wrapper.hasClass('bar'), true);
    assert.strictEqual(wrapper.hasClass(classes.root), true);
  });

  describe('List', () => {
    before(() => {
      consoleErrorMock.spy();
    });

    after(() => {
      consoleErrorMock.reset();
    });

    it('should render an Avatar', () => {
      const wrapper = shallow(
        <ListItemAvatar>
          <Avatar />
        </ListItemAvatar>,
        {
          context: {
            dense: true,
          },
        },
      );
      assert.strictEqual(wrapper.type(), Avatar);
      assert.strictEqual(consoleErrorMock.callCount(), 0);
    });

    it('should warn in a wrong context', () => {
      shallow(
        <ListItemAvatar>
          <Avatar />
        </ListItemAvatar>,
      );
      assert.strictEqual(consoleErrorMock.callCount(), 1);
    });
  });
});