Updating Message tests to use enzyme

src/components/Message.test.js

@@ -1,25 +1,23 @@
 import React from 'react';
-import renderer from 'react-test-renderer';
+import { shallow, render } from 'enzyme';
 
 import Message from './Message';
 
 test('Message rendering', () => {
-  const component = renderer.create(
+  const component = shallow(
     <Message heading="Testing">
       <p>Message content</p>
     </Message>
   );
-  let tree = component.toJSON();
+  expect(component).toMatchSnapshot();
-  expect(tree).toMatchSnapshot();
 });
 
 test('Message rendering with icon', () => {
   const Icon = () => 'Sample icon SVG';
-  const component = renderer.create(
+  const component = render(
     <Message heading="Testing" icon={ Icon }>
       <p>Message content</p>
     </Message>
   );
-  let tree = component.toJSON();
+  expect(component).toMatchSnapshot();
-  expect(tree).toMatchSnapshot();
 });

src/components/__snapshots__/Message.test.js.snap

@@ -15,11 +15,10 @@ exports[`Message rendering 1`] = `
 
 exports[`Message rendering with icon 1`] = `
 <div
-  className="message"
+  class="message"
 >
   <h2>
-    Sample icon SVG
+    Sample icon SVGTesting
-    Testing
   </h2>
   <p>
     Message content