Chore: Add tests for annotationsLoadHandler (#540)

box · Dec 11, 2017 · ae68bcb · ae68bcb
1 parent 062ea38
commit ae68bcb
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/src/lib/viewers/__tests__/BaseViewer-test.js b/src/lib/viewers/__tests__/BaseViewer-test.js
@@ -342,7 +342,7 @@ describe('lib/viewers/BaseViewer', () => {
                 scale: 1.5
             });
 
-            base.annotationsLoadPromise.then(() => {
+            return base.annotationsLoadPromise.then(() => {
                 expect(base.annotationsLoadHandler).to.be.called;
             });
         });
@@ -782,6 +782,35 @@ describe('lib/viewers/BaseViewer', () => {
         });
     });
 
+    describe('annotationsLoadHandler()', () => {
+        const conf = {
+            annotationsEnabled: true,
+            types: {
+                point: true,
+                highlight: false
+            }
+        };
+
+        beforeEach(() => {
+            window.BoxAnnotations = function BoxAnnotations() {
+                this.determineAnnotator = sandbox.stub().returns(conf);
+            }
+
+            sandbox.stub(base, 'initAnnotations');
+        });
+
+        it('should determine the annotator', () => {
+            base.annotationsLoadHandler();
+            expect(base.annotatorConf).to.equal(conf);
+        });
+
+        it('should init annotations if a conf is present', () => {
+            base.annotationsLoadHandler();
+            expect(base.initAnnotations).to.be.called;
+
+        });
+    });
+
     describe('initAnnotations()', () => {
         beforeEach(() => {
             base.options = {