Add rct-node-expanded and rct-node-collapsed classes to expanded and collapsed parent nodes

Resolves #114.

Author: jakezatecky

CHANGELOG.md

@@ -1,5 +1,11 @@
 # CHANGELOG
 
+## v.1.4.0 (TBA)
+
+### New Features
+
+* [#114]: Add `rct-node-expanded` and `rct-node-collapsed` classes to expanded and collapsed parent nodes
+
 ## [v1.3.1](https://github.com/jakezatecky/react-checkbox-tree/compare/v1.3.0...v1.3.1) (2018-09-06)
 
 ### Bug Fixes

src/js/TreeNode.js

@@ -275,11 +275,18 @@ class TreeNode extends React.Component {
     }
 
     render() {
-        const { className, disabled, isLeaf } = this.props;
+        const {
+            className,
+            disabled,
+            expanded,
+            isLeaf,
+        } = this.props;
         const nodeClass = classNames({
             'rct-node': true,
             'rct-node-leaf': isLeaf,
             'rct-node-parent': !isLeaf,
+            'rct-node-expanded': !isLeaf && expanded,
+            'rct-node-collapsed': !isLeaf && !expanded,
             'rct-disabled': disabled,
         }, className);
 

test/TreeNode.js

@@ -177,6 +177,31 @@ describe('<TreeNode />', () => {
             assert.isTrue(wrapper.contains(<span className="rct-icon rct-icon-expand-close" />));
             assert.isTrue(wrapper.contains(<span className="rct-icon rct-icon-parent-close" />));
         });
+
+        it('should append the `rct-node-expanded` class to the node when set to true', () => {
+            const wrapper = shallow(
+                <TreeNode {...baseProps} expanded isLeaf={false} />,
+            );
+
+            assert.isTrue(wrapper.find('li').hasClass('rct-node-expanded'));
+        });
+
+        it('should append the `rct-node-collapsed` class to the node when set to true', () => {
+            const wrapper = shallow(
+                <TreeNode {...baseProps} expanded={false} isLeaf={false} />,
+            );
+
+            assert.isTrue(wrapper.find('li').hasClass('rct-node-collapsed'));
+        });
+
+        it('should not append any expanded/collapsed classes to the node when a leaf', () => {
+            const wrapper = shallow(
+                <TreeNode {...baseProps} expanded isLeaf />,
+            );
+
+            assert.isFalse(wrapper.find('li').hasClass('rct-node-expanded'));
+            assert.isFalse(wrapper.find('li').hasClass('rct-node-collapsed'));
+        });
     });
 
     describe('icon', () => {