224 lines
4.5 KiB
Go
224 lines
4.5 KiB
Go
package ndb
|
||
|
||
import (
|
||
"testing"
|
||
"fmt"
|
||
)
|
||
|
||
type screenCfg struct {
|
||
Title string
|
||
Width, Height uint16
|
||
R,G,B,A uint16
|
||
}
|
||
|
||
type netCfg struct {
|
||
Host string `ndb:"hostname"`
|
||
Vlan []int `ndb:"vlan"`
|
||
Native int `ndb:"nativevlan"`
|
||
}
|
||
|
||
var multiMap = []struct {
|
||
in string
|
||
out map[string] []string
|
||
}{
|
||
{
|
||
in: "user=clive user=david user=trenton group=dirty-dozen",
|
||
out: map[string] []string {
|
||
"user": []string {"clive", "david", "trenton"},
|
||
"group": []string {"dirty-dozen"},
|
||
},
|
||
},
|
||
}
|
||
|
||
var advancedTests = []struct {
|
||
in string
|
||
out netCfg
|
||
}{
|
||
{ in: "hostname=p2-jbs537 vlan=66 vlan=35 nativevlan=218",
|
||
out: netCfg {
|
||
Host: "p2-jbs537",
|
||
Vlan: []int {66, 35},
|
||
Native: 218,
|
||
},
|
||
},
|
||
}
|
||
|
||
var structTests = []struct {
|
||
in string
|
||
out screenCfg
|
||
}{
|
||
{
|
||
in: "Title='Hollywood movie' Width=640 Height=400 A=8",
|
||
out: screenCfg {
|
||
Title: "Hollywood movie",
|
||
Width: 640,
|
||
Height: 400,
|
||
A: 8,
|
||
},
|
||
},
|
||
}
|
||
|
||
var mapTests = []struct {
|
||
in string
|
||
out map[string] string
|
||
}{
|
||
{
|
||
in: "ipnet=murray–hill ip=135.104.0.0 ipmask=255.255.0.0",
|
||
out: map[string] string {
|
||
"ipnet": "murray-hill",
|
||
"ip": "135.104.0.0",
|
||
"ipmask": "255.255.0.0",
|
||
},
|
||
},
|
||
}
|
||
|
||
func TestStruct(t *testing.T) {
|
||
var cfg screenCfg
|
||
|
||
for _,tt := range structTests {
|
||
if err := Parse([]byte(tt.in), &cfg); err != nil {
|
||
t.Error(err)
|
||
} else if cfg != tt.out {
|
||
t.Errorf("Got %v, wanted %v", cfg, tt.out)
|
||
}
|
||
t.Logf("%s == %v", tt.in, cfg)
|
||
}
|
||
}
|
||
func TestMap(t *testing.T) {
|
||
var net map[string] string
|
||
for _,tt := range mapTests {
|
||
if err := Parse([]byte(tt.in), &net); err != nil {
|
||
t.Error(err)
|
||
} else if fmt.Sprint(net) != fmt.Sprint(tt.out) {
|
||
t.Errorf("Got %v, wanted %v", net, tt.out)
|
||
}
|
||
t.Logf("%s == %v", tt.in, net)
|
||
}
|
||
}
|
||
|
||
func TestAdvanced(t *testing.T) {
|
||
var net netCfg
|
||
for _,tt := range advancedTests {
|
||
if err := Parse([]byte(tt.in), &net); err != nil {
|
||
t.Error(err)
|
||
} else if fmt.Sprint(tt.out) != fmt.Sprint(net) {
|
||
t.Errorf("Got %v, wanted %v", net, tt.out)
|
||
}
|
||
t.Logf("%s == %v", tt.in, net)
|
||
}
|
||
}
|
||
|
||
func TestMultiMap(t *testing.T) {
|
||
var m map[string] []string
|
||
for _,tt := range multiMap {
|
||
if err := Parse([]byte(tt.in), &m); err != nil {
|
||
t.Error(err)
|
||
} else if fmt.Sprint(tt.out) != fmt.Sprint(m) {
|
||
t.Errorf("Got %v, wanted %v", m, tt.out)
|
||
}
|
||
t.Logf("%s == %v", tt.in, m)
|
||
}
|
||
}
|
||
|
||
func netEqual(t *testing.T, n1, n2 netCfg) bool {
|
||
if len(n1.Vlan) != len(n2.Vlan) {
|
||
return false
|
||
}
|
||
for i := range n1.Vlan {
|
||
if n1.Vlan[i] != n2.Vlan[i] {
|
||
return false
|
||
}
|
||
}
|
||
return n1.Host == n2.Host && n1.Native == n2.Native
|
||
}
|
||
|
||
func mapEqual(t *testing.T, m1, m2 map[string] string) bool {
|
||
for k := range m1 {
|
||
if m1[k] != m2[k] {
|
||
t.Logf("%v != %v", m1[k], m2[k])
|
||
return false
|
||
}
|
||
}
|
||
return true
|
||
}
|
||
package ndb
|
||
|
||
import (
|
||
"testing"
|
||
"bytes"
|
||
)
|
||
|
||
var parseTests = []struct {
|
||
in []byte
|
||
out []pair
|
||
}{
|
||
{
|
||
in: []byte("key1=val1 key2=val2 key3=val3"),
|
||
out: []pair {
|
||
{[]byte("key1"),[]byte("val1")},
|
||
{[]byte("key2"),[]byte("val2")},
|
||
{[]byte("key3"),[]byte("val3")}},
|
||
},
|
||
{
|
||
in: []byte("title='Some value with spaces' width=340 height=200"),
|
||
out: []pair {
|
||
{[]byte("title"),[]byte("Some value with spaces")},
|
||
{[]byte("width"),[]byte("340")},
|
||
{[]byte("height"),[]byte("200")}},
|
||
},
|
||
{
|
||
in: []byte("title='Dave''s pasta' sq=Davis cost=$$"),
|
||
out: []pair {
|
||
{[]byte("title"),[]byte("Dave's pasta")},
|
||
{[]byte("sq"),[]byte("Davis")},
|
||
{[]byte("cost"),[]byte("$$")}},
|
||
},
|
||
{
|
||
in: []byte("action=''bradley key=jay mod=ctrl+alt+shift"),
|
||
out: []pair {
|
||
{[]byte("action"),[]byte("'bradley")},
|
||
{[]byte("key"),[]byte("jay")},
|
||
{[]byte("mod"),[]byte("ctrl+alt+shift")}},
|
||
},
|
||
{
|
||
in: []byte("action=reload key='' mod=ctrl+alt+shift"),
|
||
out: []pair {
|
||
{[]byte("action"),[]byte("reload")},
|
||
{[]byte("key"),[]byte("'")},
|
||
{[]byte("mod"),[]byte("ctrl+alt+shift")}},
|
||
},
|
||
{
|
||
in: []byte("s='spaces and '' quotes'"),
|
||
out: []pair {
|
||
{[]byte("s"),[]byte("spaces and ' quotes")}},
|
||
},
|
||
{
|
||
in: []byte("esc='Use '''' to escape a '''"),
|
||
out: []pair {
|
||
{[]byte("esc"),[]byte("Use '' to escape a '")}},
|
||
},
|
||
|
||
}
|
||
|
||
func Test_parsing(t *testing.T) {
|
||
for i,tt := range parseTests {
|
||
d := NewDecoder(bytes.NewReader(tt.in))
|
||
p,err := d.getPairs()
|
||
if err != nil {
|
||
t.Error(err)
|
||
t.FailNow()
|
||
} else {
|
||
for j := range tt.out {
|
||
if j > len(p) || !match(p[j],tt.out[j]) {
|
||
t.Errorf("%d: getPairs %s => %v, want %v",i, tt.in, p, tt.out)
|
||
t.FailNow()
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func match(p1, p2 pair) bool {
|
||
return (bytes.Compare(p1.attr, p2.attr) == 0) && (bytes.Compare(p1.val, p2.val) == 0)
|
||
}
|