51 lines
1.9 KiB
Go
51 lines
1.9 KiB
Go
|
// 功能描述：验证资料更新保留头像、拒绝跨账户图片及空昵称。
|
|||
|
|
// 版本：1.0.0。
|
|||
|
|
package user
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"git.apinb.com/bsm-sdk/core/types"
|
|||
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TestProfileUpdateBoundaries 对数据库读写使用 SQL Mock，不连接远程数据。
|
|||
|
|
func TestProfileUpdateBoundaries(t *testing.T) {
|
|||
|
|
for _, tc := range []struct {
|
|||
|
|
name, body string
|
|||
|
|
valid bool
|
|||
|
|
}{
|
|||
|
|
{"昵称修改保留头像", `{"name":"新昵称"}`, true},
|
|||
|
|
{"允许保留旧头像", `{"name":"新昵称","avatar":"/uploads/avatars/old.png"}`, true},
|
|||
|
|
{"空昵称拒绝", `{"name":" "}`, false},
|
|||
|
|
{"他人头像拒绝", `{"name":"新昵称","avatar":"/uploads/avatars/other.png"}`, false},
|
|||
|
|
} {
|
|||
|
|
t.Run(tc.name, func(t *testing.T) {
|
|||
|
|
mock := primaryTestDB(t)
|
|||
|
|
mock.ExpectQuery(`SELECT .* FROM "user_account"`).WillReturnRows(sqlmock.NewRows([]string{"id", "identity", "name", "avatar"}).AddRow(1, "alice", "旧昵称", "/uploads/avatars/old.png"))
|
|||
|
|
if tc.valid {
|
|||
|
|
if strings.Contains(tc.body, "avatar") {
|
|||
|
|
mock.ExpectExec(`UPDATE "user_account" SET "avatar"=.*"name"=`).WillReturnResult(sqlmock.NewResult(0, 1))
|
|||
|
|
} else {
|
|||
|
|
mock.ExpectExec(`UPDATE "user_account" SET "name"=`).WillReturnResult(sqlmock.NewResult(0, 1))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
response := httptest.NewRecorder()
|
|||
|
|
ctx, _ := gin.CreateTestContext(response)
|
|||
|
|
ctx.Set("Auth", &types.JwtClaims{Client: "user_app", Identity: "alice"})
|
|||
|
|
ctx.Request = httptest.NewRequest("PUT", "/auth/profile", strings.NewReader(tc.body))
|
|||
|
|
ctx.Request.Header.Set("Content-Type", "application/json")
|
|||
|
|
UpdateProfile(ctx)
|
|||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
success := strings.Contains(response.Body.String(), `"code":0`)
|
|||
|
|
if success != tc.valid {
|
|||
|
|
t.Fatalf("更新状态错误: %s", response.Body.String())
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|