Merge pull request #35 from pat310/fix-toggle-function

fix toggle function
pat310 · Mar 12, 2017 · ef6d542 · ef6d542
2 parents 5df4500 + e9fc59c
commit ef6d542
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -40,7 +40,7 @@ export default class Pivot {
   }
 
   toggle(rowNum) {
-    if (rowNum in this.collapsedRows) {
+    if (this.data.table[rowNum].row in this.collapsedRows) {
       return this.expand(rowNum);
     }
     return this.collapse(rowNum);

diff --git a/test/index.spec.js b/test/index.spec.js
@@ -361,7 +361,10 @@ describe('pivot', () => {
 
       const startState = pivot.data.table;
 
-      pivot.toggle(1).toggle(1);
+      pivot.toggle(1);
+      pivot.toggle(2);
+      pivot.toggle(2);
+      pivot.toggle(1);
 
       expect(startState).to.deep.equal(pivot.data.table);
     });