test(createViewer): Test .setImage

Author: thewtex

src/createViewer.js

@@ -140,7 +140,7 @@ const createViewer = (
   }
 
   publicAPI.setImage = (image) => {
-    imageSource.SetInputData(image);
+    imageSource.setInputData(image);
     transferFunctionWidget.setDataArray(image.getPointData().getScalars());
   }
 

test/createViewerTest.js

@@ -9,8 +9,10 @@ import createViewer from '../src/createViewer'
 import userInterface from '../src/userInterface'
 
 const testImage3DPath = 'base/test/data/input/HeadMRVolume.nrrd'
+const testImage3DPath2 = 'base/test/data/input/mri3D.nrrd'
 
 import createViewerBaseline from './data/baseline/createViewer.png'
+import createViewerSetImageBaseline from './data/baseline/createViewerSetImage.png'
 
 const TEST_STYLE_CONTAINER = {
   position: 'relative',
@@ -29,6 +31,7 @@ const TEST_VIEWER_STYLE = {
   containerStyle: TEST_STYLE_CONTAINER,
 }
 
+
 test('Test createViewer', (t) => {
   const gc = testUtils.createGarbageCollector(t);
 
@@ -51,3 +54,38 @@ test('Test createViewer', (t) => {
       viewer.renderLater()
     })
 })
+
+
+test('Test createViewer.setImage', (t) => {
+  const gc = testUtils.createGarbageCollector(t);
+
+  const container = document.querySelector('body')
+  const viewerContainer = gc.registerDOMElement(document.createElement('div'))
+  container.appendChild(viewerContainer)
+
+  return axios.get(testImage3DPath, {responseType: 'blob'})
+    .then(function (response) {
+      return itkreadImageBlob(null, response.data, testImage3DPath)
+      })
+    .then(({ image: itkImage, webWorker }) => {
+      webWorker.terminate()
+
+      const imageData = vtkITKHelper.convertItkToVtkImage(itkImage)
+      const viewer = createViewer(container, { image: imageData, viewerStyle: TEST_VIEWER_STYLE })
+      return axios.get(testImage3DPath2, {responseType: 'blob'})
+        .then(function (response) {
+          return itkreadImageBlob(null, response.data, testImage3DPath2)
+          })
+        .then(({ image: itkImage, webWorker }) => {
+          webWorker.terminate()
+
+          const imageData = vtkITKHelper.convertItkToVtkImage(itkImage)
+          viewer.setImage(imageData)
+          viewer.captureImage().then((screenshot) => {
+            testUtils.compareImages(screenshot, [createViewerSetImageBaseline], 'Test createViewer.setImage', t, 1.0, gc.releaseResources)
+        })
+
+        })
+      viewer.renderLater()
+    })
+})