1 files changed, 18 insertions, 4 deletions
diff --git a/spec/javascript/components/avatar.test.jsx b/spec/javascript/components/avatar.test.jsx
index 79b7d02f4..a0f67e700 100644
--- a/spec/javascript/components/avatar.test.jsx
+++ b/spec/javascript/components/avatar.test.jsx
@@ -4,9 +4,23 @@ import { render } from 'enzyme';
 import Avatar from '../../../app/assets/javascripts/components/components/avatar'
 
 describe('<Avatar />', () => {
-  it('renders an img with the given src', () => {
-    const src = '/path/to/image.jpg';
-    const wrapper = render(<Avatar src={src} size={100} />);
-    expect(wrapper.find(`img[src="${src}"]`)).to.have.length(1);
+  const src = '/path/to/image.jpg';
+  const size = 100;
+  const wrapper = render(<Avatar src={src} size={100} />);
+
+  it('renders an img element with the given src', () => {
+    expect(wrapper.find('img')).to.have.attr('src', `${src}`);
+  });
+
+  it('renders an img element of the given size', () => {
+    ['width', 'height'].map((attr) => {
+      expect(wrapper.find('img')).to.have.attr(attr, `${size}`);
+    });
+  });
+
+  it('renders a div element of the given size', () => {
+    ['width', 'height'].map((attr) => {
+      expect(wrapper.find('div')).to.have.style(attr, `${size}px`);
+    });
   });
 });