fix: fix response of POST /register

Fix #8
matheuspiment · Oct 8, 2019 · 3922868 · 3922868
1 parent 414017c
commit 3922868
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/controllers/UserController.ts b/src/controllers/UserController.ts
@@ -1,6 +1,6 @@
 import { Request, Response } from 'express'
 import * as Yup from 'yup'
-import omit from 'lodash/omit'
+import pick from 'lodash/pick'
 
 import User from '../schemas/User'
 import CheckUserEmail from '../services/CheckUserEmail'
@@ -25,7 +25,7 @@ class UserController {
       }
 
       const user = await User.create(req.body)
-      return res.json(omit(user.toObject(), ['password']))
+      return res.json(pick(user.toObject(), ['_id', 'name', 'email']))
     } catch (err) {
       return res.status(400).json({ error: 'Registration failed' })
     }

diff --git a/src/schemas/User.ts b/src/schemas/User.ts
@@ -20,8 +20,7 @@ const UserSchema = new Schema({
   },
   password: {
     type: String,
-    required: true,
-    select: false
+    required: true
   }
 }, {
   versionKey: false,