op_test.go

package gocassa

import (
	"strconv"
	"strings"

	"github.com/gocql/gocql"
)

type OpTestStruct struct {
	A string `cql:"id,omitempty"`
	B int
	C int `cql:"-"`
	D map[string]interface{}
	E []interface{}
	F []string
	G []OpTestStructAlias

	OpTestStructEmbed `cql:",squash"`
}

type OpTestStructEmbed struct {
	XA int
	XB OpTestStructAlias
	XC []string
}

type OpTestStructAlias string

var opTestRow = map[string]interface{}{
	"id": "A",
	"B":  1,
	"C":  1,
	"D": map[string]interface{}{
		"D1": 1,
		"D2": "2",
	},
	"E": []interface{}{
		"E1",
		"E2", "E3", 4,
	},
	"F": []string{
		"F1", "F2", "F3",
	},
	"G": []string{
		"F1", "F2", "F3",
	},

	"XA": 1,
	"XB": "XB",
	"XC": []string{
		"XC1", "XC2",
	},
}

var opTestRowExpected = OpTestStruct{
	A: "A",
	B: 1,
	C: 0,
	D: map[string]interface{}{"D1": 1, "D2": "2"},
	E: []interface{}{"E1", "E2", "E3", 4},
	F: []string{"F1", "F2", "F3"},
	G: []OpTestStructAlias{"F1", "F2", "F3"},
	OpTestStructEmbed: OpTestStructEmbed{
		XA: 1,
		XB: "XB",
		XC: []string{"XC1", "XC2"},
	},
}

type OpTestCompositeType struct {
	Num int
	Str string
}

func (t *OpTestCompositeType) UnmarshalCQL(_ gocql.TypeInfo, data []byte) error {
	parts := strings.Split(string(data), " ")
	num, err := strconv.Atoi(parts[0])
	if err != nil {
		return err
	}
	t.Num = num
	t.Str = parts[1]
	return nil
}

type OpTestEnumType int32

const (
	OpTestEnumOne   OpTestEnumType = 1
	OpTestEnumTwo   OpTestEnumType = 2
	OpTestEnumThree OpTestEnumType = 3
)

func (t *OpTestEnumType) UnmarshalCQL(_ gocql.TypeInfo, data []byte) error {
	values := map[string]OpTestEnumType{
		"EnumOne":   1,
		"EnumTwo":   2,
		"EnumThree": 3,
	}
	*t = values[string(data)]
	return nil
}