lbsadmin/Beego
1
0
Fork 0
mirror of https://github.com/astaxie/beego.git synced 2024-11-22 09:20:55 +00:00
Code Issues Releases Activity

orm test add extra custom JsonField

Browse Source
This commit is contained in:
slene 2013-11-26 21:34:40 +08:00
parent bcc8f60677
commit 690d77e985
2 changed files with 39 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
orm/models_test.go
View File

@ -1,6 +1,7 @@
package orm
import (
"encoding/json"
"fmt"
"os"
"strings"
@ -11,7 +12,7 @@ import (
_ "github.com/mattn/go-sqlite3"
)
// A true/false field.
// A slice string field.
type SliceStringField []string
func (e SliceStringField) Value() []string {
@ -57,10 +58,39 @@ func (e *SliceStringField) RawValue() interface{} {
return e.String()
}
func (e *SliceStringField) Clean() error {
var _ Fielder = new(SliceStringField)
// A json field.
type JsonField struct {
Name string
Data string
}
func (e *JsonField) String() string {
data, _ := json.Marshal(e)
return string(data)
}
func (e *JsonField) FieldType() int {
return TypeTextField
}
func (e *JsonField) SetRaw(value interface{}) error {
switch d := value.(type) {
case string:
return json.Unmarshal([]byte(d), e)
default:
return fmt.Errorf("<JsonField.SetRaw> unknown value `%v`", value)
}
return nil
}
func (e *JsonField) RawValue() interface{} {
return e.String()
}
var _ Fielder = new(JsonField)
type Data struct {
Id int
Boolean bool
@ -130,6 +160,7 @@ type User struct {
ShouldSkip string `orm:"-"`
Nums int
Langs SliceStringField `orm:"size(100)"`
Extra JsonField `orm:"type(text)"`
}
func (u *User) TableIndex() [][]string {

7
orm/orm_test.go
View File

@ -478,7 +478,9 @@ func TestCustomField(t *testing.T) {
throwFailNow(t, err)
user.Langs = append(user.Langs, "zh-CN", "en-US")
_, err = dORM.Update(&user, "Langs")
user.Extra.Name = "beego"
user.Extra.Data = "orm"
_, err = dORM.Update(&user, "Langs", "Extra")
throwFailNow(t, err)
user = User{Id: 2}
@ -487,6 +489,9 @@ func TestCustomField(t *testing.T) {
throwFailNow(t, AssertIs(len(user.Langs), 2))
throwFailNow(t, AssertIs(user.Langs[0], "zh-CN"))
throwFailNow(t, AssertIs(user.Langs[1], "en-US"))
throwFailNow(t, AssertIs(user.Extra.Name, "beego"))
throwFailNow(t, AssertIs(user.Extra.Data, "orm"))
}
func TestExpr(t *testing.T) {