summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main_test.go47
1 files changed, 24 insertions, 23 deletions
diff --git a/main_test.go b/main_test.go
index 0927a58..287c792 100644
--- a/main_test.go
+++ b/main_test.go
@@ -58,13 +58,23 @@ func md5hex(buf []byte) string {
return hash
}
-// Read the whole file and return number of bytes read
-func readSize(fn string) int {
- buf, err := ioutil.ReadFile(fn)
+// Verify that the file size equals "want". This checks:
+// 1) Size reported by Stat()
+// 2) Number of bytes returned when reading the whole file
+func verifySize(t *testing.T, path string, want int) {
+ buf, err := ioutil.ReadFile(path)
if err != nil {
- fmt.Printf("ReadFile: %v\n", err)
+ t.Errorf("ReadFile failed: %v", err)
+ } else if len(buf) != want {
+ t.Errorf("wrong read size: got=%d want=%d", len(buf), want)
+ }
+
+ fi, err := os.Stat(path)
+ if err != nil {
+ t.Errorf("Stat failed: %v", err)
+ } else if fi.Size() != int64(want) {
+ t.Errorf("wrong stat file size, got=%d want=%d", fi.Size(), want)
}
- return len(buf)
}
// This is the entry point for the tests
@@ -111,20 +121,7 @@ func testWriteN(t *testing.T, fn string, n int) string {
}
file.Close()
- fi, err := os.Stat(plainDir + fn)
- if err != nil {
- t.Errorf("Stat on file %s failed: %v", fn, err)
- } else {
- if fi.Size() != int64(n) {
- t.Errorf("Wrong stat file size, got=%d want=%d", fi.Size(), n)
- }
- }
-
- rs := readSize(plainDir + fn)
- if rs != n {
- t.Errorf("Wrong read file fize, got=%d want=%d", rs, n)
- }
-
+ verifySize(t, plainDir + fn, n)
bin := md5.Sum(d)
hashWant := hex.EncodeToString(bin[:])
@@ -173,23 +170,27 @@ func TestTruncate(t *testing.T) {
}
// Grow to two blocks
file.Truncate(7000)
+ verifySize(t, fn, 7000)
if md5fn(fn) != "95d4ec7038e3e4fdbd5f15c34c3f0b34" {
- t.Errorf("Fail 7000")
+ t.Errorf("wrong content")
}
// Shrink - needs RMW
file.Truncate(6999)
+ verifySize(t, fn, 6999)
if md5fn(fn) != "35fd15873ec6c35380064a41b9b9683b" {
- t.Errorf("Fail 6999")
+ t.Errorf("wrong content")
}
// Shrink to one partial block
file.Truncate(465)
+ verifySize(t, fn, 465)
if md5fn(fn) != "a1534d6e98a6b21386456a8f66c55260" {
- t.Errorf("Fail 465")
+ t.Errorf("wrong content")
}
// Grow to exactly one block
file.Truncate(4096)
+ verifySize(t, fn, 4096)
if md5fn(fn) != "620f0b67a91f7f74151bc5be745b7110" {
- t.Errorf("Fail 4096")
+ t.Errorf("wrong content")
}
}