Apply the same treatment to enable_test

doquoctoan · Sep 24, 2018 · 67e3ca2 · 67e3ca2
1 parent 32a6426
commit 67e3ca2
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/test/enable_test.js b/test/enable_test.js
@@ -20,14 +20,19 @@ describe('Enable a DOM Element for Canvas Renderer', function () {
   it('should fire CornerstoneElementEnabled', function (done) {
     const element = document.createElement('div');
 
+    element.id = 'target-element';
+
     // Assert
-    events.addEventListener('cornerstoneelementenabled', function (event) {
-      assert.equal(event.detail.element, element);
+    const handler = function (event) {
+      assert.equal(event.detail.element.getAttribute('id'), element.id);
 
-      // Cleanup
+      // Clean-up
+      events.removeEventListener('cornerstoneelementenabled', handler);
       disable(element);
       done();
-    });
+    };
+
+    events.addEventListener('cornerstoneelementenabled', handler);
     enable(element, {});
   });