Generator - rearrange (#4961)

* Generator - rearrange

* Reove default prop

Author: EgoziE

.wuf/generator/templates/$ComponentName/$ComponentName.js

@@ -8,19 +8,6 @@ import { dataHooks } from './constants';
 
 /** <%= description %> */
 class <%= ComponentName %> extends React.PureComponent {
-  static displayName = '<%= ComponentName %>';
-
-  static propTypes = {
-    dataHook: PropTypes.string,
-
-    /** Text for the button */
-    buttonText: PropTypes.string,
-  };
-
-  static defaultProps = {
-    buttonText: 'Click me!',
-  };
-
   state = {
     count: 0,
   };
@@ -37,7 +24,10 @@ class <%= ComponentName %> extends React.PureComponent {
     const isEven = count % 2 === 0;
 
     return (
-      <div {...styles('root', { even: isEven, odd: !isEven }, this.props)} data-hook={dataHook}>
+      <div
+        {...styles('root', { even: isEven, odd: !isEven }, this.props)}
+        data-hook={dataHook}
+      >
         <Text dataHook={dataHooks.<%= componentName %>Count}>
           You clicked this button {isEven ? 'even' : 'odd'} number (
           <span className={styles.number}>
@@ -59,4 +49,20 @@ class <%= ComponentName %> extends React.PureComponent {
   }
 }
 
+<%= ComponentName %>.displayName = '<%= ComponentName %>';
+
+<%= ComponentName %>.propTypes = {
+  /** Applied as data-hook HTML attribute that can be used in the tests */
+  dataHook: PropTypes.string,
+
+  /** A css class to be applied to the component's root element */
+  className: PropTypes.string,
+
+  /** Text for the button */
+  buttonText: PropTypes.string,
+};
+
+<%= ComponentName %>.defaultProps = {
+};
+
 export default <%= ComponentName %>;

.wuf/generator/templates/$ComponentName/$ComponentName.uni.driver.js

@@ -1,9 +1,9 @@
 import { baseUniDriverFactory } from 'wix-ui-test-utils/base-driver';
-import { dataHooks } from "./constants";
+import { dataHooks } from './constants';
 
-export const <%= componentName %>DriverFactory = base => {
+export const <%= componentName %>DriverFactory = (base, body) => {
   return {
-    ...baseUniDriverFactory(base),
+    ...baseUniDriverFactory(base, body),
 
     /** Get the current count */
     getCountText: async () =>

.wuf/generator/templates/$ComponentName/index.d.ts

@@ -1,8 +1,8 @@
 import * as React from 'react';
 
-
 export interface <%= ComponentName %>Props {
   dataHook?: string;
+  className?: string;
   buttonText?: string;
 }
 

.wuf/generator/templates/$ComponentName/test/$ComponentName.private.uni.driver.js

@@ -1,8 +1,8 @@
 import { <%= componentName %>DriverFactory as publicDriverFactory } from '../<%= ComponentName %>.uni.driver';
 
-export const <%= componentName %>PrivateDriverFactory = base => {
+export const <%= componentName %>PrivateDriverFactory = (base, body) => {
   return {
-    ...publicDriverFactory(base),
+    ...publicDriverFactory(base, body),
 
     // Add here driver methods that considered "private"
   };