Testing methods on our class

Author: Ashley Bibizadeh

src/App.js

@@ -18,19 +18,39 @@ const tempArr = [{
   onlineStatus: true
 }];
 
+const initialState = {
+  hideBtn: false
+};
+
 class App extends Component {
 
   constructor(props){
     super(props);
+    this.state = {
+      ...initialState
+    }
     this.fetch = this.fetch.bind(this);
   }
 
   fetch(){
     this.props.fetchPosts();
+    this.exampleMethod_updatesState();
+  }
+
+  exampleMethod_updatesState() {
+    const { hideBtn } = this.state;
+    this.setState({
+      hideBtn: !hideBtn
+    });
+  }
+
+  exampleMethod_returnsAValue(number) {
+    return number + 1;
   }
 
   render() {
     const { posts } = this.props;
+    const { hideBtn } = this.state;
 
     const configButton = {
       buttonText: 'Get posts',
@@ -42,7 +62,11 @@ class App extends Component {
         <Header />
         <section className="main">
           <Headline header="Posts" desc="Click the button to render posts!" tempArr={tempArr} />
-          <SharedButton {...configButton} />
+          
+          {!hideBtn &&
+            <SharedButton {...configButton} />
+          }
+          
           {posts.length > 0 &&
             <div>
               {posts.map((post, index) => {

src/App.test.js

@@ -33,5 +33,18 @@ describe('App Component', () => {
         expect(component.length).toBe(1);
     });
 
+    it('exampleMethod_updatesState Method should update state as expected', () => {
+        const classInstance = wrapper.instance();
+        classInstance.exampleMethod_updatesState();
+        const newState = classInstance.state.hideBtn;
+        expect(newState).toBe(true);
+    });
+
+    it('exampleMethod_returnsAValue Method should return value as expected', () => {
+        const classInstance = wrapper.instance();
+        const newValue = classInstance.exampleMethod_returnsAValue(6);
+        expect(newValue).toBe(7);
+    });
+
 
 });