harness-drone/vendor/github.com/russross/meddler/meddler_test.go
2015-09-29 17:34:44 -07:00

113 lines
2.5 KiB
Go

package meddler
import (
"testing"
)
type ItemJson struct {
ID int64 `meddler:"id,pk"`
Stuff map[string]bool `meddler:"stuff,json"`
StuffZ map[string]bool `meddler:"stuffz,jsongzip"`
}
type ItemGob struct {
ID int64 `meddler:"id,pk"`
Stuff map[string]bool `meddler:"stuff,gob"`
StuffZ map[string]bool `meddler:"stuffz,gobgzip"`
}
func TestJsonMeddler(t *testing.T) {
once.Do(setup)
// save a value
elt := &ItemJson{
ID: 0,
Stuff: map[string]bool{
"hello": true,
"world": true,
},
StuffZ: map[string]bool{
"goodbye": true,
"cruel": true,
"world": true,
},
}
if err := Save(db, "item", elt); err != nil {
t.Errorf("Save error: %v", err)
}
id := elt.ID
// load it again
elt = new(ItemJson)
if err := Load(db, "item", elt, id); err != nil {
t.Errorf("Load error: %v", err)
}
if elt.ID != id {
t.Errorf("expected id of %d, found %d", id, elt.ID)
}
if len(elt.Stuff) != 2 {
t.Errorf("expected %d items in Stuff, found %d", 2, len(elt.Stuff))
}
if !elt.Stuff["hello"] || !elt.Stuff["world"] {
t.Errorf("contents of stuff wrong: %v", elt.Stuff)
}
if len(elt.StuffZ) != 3 {
t.Errorf("expected %d items in StuffZ, found %d", 3, len(elt.StuffZ))
}
if !elt.StuffZ["goodbye"] || !elt.StuffZ["cruel"] || !elt.StuffZ["world"] {
t.Errorf("contents of stuffz wrong: %v", elt.StuffZ)
}
if _, err := db.Exec("delete from `item`"); err != nil {
t.Errorf("error wiping item table: %v", err)
}
}
func TestGobMeddler(t *testing.T) {
once.Do(setup)
// save a value
elt := &ItemGob{
ID: 0,
Stuff: map[string]bool{
"hello": true,
"world": true,
},
StuffZ: map[string]bool{
"goodbye": true,
"cruel": true,
"world": true,
},
}
if err := Save(db, "item", elt); err != nil {
t.Errorf("Save error: %v", err)
}
id := elt.ID
// load it again
elt = new(ItemGob)
if err := Load(db, "item", elt, id); err != nil {
t.Errorf("Load error: %v", err)
}
if elt.ID != id {
t.Errorf("expected id of %d, found %d", id, elt.ID)
}
if len(elt.Stuff) != 2 {
t.Errorf("expected %d items in Stuff, found %d", 2, len(elt.Stuff))
}
if !elt.Stuff["hello"] || !elt.Stuff["world"] {
t.Errorf("contents of stuff wrong: %v", elt.Stuff)
}
if len(elt.StuffZ) != 3 {
t.Errorf("expected %d items in StuffZ, found %d", 3, len(elt.StuffZ))
}
if !elt.StuffZ["goodbye"] || !elt.StuffZ["cruel"] || !elt.StuffZ["world"] {
t.Errorf("contents of stuffz wrong: %v", elt.StuffZ)
}
if _, err := db.Exec("delete from `item`"); err != nil {
t.Errorf("error wiping item table: %v", err)
}
}