You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.3 KiB
72 lines
1.3 KiB
package main |
|
|
|
import ( |
|
"crypto/md5" |
|
"encoding/hex" |
|
"io" |
|
"io/ioutil" |
|
"os" |
|
"path/filepath" |
|
"strings" |
|
"testing" |
|
) |
|
|
|
const TESTDIR = ".test" |
|
|
|
func TestBuild(t *testing.T) { |
|
files, _ := ioutil.ReadDir("testdata") |
|
for _, f := range files { |
|
if f.IsDir() { |
|
testBuild(filepath.Join("testdata", f.Name()), t) |
|
} |
|
} |
|
} |
|
|
|
func testBuild(path string, t *testing.T) { |
|
wd, _ := os.Getwd() |
|
os.Chdir(path) |
|
args := os.Args[:] |
|
os.Args = []string{"zs", "build"} |
|
t.Log("--- BUILD", path) |
|
main() |
|
|
|
compare(PUBDIR, TESTDIR, t) |
|
|
|
os.Chdir(wd) |
|
os.Args = args |
|
} |
|
|
|
func compare(pub, test string, t *testing.T) { |
|
a := md5dir(pub) |
|
b := md5dir(test) |
|
for k, v := range a { |
|
if s, ok := b[k]; !ok { |
|
t.Error("Unexpected file:", k, v) |
|
} else if s != v { |
|
t.Error("Different file:", k, v, s) |
|
} else { |
|
t.Log("Matching file", k, v) |
|
} |
|
} |
|
for k, v := range b { |
|
if _, ok := a[k]; !ok { |
|
t.Error("Missing file:", k, v) |
|
} |
|
} |
|
} |
|
|
|
func md5dir(path string) map[string]string { |
|
files := map[string]string{} |
|
filepath.Walk(path, func(s string, info os.FileInfo, err error) error { |
|
if err == nil && !info.IsDir() { |
|
if f, err := os.Open(s); err == nil { |
|
defer f.Close() |
|
hash := md5.New() |
|
io.Copy(hash, f) |
|
files[strings.TrimPrefix(s, path)] = hex.EncodeToString(hash.Sum(nil)) |
|
} |
|
} |
|
return nil |
|
}) |
|
return files |
|
}
|
|
|