Merge 3cf84c8 into b19de9c

pat310 · Mar 10, 2017 · 2c4ea82 · 2c4ea82
2 parents b19de9c + 3cf84c8
commit 2c4ea82
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 3 deletions.
diff --git a/lib/quick-pivot.js b/lib/quick-pivot.js
diff --git a/lib/quick-pivot.js.map b/lib/quick-pivot.js.map
diff --git a/lib/quick-pivot.min.js b/lib/quick-pivot.min.js
diff --git a/lib/quick-pivot.min.js.map b/lib/quick-pivot.min.js.map
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,