1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- package schema_test
- import (
- "reflect"
- "sync"
- "testing"
- "github.com/jinzhu/gorm/schema"
- )
- type UserIndex struct {
- Name string `gorm:"index"`
- Name2 string `gorm:"index:idx_name,unique"`
- Name3 string `gorm:"index:,sort:desc,collate:utf8,type:btree,length:10,where:name3 != 'jinzhu'"`
- Name4 string `gorm:"unique_index"`
- Name5 int64 `gorm:"index:,class:FULLTEXT,comment:hello \\, world,where:age > 10"`
- Name6 int64 `gorm:"index:profile,comment:hello \\, world,where:age > 10"`
- Age int64 `gorm:"index:profile,expression:ABS(age)"`
- }
- func TestParseIndex(t *testing.T) {
- user, _, err := schema.Parse(&UserIndex{}, &sync.Map{}, schema.NamingStrategy{})
- if err != nil {
- t.Fatalf("failed to parse user index, got error %v", err)
- }
- results := map[string]schema.Index{
- "idx_user_indices_name": {
- Name: "idx_user_indices_name",
- Fields: []schema.IndexOption{{}},
- },
- "idx_name": {
- Name: "idx_name",
- Class: "UNIQUE",
- Fields: []schema.IndexOption{{}},
- },
- "idx_user_indices_name3": {
- Name: "idx_user_indices_name3",
- Type: "btree",
- Where: "name3 != 'jinzhu'",
- Fields: []schema.IndexOption{{
- Sort: "desc",
- Collate: "utf8",
- Length: 10,
- }},
- },
- "idx_user_indices_name4": {
- Name: "idx_user_indices_name4",
- Class: "UNIQUE",
- Fields: []schema.IndexOption{{}},
- },
- "idx_user_indices_name5": {
- Name: "idx_user_indices_name5",
- Class: "FULLTEXT",
- Comment: "hello , world",
- Where: "age > 10",
- Fields: []schema.IndexOption{{}},
- },
- "profile": {
- Name: "profile",
- Comment: "hello , world",
- Where: "age > 10",
- Fields: []schema.IndexOption{{}, {
- Expression: "ABS(age)",
- }},
- },
- }
- indices := user.ParseIndexes()
- for k, result := range results {
- v, ok := indices[k]
- if !ok {
- t.Errorf("Failed to found index %v from parsed indices %+v", k, indices)
- }
- for _, name := range []string{"Name", "Class", "Type", "Where", "Comment"} {
- if reflect.ValueOf(result).FieldByName(name).Interface() != reflect.ValueOf(v).FieldByName(name).Interface() {
- t.Errorf(
- "index %v %v should equal, expects %v, got %v",
- k, name, reflect.ValueOf(result).FieldByName(name).Interface(), reflect.ValueOf(v).FieldByName(name).Interface(),
- )
- }
- }
- for idx, ef := range result.Fields {
- rf := v.Fields[idx]
- for _, name := range []string{"Expression", "Sort", "Collate", "Length"} {
- if reflect.ValueOf(ef).FieldByName(name).Interface() != reflect.ValueOf(rf).FieldByName(name).Interface() {
- t.Errorf(
- "index %v field #%v's %v should equal, expects %v, got %v", k, idx+1, name,
- reflect.ValueOf(ef).FieldByName(name).Interface(), reflect.ValueOf(rf).FieldByName(name).Interface(),
- )
- }
- }
- }
- }
- }
|