mirror of
https://github.com/geektutu/7days-golang.git
synced 2024-04-21 12:32:11 +00:00
29 lines
513 B
Go
29 lines
513 B
Go
package session
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
type User struct {
|
|
Name string `geeorm:"PRIMARY KEY"`
|
|
Age int
|
|
}
|
|
|
|
func TestSession_CreateTable(t *testing.T) {
|
|
s := NewSession().Model(&User{})
|
|
_ = s.DropTable()
|
|
_ = s.CreateTable()
|
|
if !s.HasTable() {
|
|
t.Fatal("Failed to create table User")
|
|
}
|
|
}
|
|
|
|
func TestSession_Model(t *testing.T) {
|
|
s := NewSession().Model(&User{})
|
|
table := s.RefTable()
|
|
s.Model(&Session{})
|
|
if table.Name != "User" || s.RefTable().Name != "Session" {
|
|
t.Fatal("Failed to change model")
|
|
}
|
|
}
|