diff --git a/filter_test.go b/filter_test.go index 079f553..60b2bd5 100644 --- a/filter_test.go +++ b/filter_test.go @@ -53,7 +53,7 @@ func TestFilterScope(t *testing.T) { schema := &schema.Schema{ DBNames: []string{"name"}, FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name", DBName: "name", DataType: schema.String}, + "name": {Name: "Name", DBName: "name", GORMDataType: schema.String}, }, Table: "test_scope_models", } @@ -84,7 +84,7 @@ func TestFilterScopeBlacklisted(t *testing.T) { schema := &schema.Schema{ DBNames: []string{"name"}, FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name"}, + "name": {Name: "Name", GORMDataType: schema.String}, }, } @@ -627,7 +627,7 @@ func TestFilterScopeWithUnsupportedDataType(t *testing.T) { schema := &schema.Schema{ DBNames: []string{"name"}, FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name", DBName: "name", DataType: "CHARACTER VARYING(255)"}, + "name": {Name: "Name", DBName: "name", GORMDataType: "custom", DataType: "CHARACTER VARYING(255)"}, }, Table: "test_scope_models", } diff --git a/search_test.go b/search_test.go index 9b7b1b6..e40870c 100644 --- a/search_test.go +++ b/search_test.go @@ -25,9 +25,9 @@ func TestSearchScope(t *testing.T) { schema := &schema.Schema{ FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name", DBName: "name", DataType: schema.String}, - "email": {Name: "Email", DBName: "email", DataType: schema.String}, - "role": {Name: "Role", DBName: "role", DataType: schema.String}, + "name": {Name: "Name", DBName: "name", GORMDataType: schema.String}, + "email": {Name: "Email", DBName: "email", GORMDataType: schema.String}, + "role": {Name: "Role", DBName: "role", GORMDataType: schema.String}, }, Table: "test_models", } @@ -88,9 +88,9 @@ func TestSearchScopeEmptyField(t *testing.T) { } schema := &schema.Schema{ FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name"}, - "email": {Name: "Email"}, - "role": {Name: "Role"}, + "name": {Name: "Name", GORMDataType: schema.String}, + "email": {Name: "Email", GORMDataType: schema.String}, + "role": {Name: "Role", GORMDataType: schema.String}, }, Table: "test_models", } @@ -454,9 +454,9 @@ func TestSearchScopeWithUnsupportedDataType(t *testing.T) { schema := &schema.Schema{ FieldsByDBName: map[string]*schema.Field{ - "name": {Name: "Name", DBName: "name", DataType: schema.String}, - "email": {Name: "Email", DBName: "email", DataType: "CHARACTER VARYING(255)"}, - "role": {Name: "Role", DBName: "role", DataType: schema.String}, + "name": {Name: "Name", DBName: "name", GORMDataType: schema.String}, + "email": {Name: "Email", DBName: "email", GORMDataType: "custom", DataType: "CHARACTER VARYING(255)"}, + "role": {Name: "Role", DBName: "role", GORMDataType: schema.String}, }, Table: "test_models", } diff --git a/util.go b/util.go index 74cbcbc..50496b8 100644 --- a/util.go +++ b/util.go @@ -127,7 +127,6 @@ func getDataType(field *schema.Field) DataType { case 64: return DataTypeFloat64 } - return DataTypeFloat64 case schema.Int: switch field.Size { case 8: @@ -139,7 +138,6 @@ func getDataType(field *schema.Field) DataType { case 64: return DataTypeInt64 } - return DataTypeInt64 case schema.Uint: switch field.Size { case 8: @@ -151,7 +149,6 @@ func getDataType(field *schema.Field) DataType { case 64: return DataTypeUint64 } - return DataTypeUint64 case schema.Time: return DataTypeTime }