gorm/scaner_test.go

package gorm_test

import (
	"database/sql/driver"
	"encoding/json"
	"errors"
	"testing"

	"github.com/jinzhu/gorm"
)

func TestScannableSlices(t *testing.T) {
	if err := DB.AutoMigrate(&RecordWithSlice{}).Error; err != nil {
		t.Errorf("Should create table with slice values correctly: %s", err)
	}

	r1 := RecordWithSlice{
		Strings: ExampleStringSlice{"a", "b", "c"},
		Structs: ExampleStructSlice{
			{"name1", "value1"},
			{"name2", "value2"},
		},
	}

	if err := DB.Save(&r1).Error; err != nil {
		t.Errorf("Should save record with slice values")
	}

	var r2 RecordWithSlice

	if err := DB.Find(&r2).Error; err != nil {
		t.Errorf("Should fetch record with slice values")
	}

	if len(r2.Strings) != 3 || r2.Strings[0] != "a" || r2.Strings[1] != "b" || r2.Strings[2] != "c" {
		t.Errorf("Should have serialised and deserialised a string array")
	}

	if len(r2.Structs) != 2 || r2.Structs[0].Name != "name1" || r2.Structs[0].Value != "value1" || r2.Structs[1].Name != "name2" || r2.Structs[1].Value != "value2" {
		t.Errorf("Should have serialised and deserialised a struct array")
	}
}

type RecordWithSlice struct {
	ID      uint64
	Strings ExampleStringSlice `sql:"type:text"`
	Structs ExampleStructSlice `sql:"type:text"`
}

type ExampleStringSlice []string

func (l ExampleStringSlice) Value() (driver.Value, error) {
	return json.Marshal(l)
}

func (l *ExampleStringSlice) Scan(input interface{}) error {
	switch value := input.(type) {
	case string:
		return json.Unmarshal([]byte(value), l)
	case []byte:
		return json.Unmarshal(value, l)
	default:
		return errors.New("not supported")
	}
}

type ExampleStruct struct {
	Name  string
	Value string
}

type ExampleStructSlice []ExampleStruct

func (l ExampleStructSlice) Value() (driver.Value, error) {
	return json.Marshal(l)
}

func (l *ExampleStructSlice) Scan(input interface{}) error {
	switch value := input.(type) {
	case string:
		return json.Unmarshal([]byte(value), l)
	case []byte:
		return json.Unmarshal(value, l)
	default:
		return errors.New("not supported")
	}
}

type ScannerDataType struct {
	Street string `sql:"TYPE:varchar(24)"`
}

func (ScannerDataType) Value() (driver.Value, error) {
	return nil, nil
}

func (*ScannerDataType) Scan(input interface{}) error {
	return nil
}

type ScannerDataTypeTestStruct struct {
	Field1          int
	ScannerDataType *ScannerDataType `sql:"TYPE:json"`
}

type ScannerDataType2 struct {
	Street string `sql:"TYPE:varchar(24)"`
}

func (ScannerDataType2) Value() (driver.Value, error) {
	return nil, nil
}

func (*ScannerDataType2) Scan(input interface{}) error {
	return nil
}

type ScannerDataTypeTestStruct2 struct {
	Field1          int
	ScannerDataType *ScannerDataType2
}

func TestScannerDataType(t *testing.T) {
	scope := gorm.Scope{Value: &ScannerDataTypeTestStruct{}}
	if field, ok := scope.FieldByName("ScannerDataType"); ok {
		if DB.Dialect().DataTypeOf(field.StructField) != "json" {
			t.Errorf("data type for scanner is wrong")
		}
	}

	scope = gorm.Scope{Value: &ScannerDataTypeTestStruct2{}}
	if field, ok := scope.FieldByName("ScannerDataType"); ok {
		if DB.Dialect().DataTypeOf(field.StructField) != "varchar(24)" {
			t.Errorf("data type for scanner is wrong")
		}
	}
}