fixup! apply feedback

loopbackio · Apr 22, 2019 · 70de24c · 70de24c
1 parent 7671446
commit 70de24c
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/packages/booter-lb3app/package-lock.json b/packages/booter-lb3app/package-lock.json
diff --git a/packages/booter-lb3app/package.json b/packages/booter-lb3app/package.json
@@ -39,6 +39,7 @@
     "@loopback/testlab": "^1.2.5",
     "@loopback/tslint-config": "^2.0.4",
     "@types/node": "^10.11.2",
+    "@types/lodash": "^4.14.123",
     "loopback-boot": "^3.3.0",
     "compression": "^1.7.4",
     "cors": "^2.8.5",

diff --git a/packages/booter-lb3app/src/__tests__/acceptance/booter-lb3app.acceptance.ts b/packages/booter-lb3app/src/__tests__/acceptance/booter-lb3app.acceptance.ts
@@ -4,6 +4,7 @@
 // License text available at https://opensource.org/licenses/MIT
 
 import {Client, expect} from '@loopback/testlab';
+import * as _ from 'lodash';
 import {
   CoffeeApplication,
   givenCoffeeShop,
@@ -36,9 +37,11 @@ describe('booter-lb3app', () => {
         .send(coffeeShop)
         .expect(200);
 
-      expect(response.body).to.containDeep(coffeeShop.__data);
+      expect(response.body).to.containDeep(
+        _.pick(coffeeShop, ['name', 'city']),
+      );
       const result = await client.get(`/api/CoffeeShops/${response.body.id}`);
-      expect(result.body).to.containDeep(coffeeShop.__data);
+      expect(result.body).to.containDeep(_.pick(coffeeShop, ['name', 'city']));
     });
 
     it('gets route defined outside a model', async () => {
@@ -126,12 +129,14 @@ describe('booter-lb3app', () => {
         .post('/coffees/CoffeeShops')
         .send(coffeeShop)
         .expect(200);
-      expect(response.body).to.containDeep(coffeeShop.__data);
+      expect(response.body).to.containDeep(
+        _.pick(coffeeShop, ['name', 'city']),
+      );
 
       const result = await client.get(
         `/coffees/CoffeeShops/${response.body.id}`,
       );
-      expect(result.body).to.containDeep(coffeeShop.__data);
+      expect(result.body).to.containDeep(_.pick(coffeeShop, ['name', 'city']));
     });
 
     it('does not get route defined outside a model', async () => {