fix(discoverability): add getInitialAnnotationMode override (#1273)

* fix(discoverability): add getInitialAnnotationMode override
box · Oct 21, 2020 · b64d8fc · b64d8fc
1 parent b42768a
commit b64d8fc
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/lib/viewers/image/ImageViewer.js b/src/lib/viewers/image/ImageViewer.js
@@ -164,6 +164,11 @@ class ImageViewer extends ImageBaseViewer {
         this.setScale(this.imageEl.offsetwidth, this.imageEl.offsetHeight);
     }
 
+    // Annotation overrides
+    getInitialAnnotationMode() {
+        return this.options.enableAnnotationsImageDiscoverability ? AnnotationMode.REGION : AnnotationMode.NONE;
+    }
+
     /**
      * Gets the width & height after the transforms are applied.
      * When an image is rotated, the width & height of an image, e.g. offsetWidth & offsetHeight,

diff --git a/src/lib/viewers/image/__tests__/ImageViewer-test.js b/src/lib/viewers/image/__tests__/ImageViewer-test.js
@@ -423,6 +423,21 @@ describe('lib/viewers/image/ImageViewer', () => {
         });
     });
 
+    describe('getInitialAnnotationMode()', () => {
+        test.each`
+            mode                     | enableAnnotationsImageDiscoverability
+            ${AnnotationMode.REGION} | ${true}
+            ${AnnotationMode.NONE}   | ${false}
+        `(
+            'should return $mode if enableAnnotationsImageDiscoverability is $enableAnnotationsImageDiscoverability',
+            ({ enableAnnotationsImageDiscoverability, mode }) => {
+                image.options.enableAnnotationsImageDiscoverability = enableAnnotationsImageDiscoverability;
+
+                expect(image.getInitialAnnotationMode()).toBe(mode);
+            },
+        );
+    });
+
     describe('getTransformWidthAndHeight', () => {
         test('should return the same width & height if the image is not rotated', () => {
             const width = 100;