fix components getting className set incorrectly, it should be in pro…

…ps always
infernojs · Mar 24, 2017 · c203e28 · c203e28
1 parent 7400a3b
commit c203e28
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -213,7 +213,7 @@ function getVNodeProps(t, astProps, isComponent) {
 				propName = propName.namespace.name + ':' + propName.name.name;
 			}
 
-			if (propName === 'className' || propName === 'class') {
+			if (!isComponent && (propName === 'className' || propName === 'class')) {
 				className = getValue(t, astProp.value);
 			} else if (propName.substr(0, 11) === 'onComponent' && isComponent) {
 				if (!ref) {

diff --git a/tests.js b/tests.js
@@ -36,10 +36,14 @@ describe('Array', function() {
 			expect(transform('<div>1</div>')).to.equal('createVNode(2, "div", null, "1");');
 		});
 
-		it('className should be in third parameter as string', function () {
+		it('className should be in third parameter as string when its element', function () {
 			expect(transform('<div className="first second">1</div>')).to.equal('createVNode(2, "div", "first second", "1");');
 		});
 
+		it('className should be in fifth parameter as string when its component', function () {
+			expect(transform('<UnkownClass className="first second">1</UnkownClass>')).to.equal('createVNode(16, UnkownClass, null, null, {\n  "className": "first second",\n  children: "1"\n});');
+		});
+
 		it('class should be in third parameter as variable', function () {
 			expect(transform('<div class={variable}>1</div>')).to.equal('createVNode(2, "div", variable, "1");');
 		});