diff --git a/package.json b/package.json
index 133244650f..3d1878e15a 100644
--- a/package.json
+++ b/package.json
@@ -15,6 +15,7 @@
"browserify": "^13.1.0",
"browserify-incremental": "^3.1.1",
"chai": "^3.5.0",
+ "chai-enzyme": "^0.5.2",
"enzyme": "^2.4.1",
"es6-promise": "^3.2.1",
"immutable": "^3.8.1",
diff --git a/spec/javascript/components/avatar.test.jsx b/spec/javascript/components/avatar.test.jsx
index 79b7d02f46..852e13a89c 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('', () => {
- it('renders an img with the given src', () => {
- const src = '/path/to/image.jpg';
- const wrapper = render();
- expect(wrapper.find(`img[src="${src}"]`)).to.have.length(1);
+ const src = '/path/to/image.jpg';
+ const size = 100;
+ const wrapper = render();
+
+ 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`);
+ });
});
});
diff --git a/spec/javascript/setup.js b/spec/javascript/setup.js
index 636cdcc7e3..0d07f74d88 100644
--- a/spec/javascript/setup.js
+++ b/spec/javascript/setup.js
@@ -1,3 +1,7 @@
+import chai from 'chai';
+import chaiEnzyme from 'chai-enzyme';
+chai.use(chaiEnzyme());
+
/**
* http://airbnb.io/enzyme/docs/guides/jsdom.html
*/