104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
package bdb
|
|
|
|
import "testing"
|
|
|
|
func TestSQLColDefinitions(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cols := []Column{
|
|
{Name: "one", Type: "int64"},
|
|
{Name: "two", Type: "string"},
|
|
{Name: "three", Type: "string"},
|
|
}
|
|
|
|
defs := SQLColDefinitions(cols, []string{"one"})
|
|
if len(defs) != 1 {
|
|
t.Error("wrong number of defs:", len(defs))
|
|
}
|
|
if got := defs[0].String(); got != "one int64" {
|
|
t.Error("wrong def:", got)
|
|
}
|
|
|
|
defs = SQLColDefinitions(cols, []string{"one", "three"})
|
|
if len(defs) != 2 {
|
|
t.Error("wrong number of defs:", len(defs))
|
|
}
|
|
if got := defs[0].String(); got != "one int64" {
|
|
t.Error("wrong def:", got)
|
|
}
|
|
if got := defs[1].String(); got != "three string" {
|
|
t.Error("wrong def:", got)
|
|
}
|
|
}
|
|
|
|
func TestSQLDefStrings(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cols := []Column{
|
|
{Name: "one", Type: "int64"},
|
|
{Name: "two", Type: "string"},
|
|
{Name: "three", Type: "string"},
|
|
}
|
|
|
|
defs := SQLColDefinitions(cols, []string{"one", "three"})
|
|
strs := SQLColDefStrings(defs)
|
|
|
|
if got := strs[0]; got != "one int64" {
|
|
t.Error("wrong str:", got)
|
|
}
|
|
if got := strs[1]; got != "three string" {
|
|
t.Error("wrong str:", got)
|
|
}
|
|
}
|
|
|
|
func TestAutoIncPrimaryKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := map[string]struct {
|
|
Ok bool
|
|
Expect Column
|
|
Pkey *PrimaryKey
|
|
Columns []Column
|
|
}{
|
|
"nillcase": {
|
|
Ok: false,
|
|
Pkey: nil,
|
|
Columns: nil,
|
|
},
|
|
"easycase": {
|
|
Ok: true,
|
|
Expect: Column{Name: "one", Type: "int32", Nullable: false, Default: `nextval('abc'::regclass)`},
|
|
Pkey: &PrimaryKey{Name: "pkey", Columns: []string{"one"}},
|
|
Columns: []Column{{Name: "one", Type: "int32", Nullable: false, Default: `nextval('abc'::regclass)`}},
|
|
},
|
|
"missingcase": {
|
|
Ok: false,
|
|
Pkey: &PrimaryKey{Name: "pkey", Columns: []string{"two"}},
|
|
Columns: []Column{{Name: "one", Type: "int32", Nullable: false, Default: `nextval('abc'::regclass)`}},
|
|
},
|
|
"wrongtype": {
|
|
Ok: false,
|
|
Pkey: &PrimaryKey{Name: "pkey", Columns: []string{"one"}},
|
|
Columns: []Column{{Name: "one", Type: "string", Nullable: false, Default: `nextval('abc'::regclass)`}},
|
|
},
|
|
"nodefault": {
|
|
Ok: false,
|
|
Pkey: &PrimaryKey{Name: "pkey", Columns: []string{"one"}},
|
|
Columns: []Column{{Name: "one", Type: "string", Nullable: false, Default: ``}},
|
|
},
|
|
"nullable": {
|
|
Ok: false,
|
|
Pkey: &PrimaryKey{Name: "pkey", Columns: []string{"one"}},
|
|
Columns: []Column{{Name: "one", Type: "string", Nullable: true, Default: `nextval('abc'::regclass)`}},
|
|
},
|
|
}
|
|
|
|
for testName, test := range tests {
|
|
pkey := AutoIncPrimaryKey(test.Columns, test.Pkey)
|
|
if ok := (pkey != nil); ok != test.Ok {
|
|
t.Errorf("%s) found state was wrong, want: %t, got: %t", testName, test.Ok, ok)
|
|
} else if test.Ok && *pkey != test.Expect {
|
|
t.Errorf("%s) wrong primary key, want: %#v, got %#v", testName, test.Expect, pkey)
|
|
}
|
|
}
|
|
}
|