fix: Allow same option to be reselected right after it's deselected (#129) 🐛

Closes #128

Author: mrchief

docs/bundle.js

@@ -98,11 +98,18 @@ class TreeManager {
 
   togglePreviousChecked(id) {
     const prevChecked = this.currentChecked
-    if (prevChecked) this.getNodeById(prevChecked).checked = false
+
+    // if id is same as previously selected node, then do nothing (since it's state is already set correctly by setNodeCheckedState)
+    // but if they ar not same, then toggle the previous one
+    if (prevChecked && prevChecked !== id) this.getNodeById(prevChecked).checked = false
+
     this.currentChecked = id
+
+    console.table({ id, prevChecked, currentChecked: this.currentChecked })
   }
 
   setNodeCheckedState(id, checked) {
+    console.table({ id, checked })
     const node = this.getNodeById(id)
     node.checked = checked
 

src/tree-manager/index.js

@@ -0,0 +1,19 @@
+import test from 'ava'
+import TreeManager from '..'
+
+test('should select the same node if it is selected and then deselected right again', t => {
+  const tree = [{ id: 'nodeA' }, { id: 'nodeB' }, { id: 'nodeC' }]
+
+  const manager = new TreeManager(tree, true)
+
+  // first select a node
+  manager.setNodeCheckedState('nodeA', true)
+
+  // then deselect the same node
+  manager.setNodeCheckedState('nodeA', false)
+
+  // then reselect the same node right away
+  manager.setNodeCheckedState('nodeA', true)
+
+  t.true(manager.getNodeById('nodeA').checked)
+})