From a02cb39a45483955fb45cd16168c6ed68af8c7ed Mon Sep 17 00:00:00 2001
From: Jinzhu <wosmvp@gmail.com>
Date: Mon, 1 Jun 2020 00:36:18 +0800
Subject: [PATCH] Add more tests

---
 finisher_api.go     |  2 +-
 tests/query_test.go | 43 +++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 42 insertions(+), 3 deletions(-)

diff --git a/finisher_api.go b/finisher_api.go
index 334aea58..780de267 100644
--- a/finisher_api.go
+++ b/finisher_api.go
@@ -273,7 +273,7 @@ func (db *DB) Scan(dest interface{}) (tx *DB) {
 //     db.Find(&users).Pluck("age", &ages)
 func (db *DB) Pluck(column string, dest interface{}) (tx *DB) {
 	tx = db.getInstance()
-	tx.Statement.AddClause(clause.Select{Columns: []clause.Column{{Name: column}}})
+	tx.Statement.AddClauseIfNotExists(clause.Select{Columns: []clause.Column{{Name: column}}})
 	tx.Statement.Dest = dest
 	tx.callbacks.Query().Execute(tx)
 	return
diff --git a/tests/query_test.go b/tests/query_test.go
index b7c619d7..a4fe1243 100644
--- a/tests/query_test.go
+++ b/tests/query_test.go
@@ -2,8 +2,10 @@ package tests_test
 
 import (
 	"reflect"
+	"sort"
 	"strconv"
 	"testing"
+	"time"
 
 	. "github.com/jinzhu/gorm/tests"
 )
@@ -81,6 +83,24 @@ func TestFind(t *testing.T) {
 	}
 }
 
+func TestFillSmallerStruct(t *testing.T) {
+	user := User{Name: "SmallerUser", Age: 100}
+	DB.Save(&user)
+	type SimpleUser struct {
+		Name      string
+		ID        int64
+		UpdatedAt time.Time
+		CreatedAt time.Time
+	}
+
+	var simpleUser SimpleUser
+	if err := DB.Table("users").Where("name = ?", user.Name).First(&simpleUser).Error; err != nil {
+		t.Fatalf("Failed to query smaller user, got error %v", err)
+	}
+
+	AssertObjEqual(t, user, simpleUser, "Name", "ID", "UpdatedAt", "CreatedAt")
+}
+
 func TestPluck(t *testing.T) {
 	users := []*User{
 		GetUser("pluck-user1", Config{}),
@@ -92,12 +112,12 @@ func TestPluck(t *testing.T) {
 
 	var names []string
 	if err := DB.Model(User{}).Where("name like ?", "pluck-user%").Order("name").Pluck("name", &names).Error; err != nil {
-		t.Errorf("Raise error when pluck name, got %v", err)
+		t.Errorf("got error when pluck name: %v", err)
 	}
 
 	var ids []int
 	if err := DB.Model(User{}).Where("name like ?", "pluck-user%").Order("name").Pluck("id", &ids).Error; err != nil {
-		t.Errorf("Raise error when pluck id, got %v", err)
+		t.Errorf("got error when pluck id: %v", err)
 	}
 
 	for idx, name := range names {
@@ -112,3 +132,22 @@ func TestPluck(t *testing.T) {
 		}
 	}
 }
+
+func TestPluckWithSelect(t *testing.T) {
+	users := []User{
+		{Name: "pluck_with_select_1", Age: 25},
+		{Name: "pluck_with_select_2", Age: 26},
+	}
+
+	DB.Create(&users)
+
+	var userAges []int
+	err := DB.Model(&User{}).Where("name like ?", "pluck_with_select%").Select("age + 1  as user_age").Pluck("user_age", &userAges).Error
+	if err != nil {
+		t.Fatalf("got error when pluck user_age: %v", err)
+	}
+
+	sort.Ints(userAges)
+
+	AssertEqual(t, userAges, []int{26, 27})
+}