From 99dcd25a7ea46003d8a0a4754dd9b270332c1979 Mon Sep 17 00:00:00 2001 From: Vincent Batts Date: Wed, 6 Apr 2016 12:45:08 -0400 Subject: [PATCH] walk: test passes and validates parse Signed-off-by: Vincent Batts --- walk_test.go | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/walk_test.go b/walk_test.go index c37fb87..75c404a 100644 --- a/walk_test.go +++ b/walk_test.go @@ -2,6 +2,7 @@ package mtree import ( "io/ioutil" + "os" "testing" ) @@ -10,26 +11,27 @@ func TestWalk(t *testing.T) { if err != nil { t.Fatal(err) } - - //log.Fatalf("%#v", dh) + numEntries = countTypes(dh) fh, err := ioutil.TempFile("", "walk.") if err != nil { t.Fatal(err) } + defer os.Remove(fh.Name()) + defer fh.Close() if _, err = dh.WriteTo(fh); err != nil { - t.Error(err) + t.Fatal(err) } - fh.Close() - t.Fatal(fh.Name()) - //os.Remove(fh.Name()) -} - -func TestReadNames(t *testing.T) { - names, err := readOrderedDirNames(".") - if err != nil { - t.Error(err) + if _, err := fh.Seek(0, 0); err != nil { + t.Fatal(err) + } + if dh, err = ParseSpec(fh); err != nil { + t.Fatal(err) + } + for k, v := range countTypes(dh) { + if numEntries[k] != v { + t.Errorf("for type %s: expected %d, got %d", k, numEntries[k], v) + } } - t.Errorf("names: %q", names) }