From 88bff22526fedcd8c8e4453653cc2fbd55e9797b Mon Sep 17 00:00:00 2001
From: Havoc412 <2993167370@qq.com>
Date: Wed, 16 Oct 2024 14:12:23 +0800
Subject: [PATCH] update animal insert test

---
 app/model/animal.go       |  4 ++--
 test/insertAnimal_test.go | 36 +++++++++++++++++++++++++++---------
 2 files changed, 29 insertions(+), 11 deletions(-)

diff --git a/app/model/animal.go b/app/model/animal.go
index d8b9656..4f2a81c 100644
--- a/app/model/animal.go
+++ b/app/model/animal.go
@@ -18,7 +18,7 @@ type Animal struct {
 	Gender        uint8  `json:"gender,omitempty"`                                                  // 性别
 	Breed         uint8  `json:"breed,omitempty"`                                                   // 品种
 	Sterilization uint8  `json:"sterilization,omitempty"`                                           // 1 不明 2 未绝育 3 已绝育
-	NickName      string `gorm:"type:varchar(31)" json:"nick_name,omitempty"`                       // 别称，辅助查询；存储上采取 , 间隔符的方式; VARCHAR 会比较合适
+	NickNames     string `gorm:"type:varchar(31)" json:"nick_names,omitempty"`                      // 别称，辅助查询；存储上采取 , 间隔符的方式; VARCHAR 会比较合适
 	Status        uint8  `json:"status,omitempty"`                                                  // 状态
 	Description   string `gorm:"column:description;type:varchar(255)" json:"description,omitempty"` // 简明介绍
 	Tags          string `json:"tags,omitempty"`
@@ -34,7 +34,7 @@ type Animal struct {
 	ActivityRadius uint64  `json:"activity_radius,omitempty"` // 活动半径
 	// CatFace
 	FaceBreeds     string `json:"face_breeds,omitempty" gorm:"size:20"`
-	FaceBreedProbs string `json:"face_breed_probs" gorm:"size:20"`
+	FaceBreedProbs string `json:"face_breed_probs,omitempty" gorm:"size:20"`
 }
 
 func (a *Animal) TableName() string {
diff --git a/test/insertAnimal_test.go b/test/insertAnimal_test.go
index 723b250..32582a1 100644
--- a/test/insertAnimal_test.go
+++ b/test/insertAnimal_test.go
@@ -41,7 +41,10 @@ const (
 		"HeadImg":        "1.jpg",
 		"Photos":         "",
 		"ActivityRadius": 100,
-		"NickName": "胖北,北子"
+		"Tags": "",
+		"NickName": "胖北,北子",
+		"face_breeds":     "10,8,2",
+		"face_breed_probs": "0.9,0.09,0.01"
 	}, {
 		"Name": "打三",
 		"Birthday": "2019-01-01",
@@ -58,7 +61,10 @@ const (
 		"HeadImg":        "2.jpg",
 		"Photos":         "",
 		"ActivityRadius": 100,
-		"NickName": "打人三花,小花"
+		"NickName": "打人三花,小花",
+		"Tags": "",
+		"face_breeds":     "10,8,2",
+		"face_breed_probs": "0.9,0.09,0.01"
 	}, {
 		"Name": "猪皮",
 		"Birthday": "2017-01-01",
@@ -71,18 +77,20 @@ const (
 		"Longitude": 114.367661,
 		"Avatar": "4.jpg",
 		"AvatarHeight":   191,
-		"AvatarWidth":    160,
+		"AvatarWidth":    160,		
 		"HeadImg":        "4.jpg",
 		"Photos":         "0.png,1.jpg,3.jpg",
 		"ActivityRadius": 100,
-		"ActivityRadius": 100,
-		"NickName": "猜皮,猪",
-		"Tags": "臭脸,猜皮,玉玉"
+		"nick_names": "猜皮,猪",
+		"Tags": "臭脸,猜皮,玉玉",
+		"face_breeds":     "2,6,4",
+		"face_breed_probs": "0.9,0.09,0.01"
 	}, {
 		"Name": "切糕",
 		"Birthday": "2022-01-01",
 		"Gender": 2,
 		"Breed": 8,
+		"Tags": "",
 		"Status": 1,
 		"Sterilization": 1,
 		"Description": "脾气老实的男妈妈；活动在湖滨区域；身边还有一只狸花猫叫做小蝶；",
@@ -93,7 +101,9 @@ const (
 		"AvatarWidth":    306,
 		"HeadImg":        "6.jpg",
 		"Photos":         "",
-		"ActivityRadius": 100
+		"ActivityRadius": 100,
+		"face_breeds":     "8,7,5",
+		"face_breed_probs": "0.8,0.19,0.01"
 	}]`
 )
 
@@ -112,7 +122,7 @@ func TestCreateAnimal(t *testing.T) {
 		Gender:         3,
 		Breed:          3, // 示例品种 ID
 		Sterilization:  2, // 已绝育
-		NickName:       "cc5,西西务,西西务小姐",
+		NickNames:      "cc5,西西务,西西务小姐",
 		Status:         1, // 示例状态
 		Description:    "信部资深学姐,优雅的代名。",
 		Latitude:       30.532645,
@@ -123,6 +133,9 @@ func TestCreateAnimal(t *testing.T) {
 		HeadImg:        "0.jpg",
 		Photos:         "",
 		ActivityRadius: 100,
+		Tags:           "优雅",
+		FaceBreeds:     "3,4,5",
+		FaceBreedProbs: "0.9,0.05,0.05",
 	}
 
 	// 插入数据到数据库
@@ -148,13 +161,14 @@ func TestCreateAnimalList(t *testing.T) {
 	}
 
 	for _, animal := range animals {
+		fmt.Println("animal:", animal)
 		animal := model.Animal{
 			Name:           animal.Name,
 			Birthday:       animal.Birthday,
 			Gender:         animal.Gender,
 			Breed:          animal.Breed,
 			Sterilization:  animal.Sterilization,
-			NickName:       animal.NickName,
+			NickNames:      animal.NickNames,
 			Status:         animal.Status,
 			Description:    animal.Description,
 			Latitude:       animal.Latitude,
@@ -165,9 +179,13 @@ func TestCreateAnimalList(t *testing.T) {
 			HeadImg:        animal.HeadImg,
 			Photos:         animal.Photos,
 			ActivityRadius: animal.ActivityRadius,
+			FaceBreeds:     animal.FaceBreeds,
+			FaceBreedProbs: animal.FaceBreedProbs,
 			Tags:           animal.Tags,
 		}
 
+		fmt.Println("TEST: ", animal.FaceBreeds, animal.FaceBreedProbs)
+
 		result := DB.Create(&animal)
 
 		// 检查插入是否成功