1
0
Fork 0
mirror of https://github.com/vbatts/go-mtree.git synced 2024-12-22 22:06:29 +00:00
go-mtree/mtree_test.go
Vincent Batts 34b9f4dc46 hierarchy: testing works
Signed-off-by: Vincent Batts <vbatts@hashbangbash.com>
2016-04-05 17:16:44 -04:00

68 lines
1.2 KiB
Go

package mtree
import (
"io/ioutil"
"os"
"testing"
)
var (
testFiles = []string{"testdata/source.mtree"}
numEntries = map[EntryType]int{
FullType: 0,
RelativeType: 45,
CommentType: 37,
SpecialType: 7,
DotDotType: 17,
BlankType: 34,
}
expectedLength = int64(7887)
)
func TestParser(t *testing.T) {
for _, file := range testFiles {
func() {
fh, err := os.Open(file)
if err != nil {
t.Error(err)
return
}
defer fh.Close()
dh, err := ParseSpec(fh)
if err != nil {
t.Error(err)
}
gotNums := countTypes(dh)
for typ, num := range numEntries {
if gNum, ok := gotNums[typ]; ok {
if num != gNum {
t.Errorf("for type %s: expected %d, got %d", typ, num, gNum)
}
}
}
i, err := dh.WriteTo(ioutil.Discard)
if err != nil {
t.Error(err)
}
if i != expectedLength {
t.Errorf("expected to write %d, but wrote %d", expectedLength, i)
}
}()
}
}
func countTypes(dh *DirectoryHierarchy) map[EntryType]int {
nT := map[EntryType]int{}
for i := range dh.Entries {
typ := dh.Entries[i].Type
if _, ok := nT[typ]; !ok {
nT[typ] = 1
} else {
nT[typ]++
}
}
return nT
}