Merge pull request #31 from pat310/add-toggle-function

add toggle function
pat310 · Mar 11, 2017 · 13ee601 · 13ee601
2 parents 6f56fe3 + 1b2f075
commit 13ee601
Show file tree

Hide file tree

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

diff --git a/test/index.spec.js b/test/index.spec.js
@@ -348,6 +348,22 @@ describe('pivot', () => {
       expect(pivot.getData(1)).to.deep.equal(uncollapsedData);
     });
 
+    it('should return table to normal state when toggled twice', () => {
+      const pivot = new Pivot(
+        dataArray,
+        rowsToPivot,
+        colsToPivot,
+        aggregationCategory,
+        aggregationType,
+      );
+
+      const startState = pivot.data.table;
+
+      pivot.toggle(1).toggle(1);
+
+      expect(startState).to.deep.equal(pivot.data.table);
+    });
+
     it('should expand correct rows', () => {
       const pivot = new Pivot(
         dataArray,