some fix / add test

orm/models_test.go

@@ -0,0 +1,152 @@
+package orm
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	_ "github.com/bmizerany/pq"
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+type User struct {
+	Id       int       `orm:"auto"`
+	UserName string    `orm:"size(30);unique"`
+	Email    string    `orm:"size(100)"`
+	Password string    `orm:"size(100)"`
+	Status   int16     `orm:"choices(0,1,2,3);defalut(0)"`
+	IsStaff  bool      `orm:"default(false)"`
+	IsActive bool      `orm:"default(1)"`
+	Created  time.Time `orm:"auto_now_add;type(date)"`
+	Updated  time.Time `orm:"auto_now"`
+	Profile  *Profile  `orm:"null;rel(one);on_delete(set_null)"`
+	Posts    []*Post   `orm:"reverse(many)" json:"-"`
+	Manager  `json:"-"`
+}
+
+func NewUser() *User {
+	obj := new(User)
+	obj.Manager.Init(obj)
+	return obj
+}
+
+type Profile struct {
+	Id      int     `orm:"auto"`
+	Age     int16   ``
+	Money   float64 ``
+	User    *User   `orm:"reverse(one)" json:"-"`
+	Manager `json:"-"`
+}
+
+func (u *Profile) TableName() string {
+	return "user_profile"
+}
+
+func NewProfile() *Profile {
+	obj := new(Profile)
+	obj.Manager.Init(obj)
+	return obj
+}
+
+type Post struct {
+	Id      int       `orm:"auto"`
+	User    *User     `orm:"rel(fk)"` //
+	Title   string    `orm:"size(60)"`
+	Content string    ``
+	Created time.Time `orm:"auto_now_add"`
+	Updated time.Time `orm:"auto_now"`
+	Tags    []*Tag    `orm:"rel(m2m)"`
+	Manager `json:"-"`
+}
+
+func NewPost() *Post {
+	obj := new(Post)
+	obj.Manager.Init(obj)
+	return obj
+}
+
+type Tag struct {
+	Id      int     `orm:"auto"`
+	Name    string  `orm:"size(30)"`
+	Posts   []*Post `orm:"reverse(many)" json:"-"`
+	Manager `json:"-"`
+}
+
+func NewTag() *Tag {
+	obj := new(Tag)
+	obj.Manager.Init(obj)
+	return obj
+}
+
+type Comment struct {
+	Id      int       `orm:"auto"`
+	Post    *Post     `orm:"rel(fk)"`
+	Content string    ``
+	Parent  *Comment  `orm:"null;rel(fk)"`
+	Created time.Time `orm:"auto_now_add"`
+	Manager `json:"-"`
+}
+
+func NewComment() *Comment {
+	obj := new(Comment)
+	obj.Manager.Init(obj)
+	return obj
+}
+
+var DBARGS = struct {
+	Driver string
+	Source string
+}{
+	os.Getenv("ORM_DRIVER"),
+	os.Getenv("ORM_SOURCE"),
+}
+
+var dORM Ormer
+
+func init() {
+	RegisterModel(new(User))
+	RegisterModel(new(Profile))
+	RegisterModel(new(Post))
+	RegisterModel(new(Tag))
+	RegisterModel(new(Comment))
+
+	if DBARGS.Driver == "" || DBARGS.Source == "" {
+		fmt.Println(`need driver and source!
+
+Default DB Drivers.
+
+  driver: url
+   mysql: https://github.com/go-sql-driver/mysql
+ sqlite3: https://github.com/mattn/go-sqlite3
+postgres: https://github.com/bmizerany/pq
+
+eg: mysql
+ORM_DRIVER=mysql ORM_SOURCE="root:root@/my_db?charset=utf8" go test github.com/astaxie/beego/orm
+`)
+		os.Exit(2)
+	}
+
+	RegisterDataBase("default", DBARGS.Driver, DBARGS.Source, 20)
+
+	BootStrap()
+
+	truncateTables()
+
+	dORM = NewOrm()
+}
+
+func truncateTables() {
+	logs := "truncate tables for test\n"
+	o := NewOrm()
+	for _, m := range modelCache.allOrdered() {
+		query := fmt.Sprintf("truncate table `%s`", m.table)
+		_, err := o.Raw(query).Exec()
+		logs += query + "\n"
+		if err != nil {
+			fmt.Println(logs)
+			fmt.Println(err)
+			os.Exit(2)
+		}
+	}
+}