Made all tests pass

Author: mbj36

.babelrc

@@ -1,3 +1,20 @@
 {
-  "presets": ["env", "react"]
+  "presets": [
+    [
+      "es2015",
+      {
+        "modules": false
+      }
+    ],
+    "react",
+    "stage-0"
+  ],
+  "env": {
+    "start": {
+      "presets": ["react-hmre"]
+    },
+    "test": {
+      "presets": ["es2015", "react", "stage-0"]
+    }
+  }
 }

package.json

@@ -4,8 +4,7 @@
   "description": "React component for syntax highlighting",
   "main": "index.js",
   "scripts": {
-    "prepublish":
-      "babel ./src --out-dir ./lib --plugins=transform-class-properties,transform-react-jsx --presets=env",
+    "prepublish": "babel ./src --out-dir ./lib --plugins=transform-class-properties,transform-react-jsx --presets=env",
     "test": "jest"
   },
   "repository": {
@@ -39,7 +38,9 @@
     "babel-plugin-transform-class-properties": "^6.24.1",
     "babel-plugin-transform-react-jsx": "^6.24.1",
     "babel-preset-env": "^1.6.1",
+    "babel-preset-es2015": "^6.24.1",
     "babel-preset-react": "^6.24.1",
+    "babel-preset-stage-0": "^6.24.1",
     "eslint": "^3.19.0",
     "eslint-plugin-react": "^6.10.3",
     "html-loader": "^0.4.5",

test/highlight.test.js

@@ -1,54 +1,55 @@
-import Highlight from '../src';
-import ReactDOM from 'react-dom';
-import TestUtils from 'react-dom/test-utils';
-import ReactDOMServer from 'react-dom/server';
+import Highlight from '../src'
+import ReactDOM from 'react-dom'
+import TestUtils from 'react-dom/test-utils'
+import ReactDOMServer from 'react-dom/server'
+import React from 'react'
 
 describe('highlight', () => {
-    it('should display test inside it', () => {
-        const text = TestUtils.renderIntoDocument(
-            <Highlight>Some text</Highlight>
-        );
-
-        expect(ReactDOM.findDOMNode(text).textContent).toBe('Some text');
-    });
-
-    it('should have pre and code tags in markup', () => {
-        const text = ReactDOMServer.renderToStaticMarkup(
-            <Highlight>Some text</Highlight>
-        );
-
-        expect(text).toBe('<pre><code>Some text</code></pre>');
-    });
-
-    it('should assign className prop', () => {
-        const text = ReactDOMServer.renderToStaticMarkup(
-            <Highlight className='html'>Some text</Highlight>
-        );
-
-        expect(text).toBe('<pre><code class="html">Some text</code></pre>');
-    });
-
-    it('should render children in span', () => {
-        const text = ReactDOMServer.renderToStaticMarkup(
-            <Highlight element='span'>Some text</Highlight>
-        );
-
-        expect(text).toBe('<span>Some text</span>');
-    });
-
-    it('should render innerHTML in span', () => {
-        const text = ReactDOMServer.renderToStaticMarkup(
-            <Highlight innerHTML={true} element='span'>Some text</Highlight>
-        );
-
-        expect(text).toBe('<span>Some text</span>');
-    });
-
-    it('should accept innerHTML prop', () => {
-        const text = TestUtils.renderIntoDocument(
-            <Highlight innerHTML={true}>{"<div>Sometext</div>"}</Highlight>
-        );
-
-        expect(ReactDOM.findDOMNode(text).textContent).toBe('Sometext');
-    });
-});
+  test('should display text inside it', () => {
+    const text = TestUtils.renderIntoDocument(<Highlight>Some text</Highlight>)
+
+    expect(ReactDOM.findDOMNode(text).textContent).toBe('Some text')
+  })
+
+  test('should have pre and code tags in markup', () => {
+    const text = ReactDOMServer.renderToStaticMarkup(
+      <Highlight>Some text</Highlight>
+    )
+
+    expect(text).toBe('<pre><code>Some text</code></pre>')
+  })
+
+  test('should assign className prop', () => {
+    const text = ReactDOMServer.renderToStaticMarkup(
+      <Highlight className="html">Some text</Highlight>
+    )
+
+    expect(text).toBe('<pre><code class="html">Some text</code></pre>')
+  })
+
+  test('should render children in span', () => {
+    const text = ReactDOMServer.renderToStaticMarkup(
+      <Highlight element="span">Some text</Highlight>
+    )
+
+    expect(text).toBe('<span>Some text</span>')
+  })
+
+  test('should render innerHTML in span', () => {
+    const text = ReactDOMServer.renderToStaticMarkup(
+      <Highlight innerHTML={true} element="span">
+        Some text
+      </Highlight>
+    )
+
+    expect(text).toBe('<span>Some text</span>')
+  })
+
+  test('should accept innerHTML prop', () => {
+    const text = TestUtils.renderIntoDocument(
+      <Highlight innerHTML={true}>{'<div>Sometext</div>'}</Highlight>
+    )
+
+    expect(ReactDOM.findDOMNode(text).textContent).toBe('Sometext')
+  })
+})