66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
. "gorm.io/gorm/utils/tests"
|
|
)
|
|
|
|
type SerializerStruct struct {
|
|
gorm.Model
|
|
Name []byte `gorm:"json"`
|
|
Roles Roles `gorm:"json"`
|
|
EncryptedString EncryptedString
|
|
}
|
|
|
|
type Roles []string
|
|
type EncryptedString string
|
|
|
|
func (es *EncryptedString) Scan(ctx context.Context, field *schema.Field, dst reflect.Value, dbValue interface{}) (err error) {
|
|
switch value := dbValue.(type) {
|
|
case []byte:
|
|
*es = EncryptedString(bytes.TrimPrefix(value, []byte("hello")))
|
|
case string:
|
|
*es = EncryptedString(strings.TrimPrefix(value, "hello"))
|
|
default:
|
|
return fmt.Errorf("unsupported data %v", dbValue)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Value implements serializer interface
|
|
func (es EncryptedString) Value(ctx context.Context, field *schema.Field, dst reflect.Value, fieldValue interface{}) (interface{}, error) {
|
|
return "hello" + string(es), nil
|
|
}
|
|
|
|
func TestSerializer(t *testing.T) {
|
|
DB.Migrator().DropTable(&SerializerStruct{})
|
|
if err := DB.Migrator().AutoMigrate(&SerializerStruct{}); err != nil {
|
|
t.Fatalf("no error should happen when migrate scanner, valuer struct, got error %v", err)
|
|
}
|
|
|
|
data := SerializerStruct{
|
|
Name: []byte("jinzhu"),
|
|
Roles: []string{"r1", "r2"},
|
|
EncryptedString: EncryptedString("pass"),
|
|
}
|
|
|
|
if err := DB.Create(&data).Error; err != nil {
|
|
t.Fatalf("failed to create data, got error %v", err)
|
|
}
|
|
|
|
var result SerializerStruct
|
|
if err := DB.First(&result, data.ID).Error; err != nil {
|
|
t.Fatalf("failed to query data, got error %v", err)
|
|
}
|
|
|
|
AssertEqual(t, result, data)
|
|
}
|