fix: fix editor.remove and add a test

geops · May 14, 2024 · 1427d6e · 1427d6e
1 parent 89c470f
commit 1427d6e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/editor.js b/src/editor.js
@@ -115,7 +115,7 @@ class Editor {
   remove() {
     const controls = [...this.controls.getArray()];
     controls.forEach((control) => {
-      control.removeControl(control);
+      this.removeControl(control);
     });
     if (this.toolbar) {
       this.toolbar.destroy();

diff --git a/src/editor.test.js b/src/editor.test.js
@@ -44,4 +44,18 @@ describe('editor', () => {
     expect(cad.editor).toBe();
     expect(cad.getActive()).toBe(false);
   });
+
+  test('is removed', () => {
+    editor.addControl(cad);
+    cad.activate();
+    expect(cad.getActive()).toBe(true);
+    expect(editor.controls.getArray()[0]).toBe(cad);
+    expect(editor.activeControls.getArray()[0]).toBe(cad);
+    editor.remove();
+    expect(editor.controls.getLength()).toBe(0);
+    expect(editor.activeControls.getLength()).toBe(0);
+    expect(cad.map).toBe();
+    expect(cad.editor).toBe();
+    expect(cad.getActive()).toBe(false);
+  });
 });