fix: mock pluck and pluckMany directly

Author: jlenon7

tests/unit/database/builders/QueryBuilderTest.ts

@@ -230,28 +230,25 @@ export default class QueryBuilderTest {
 
   @Test()
   public async shouldBeAbleToPluckData({ assert }: Context) {
-    const expectedData = { id: '1', name: 'John Doe' }
-    Mock.when(FakeDriver, 'find').resolve(expectedData)
+    const expectedData = 'John Doe'
+    Mock.when(FakeDriver, 'pluck').resolve(expectedData)
 
     const queryBuilder = new QueryBuilder(FakeDriver, 'users')
     const result = await queryBuilder.pluck('name')
 
-    assert.calledOnce(FakeDriver.find)
+    assert.calledOnce(FakeDriver.pluck)
     assert.deepEqual(result, 'John Doe')
   }
 
   @Test()
   public async shouldBeAbleToPluckManyData({ assert }: Context) {
-    const expectedData = [
-      { id: '1', name: 'John Doe' },
-      { id: '2', name: 'Jane Doe' }
-    ]
-    Mock.when(FakeDriver, 'findMany').resolve(expectedData)
+    const expectedData = ['John Doe', 'Jane Doe']
+    Mock.when(FakeDriver, 'pluckMany').resolve(expectedData)
 
     const queryBuilder = new QueryBuilder(FakeDriver, 'users')
     const result = await queryBuilder.pluckMany('name')
 
-    assert.calledOnce(FakeDriver.findMany)
+    assert.calledOnce(FakeDriver.pluckMany)
     assert.deepEqual(result, ['John Doe', 'Jane Doe'])
   }
 

tests/unit/models/BaseModelTest.ts

@@ -168,7 +168,7 @@ export default class BaseModelTest {
 
   @Test()
   public async shouldBeAbleToPluckColumnValueInDatabaseUsingPluckMethod({ assert }: Context) {
-    Mock.when(Database.driver, 'find').resolve({ id: '1' })
+    Mock.when(Database.driver, 'pluck').resolve('1')
 
     const data = await User.pluck('id')
 
@@ -244,7 +244,7 @@ export default class BaseModelTest {
 
   @Test()
   public async shouldBeAbleToPluckColumnsValueInDatabaseUsingPluckManyMethod({ assert }: Context) {
-    Mock.when(Database.driver, 'findMany').resolve([{ id: '1' }])
+    Mock.when(Database.driver, 'pluckMany').resolve(['1'])
 
     const data = await User.pluckMany('id')
 

tests/unit/models/builders/ModelQueryBuilderTest.ts

@@ -341,24 +341,21 @@ export default class ModelQueryBuilderTest {
 
   @Test()
   public async shouldBeAbleToPluckData({ assert }: Context) {
-    Mock.when(Database.driver, 'find').resolve({ id: '1', name: 'John Doe' })
+    Mock.when(Database.driver, 'pluck').resolve('John Doe')
 
     const result = await User.query().pluck('name')
 
-    assert.calledOnce(Database.driver.find)
+    assert.calledOnce(Database.driver.pluck)
     assert.deepEqual(result, 'John Doe')
   }
 
   @Test()
   public async shouldBeAbleToPluckManyData({ assert }: Context) {
-    Mock.when(Database.driver, 'findMany').resolve([
-      { id: '1', name: 'John Doe' },
-      { id: '2', name: 'Jane Doe' }
-    ])
+    Mock.when(Database.driver, 'pluckMany').resolve(['John Doe', 'Jane Doe'])
 
     const result = await User.query().pluckMany('name')
 
-    assert.calledOnce(Database.driver.findMany)
+    assert.calledOnce(Database.driver.pluckMany)
     assert.deepEqual(result, ['John Doe', 'Jane Doe'])
   }