You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

85 lines
1.9 KiB

package schema
import (
"encoding/json"
"github.com/golang/protobuf/proto"
. "github.com/seaweedfs/seaweedfs/weed/pb/schema_pb"
"github.com/stretchr/testify/assert"
"testing"
)
func TestEnumScalarType(t *testing.T) {
tests := []struct {
name string
enum ScalarType
expected int32
}{
{"Boolean", ScalarType_BOOL, 0},
{"Integer", ScalarType_INT32, 1},
{"Long", ScalarType_INT64, 3},
{"Float", ScalarType_FLOAT, 4},
{"Double", ScalarType_DOUBLE, 5},
{"Bytes", ScalarType_BYTES, 6},
{"String", ScalarType_STRING, 7},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
assert.Equal(t, tt.expected, int32(tt.enum))
})
}
}
func TestField(t *testing.T) {
field := &Field{
Name: "field_name",
Type: &Type{Kind: &Type_ScalarType{ScalarType: ScalarType_INT32}},
FieldIndex: 1,
IsRepeated: false,
}
assert.NotNil(t, field)
}
func TestRecordType(t *testing.T) {
subRecord := &RecordType{
Fields: []*Field{
{
Name: "field_1",
Type: &Type{Kind: &Type_ScalarType{ScalarType: ScalarType_INT32}},
FieldIndex: 1,
IsRepeated: false,
},
{
Name: "field_2",
Type: &Type{Kind: &Type_ScalarType{ScalarType: ScalarType_STRING}},
FieldIndex: 2,
IsRepeated: false,
},
},
}
record := &RecordType{
Fields: []*Field{
{
Name: "field_key",
Type: &Type{Kind: &Type_ScalarType{ScalarType: ScalarType_INT32}},
FieldIndex: 1,
IsRepeated: false,
},
{
Name: "field_record",
Type: &Type{Kind: &Type_RecordType{RecordType: subRecord}},
FieldIndex: 2,
IsRepeated: false,
},
},
}
// serialize record to protobuf text marshalling
text := proto.MarshalTextString(record)
println(text)
bytes, _ := json.Marshal(record)
println(string(bytes))
assert.NotNil(t, record)
}