refactor the code

Relates #38
GSG-G7 · Oct 2, 2019 · 4a713be · 4a713be
1 parent 0a068bc
commit 4a713be
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/client/src/components/pages/Login/index.js b/client/src/components/pages/Login/index.js
@@ -25,7 +25,7 @@ export default class Login extends React.Component {
     e.preventDefault();
     const { tableNumber, secret } = this.state;
     const { history, updateTableNumber } = this.props;
-    logInSchema
+    return logInSchema
       .validateAsync({ tableNumber, secret })
       .then(() => {
         return fetch('/api/v1/login', {
@@ -37,11 +37,11 @@ export default class Login extends React.Component {
           }),
         });
       })
-      .then((res => {
+      .then(res => {
         if (res.status === 200) {
-          updateTableNumber(tableNumber, history.push);
-        } else
-          return this.setState({ err: 'Incorrect Table Number or Secret' });
+          return updateTableNumber(tableNumber, history.push);
+        }
+        return this.setState({ err: 'Incorrect Table Number or Secret' });
       })
       .catch(err => {
         if (err.details) {