Adding the Image component tests

src/__tests__/image.test.tsx

@@ -0,0 +1,63 @@
+import * as React from 'react';
+import Image from '../image';
+import { getMapMock } from '../jest/util';
+import { mount } from 'enzyme';
+
+describe('Image', () => {
+  it('Should add image on mount', () => {
+    const mapMock = getMapMock();
+    const onLoaded = jest.fn();
+    const onError = jest.fn();
+
+    const imageId = 'image';
+    const imageData = {};
+    const imageOptions = {};
+
+    mount(
+      <Image
+        id={imageId}
+        map={mapMock}
+        data={imageData}
+        options={imageOptions}
+        onError={onError}
+        onLoaded={onLoaded}
+      />
+    );
+
+    expect(mapMock.addImage.mock.calls[0]).toEqual([
+      imageId,
+      imageData,
+      imageOptions
+    ]);
+
+    expect(onLoaded).toBeCalled();
+    expect(onError).not.toBeCalled();
+  });
+
+  it('Should remove image on unmount', () => {
+    const mapMock = getMapMock();
+    const onLoaded = jest.fn();
+    const onError = jest.fn();
+
+    const imageId = 'image';
+    const imageData = {};
+    const imageOptions = {};
+
+    const component = mount(
+      <Image
+        id={imageId}
+        map={mapMock}
+        data={imageData}
+        options={imageOptions}
+        onError={onError}
+        onLoaded={onLoaded}
+      />
+    );
+
+    expect(mapMock.addImage).toBeCalled();
+    expect(onLoaded).toBeCalled();
+
+    component.unmount();
+    expect(mapMock.removeImage).toBeCalled();
+  });
+});

src/jest/util.tsx

@@ -13,6 +13,8 @@ export const getMapMock = (override?: { [key: string]: any }) => ({
   setLayerZoomRange: jest.fn(),
   getLayer: jest.fn(),
   addImage: jest.fn(),
+  loadImage: jest.fn(),
+  removeImage: jest.fn(),
   hasImage: jest.fn(),
   getSource: jest.fn().mockReturnValue({ setData: jest.fn() }),
   project: jest.fn(),