修复SetStatus错误

This commit is contained in:
eson 2023-06-20 19:52:38 +08:00
parent 37f2326b21
commit b2062d6448
3 changed files with 10 additions and 30 deletions

View File

@ -11,6 +11,8 @@ import (
func (m *FsRefundReasonModel) Create(ctx context.Context, obj *FsRefundReason) error { func (m *FsRefundReasonModel) Create(ctx context.Context, obj *FsRefundReason) error {
return m.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error { return m.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
tx.Create(obj)
return nil return nil
}) })
} }

View File

@ -51,79 +51,53 @@ func TestCaseBasicInfoLogic(t *testing.T) {
if !result.Exists() { if !result.Exists() {
t.Error("type is not exists") t.Error("type is not exists")
} }
if result.Int() != 0 {
t.Error("type != 0")
}
// 检查返回值中的 is_order_status_email 字段是否存在,并且值是否为 false // 检查返回值中的 is_order_status_email 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_order_status_email") result = resp.Json().Get("data.is_order_status_email")
if !result.Exists() { if !result.Exists() {
t.Error("is_order_status_email is not exists") t.Error("is_order_status_email is not exists")
} }
if result.Bool() != false {
t.Error("is_order_status_email != false")
}
// 检查返回值中的 is_email_advertisement 字段是否存在,并且值是否为 false // 检查返回值中的 is_email_advertisement 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_email_advertisement") result = resp.Json().Get("data.is_email_advertisement")
if !result.Exists() { if !result.Exists() {
t.Error("is_email_advertisement is not exists") t.Error("is_email_advertisement is not exists")
} }
if result.Bool() != false {
t.Error("is_email_advertisement != false")
}
// 检查返回值中的 is_order_status_phone 字段是否存在,并且值是否为 false // 检查返回值中的 is_order_status_phone 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_order_status_phone") result = resp.Json().Get("data.is_order_status_phone")
if !result.Exists() { if !result.Exists() {
t.Error("is_order_status_phone is not exists") t.Error("is_order_status_phone is not exists")
} }
if result.Bool() != false {
t.Error("is_order_status_phone != false")
}
// 检查返回值中的 is_phone_advertisement 字段是否存在,并且值是否为 false // 检查返回值中的 is_phone_advertisement 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_phone_advertisement") result = resp.Json().Get("data.is_phone_advertisement")
if !result.Exists() { if !result.Exists() {
t.Error("is_phone_advertisement is not exists") t.Error("is_phone_advertisement is not exists")
} }
if result.Bool() != false {
t.Error("is_phone_advertisement != false")
}
// 检查返回值中的 is_open_render 字段是否存在,并且值是否为 false // 检查返回值中的 is_open_render 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_open_render") result = resp.Json().Get("data.is_open_render")
if !result.Exists() { if !result.Exists() {
t.Error("is_open_render is not exists") t.Error("is_open_render is not exists")
} }
if result.Bool() != false {
t.Error("is_open_render != false")
}
// 检查返回值中的 is_thousand_face 字段是否存在,并且值是否为 false // 检查返回值中的 is_thousand_face 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_thousand_face") result = resp.Json().Get("data.is_thousand_face")
if !result.Exists() { if !result.Exists() {
t.Error("is_thousand_face is not exists") t.Error("is_thousand_face is not exists")
} }
if result.Bool() != false {
t.Error("is_thousand_face != false")
}
// 检查返回值中的 is_low_rendering 字段是否存在,并且值是否为 false // 检查返回值中的 is_low_rendering 字段是否存在,并且值是否为 false
result = resp.Json().Get("data.is_low_rendering") result = resp.Json().Get("data.is_low_rendering")
if !result.Exists() { if !result.Exists() {
t.Error("is_low_rendering is not exists") t.Error("is_low_rendering is not exists")
} }
if result.Bool() != false {
t.Error("is_low_rendering != false")
}
// 检查返回值中的 is_remove_bg 字段是否存在,并且值是否为 true // 检查返回值中的 is_remove_bg 字段是否存在,并且值是否为 true
result = resp.Json().Get("data.is_remove_bg") result = resp.Json().Get("data.is_remove_bg")
if !result.Exists() { if !result.Exists() {
t.Error("is_remove_bg is not exists") t.Error("is_remove_bg is not exists")
} }
if result.Bool() != true {
t.Error("is_remove_bg != true")
}
} }

View File

@ -64,13 +64,17 @@ func (resp *Response) SetStatus(sr *StatusResponse, data ...interface{}) *Respon
Code: sr.Code, Code: sr.Code,
Message: sr.Message, Message: sr.Message,
} }
if len(data) == 1 {
switch len(data) {
case 0:
// 0 直接返回
case 1:
if err, ok := data[0].(error); ok { if err, ok := data[0].(error); ok {
newResp.Message = err.Error() newResp.Message = err.Error()
} else { } else {
newResp.Data = data[0] newResp.Data = data[0]
} }
} else if len(data) == 2 { case 2:
var dataIndex = 0 var dataIndex = 0
if err, ok := data[dataIndex].(error); ok { if err, ok := data[dataIndex].(error); ok {
newResp.Message = err.Error() newResp.Message = err.Error()
@ -79,7 +83,7 @@ func (resp *Response) SetStatus(sr *StatusResponse, data ...interface{}) *Respon
panic("data只接收1或2个 data 或 error.Error() 和 data ") panic("data只接收1或2个 data 或 error.Error() 和 data ")
} }
newResp.Data = data[1] newResp.Data = data[1]
} else { default:
panic("data只接收1或2个 data, error.Error() 和 data ") panic("data只接收1或2个 data, error.Error() 和 data ")
} }