Added unit tests for the new wrap() method

Author: JRogaishio

tests/model.test.js

@@ -55,7 +55,7 @@ describe('Model methods', () => {
     })
   })
 
-  test('$first() returns first object in array as instance of such Model', async () => {
+  test('$first() returns first object in array as instance of such Model with "data" wrapper', async () => {
     axiosMock.onGet('http://localhost/posts').reply(200, postsEmbedResponse)
 
     const post = await Post.$first()
@@ -285,6 +285,55 @@ describe('Model methods', () => {
     expect(postsAll).toStrictEqual(postsGet)
   })
 
+  test('find() handles request with "data" wrapper when wrap() is set to "data"', async () => {
+    // Set the wrap method to 'data'
+    Post.prototype['wrap'] = () => {
+      return 'data'
+    }
+
+    axiosMock.onGet('http://localhost/posts/1').reply(200, postEmbedResponse)
+
+    const post = await Post.find(1)
+
+    expect(post).toEqual(postEmbedResponse.data)
+    expect(post).toBeInstanceOf(Post)
+    expect(post.user).toBeInstanceOf(User)
+    post.relationships.tags.data.forEach((tag) => {
+      expect(tag).toBeInstanceOf(Tag)
+    })
+  })
+
+  test('get() handles request with "data" wrapper when wrap() is set to "data"', async () => {
+    // Set the wrap method to 'data'
+    Post.prototype['wrap'] = () => {
+      return 'data'
+    }
+
+    axiosMock.onGet('http://localhost/posts').reply(200, postsEmbedResponse)
+
+    const posts = await Post.get()
+
+    expect(posts).toEqual(postsEmbedResponse.data)
+  })
+
+  test('first() returns first object in array as instance of such Model with "data" wrapper when wrap() is set to "data"', async () => {
+    // Set the wrap method to 'data'
+    Post.prototype['wrap'] = () => {
+      return 'data'
+    }
+
+    axiosMock.onGet('http://localhost/posts').reply(200, postsEmbedResponse)
+
+    const post = await Post.first()
+
+    expect(post).toEqual(postsEmbedResponse.data[0])
+    expect(post).toBeInstanceOf(Post)
+    expect(post.user).toBeInstanceOf(User)
+    post.relationships.tags.forEach((tag) => {
+      expect(tag).toBeInstanceOf(Tag)
+    })
+  })
+
   test('save() method makes a POST request when ID of object does not exists', async () => {
     let post
     const _postResponse = {