343 lines
9.4 KiB
Go
343 lines
9.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// perform integration test using the compiled binary
|
|
|
|
var testDir = "/tmp/chkbit"
|
|
|
|
func getCmd() string {
|
|
_, filename, _, _ := runtime.Caller(0)
|
|
prjRoot := filepath.Dir(filepath.Dir(filename))
|
|
return filepath.Join(prjRoot, "chkbit")
|
|
}
|
|
|
|
func checkOut(t *testing.T, sout string, expected string) {
|
|
if !strings.Contains(sout, expected) {
|
|
t.Errorf("Expected '%s' in output, got '%s'\n", expected, sout)
|
|
}
|
|
}
|
|
|
|
func checkNotOut(t *testing.T, sout string, notExpected string) {
|
|
if strings.Contains(sout, notExpected) {
|
|
t.Errorf("Did not expect '%s' in output, got '%s'\n", notExpected, sout)
|
|
}
|
|
}
|
|
|
|
// misc files
|
|
|
|
var (
|
|
startList = []string{"time", "year", "people", "way", "day", "thing"}
|
|
wordList = []string{"life", "world", "school", "state", "family", "student", "group", "country", "problem", "hand", "part", "place", "case", "week", "company", "system", "program", "work", "government", "number", "night", "point", "home", "water", "room", "mother", "area", "money", "story", "fact", "month", "lot", "right", "study", "book", "eye", "job", "word", "business", "issue", "side", "kind", "head", "house", "service", "friend", "father", "power", "hour", "game", "line", "end", "member", "law", "car", "city", "community", "name", "president", "team", "minute", "idea", "kid", "body", "information", "back", "face", "others", "level", "office", "door", "health", "person", "art", "war", "history", "party", "result", "change", "morning", "reason", "research", "moment", "air", "teacher", "force", "education"}
|
|
extList = []string{"txt", "md", "pdf", "jpg", "jpeg", "png", "mp4", "mp3", "csv"}
|
|
startDate = time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
endDate = time.Date(2024, 12, 1, 0, 0, 0, 0, time.UTC)
|
|
dateList = []time.Time{}
|
|
wordIdx = 0
|
|
extIdx = 0
|
|
dateIdx = 0
|
|
)
|
|
|
|
func nextWord() string {
|
|
word := wordList[wordIdx%len(wordList)]
|
|
wordIdx++
|
|
return word
|
|
}
|
|
|
|
func nextExt() string {
|
|
ext := extList[extIdx%len(extList)]
|
|
extIdx++
|
|
return ext
|
|
}
|
|
|
|
func setDate(filename string, r int) {
|
|
date := dateList[dateIdx%len(dateList)]
|
|
m := 17 * dateIdx / len(dateList)
|
|
date = date.Add(time.Duration(m) * time.Hour)
|
|
dateIdx++
|
|
os.Chtimes(filename, date, date)
|
|
}
|
|
|
|
func genFile(path string, size int) {
|
|
os.WriteFile(path, make([]byte, size), 0644)
|
|
setDate(path, size*size)
|
|
}
|
|
|
|
func genFiles(dir string, a int) {
|
|
os.MkdirAll(dir, 0755)
|
|
for i := 1; i <= 5; i++ {
|
|
size := a*i*wordIdx*100 + extIdx
|
|
file := nextWord() + "-" + nextWord()
|
|
|
|
if i%3 == 0 {
|
|
file += "-" + nextWord()
|
|
}
|
|
|
|
file += "." + nextExt()
|
|
genFile(filepath.Join(dir, file), size)
|
|
}
|
|
}
|
|
|
|
func genDir(root string) {
|
|
for _, start := range startList {
|
|
|
|
for i := 1; i <= 5; i++ {
|
|
dir := filepath.Join(root, start, nextWord())
|
|
genFiles(dir, 1)
|
|
|
|
if wordIdx%3 == 0 {
|
|
dir = filepath.Join(dir, nextWord())
|
|
genFiles(dir, 1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func setupMiscFiles() {
|
|
|
|
var c int64 = 50
|
|
interval := (int64)(endDate.Sub(startDate).Seconds()) / c
|
|
for i := range make([]int64, c) {
|
|
dateList = append(dateList, startDate.Add(time.Duration(interval*(int64)(i))*time.Second))
|
|
}
|
|
|
|
root := filepath.Join(testDir, "root")
|
|
if err := os.RemoveAll(testDir); err != nil {
|
|
fmt.Println("Failed to clean", err)
|
|
panic(err)
|
|
}
|
|
|
|
genDir(root)
|
|
|
|
os.MkdirAll(filepath.Join(root, "day/car/empty"), 0755)
|
|
|
|
rootPeople := filepath.Join(root, "people")
|
|
testPeople := filepath.Join(testDir, "people")
|
|
|
|
err := os.Rename(rootPeople, testPeople)
|
|
if err != nil {
|
|
fmt.Println("Rename failed", err)
|
|
panic(err)
|
|
}
|
|
|
|
err = os.Symlink(testPeople, rootPeople)
|
|
if err != nil {
|
|
fmt.Println("Symlink failed", err)
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func TestRoot(t *testing.T) {
|
|
setupMiscFiles()
|
|
|
|
tool := getCmd()
|
|
root := filepath.Join(testDir, "root")
|
|
|
|
// update index, no recourse
|
|
t.Run("no-recourse", func(t *testing.T) {
|
|
cmd := exec.Command(tool, "-umR", filepath.Join(root, "day/office"))
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "Processed 5 files")
|
|
checkOut(t, sout, "- 1 directory was updated")
|
|
checkOut(t, sout, "- 5 file hashes were added")
|
|
checkOut(t, sout, "- 0 file hashes were updated")
|
|
checkNotOut(t, sout, "removed")
|
|
})
|
|
|
|
// update remaining index from root
|
|
t.Run("update-remaining", func(t *testing.T) {
|
|
cmd := exec.Command(tool, "-um", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "Processed 300 files")
|
|
checkOut(t, sout, "- 66 directories were updated")
|
|
checkOut(t, sout, "- 295 file hashes were added")
|
|
checkOut(t, sout, "- 0 file hashes were updated")
|
|
checkNotOut(t, sout, "removed")
|
|
})
|
|
|
|
// delete files, check for missing
|
|
t.Run("delete", func(t *testing.T) {
|
|
os.RemoveAll(filepath.Join(root, "thing/change"))
|
|
os.Remove(filepath.Join(root, "time/hour/minute/body-information.csv"))
|
|
|
|
cmd := exec.Command(tool, "-m", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "del /tmp/chkbit/root/thing/change/")
|
|
checkOut(t, sout, "2 files/directories would have been removed")
|
|
})
|
|
|
|
// do not report missing without -m
|
|
t.Run("no-missing", func(t *testing.T) {
|
|
cmd := exec.Command(tool, root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkNotOut(t, sout, "del ")
|
|
checkNotOut(t, sout, "removed")
|
|
})
|
|
|
|
// check for missing and update
|
|
t.Run("missing", func(t *testing.T) {
|
|
cmd := exec.Command(tool, "-um", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "del /tmp/chkbit/root/thing/change/")
|
|
checkOut(t, sout, "2 files/directories have been removed")
|
|
})
|
|
|
|
// check again
|
|
t.Run("repeat", func(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
cmd := exec.Command(tool, "-uv", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "Processed 289 files")
|
|
checkNotOut(t, sout, "removed")
|
|
checkNotOut(t, sout, "updated")
|
|
checkNotOut(t, sout, "added")
|
|
}
|
|
})
|
|
|
|
// add files only
|
|
t.Run("add-only", func(t *testing.T) {
|
|
|
|
genFiles(filepath.Join(root, "way/add"), 99)
|
|
genFile(filepath.Join(root, "time/add-file.txt"), 500)
|
|
// modify existing, will not be reported:
|
|
genFile(filepath.Join(root, "way/job/word-business.mp3"), 500)
|
|
|
|
cmd := exec.Command(tool, "-a", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "Processed 6 files")
|
|
checkOut(t, sout, "- 3 directories were updated")
|
|
checkOut(t, sout, "- 6 file hashes were added")
|
|
checkOut(t, sout, "- 0 file hashes were updated")
|
|
})
|
|
|
|
// update remaining
|
|
t.Run("update-remaining-add", func(t *testing.T) {
|
|
cmd := exec.Command(tool, "-u", root)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
}
|
|
sout := string(out)
|
|
checkOut(t, sout, "Processed 295 files")
|
|
checkOut(t, sout, "- 1 directory was updated")
|
|
checkOut(t, sout, "- 0 file hashes were added")
|
|
checkOut(t, sout, "- 1 file hash was updated")
|
|
})
|
|
}
|
|
|
|
func TestDMG(t *testing.T) {
|
|
|
|
testDmg := filepath.Join(testDir, "test_dmg")
|
|
if err := os.RemoveAll(testDmg); err != nil {
|
|
fmt.Println("Failed to clean", err)
|
|
panic(err)
|
|
}
|
|
if err := os.MkdirAll(testDmg, 0755); err != nil {
|
|
fmt.Println("Failed to create test directory", err)
|
|
panic(err)
|
|
}
|
|
|
|
if err := os.Chdir(testDmg); err != nil {
|
|
fmt.Println("Failed to cd test directory", err)
|
|
panic(err)
|
|
}
|
|
|
|
tool := getCmd()
|
|
testFile := filepath.Join(testDmg, "test.txt")
|
|
t1, _ := time.Parse(time.RFC3339, "2022-02-01T11:00:00Z")
|
|
t2, _ := time.Parse(time.RFC3339, "2022-02-01T12:00:00Z")
|
|
t3, _ := time.Parse(time.RFC3339, "2022-02-01T13:00:00Z")
|
|
|
|
// create test and set the modified time"
|
|
t.Run("create", func(t *testing.T) {
|
|
os.WriteFile(testFile, []byte("foo1"), 0644)
|
|
os.Chtimes(testFile, t2, t2)
|
|
|
|
cmd := exec.Command(tool, "-u", ".")
|
|
if out, err := cmd.Output(); err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
} else {
|
|
checkOut(t, string(out), "new test.txt")
|
|
}
|
|
})
|
|
|
|
// update test with different content & old modified (expect 'old')"
|
|
t.Run("expect-old", func(t *testing.T) {
|
|
os.WriteFile(testFile, []byte("foo2"), 0644)
|
|
os.Chtimes(testFile, t1, t1)
|
|
|
|
cmd := exec.Command(tool, "-u", ".")
|
|
if out, err := cmd.Output(); err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
} else {
|
|
checkOut(t, string(out), "old test.txt")
|
|
}
|
|
})
|
|
|
|
// update test & new modified (expect 'upd')"
|
|
t.Run("expect-upd", func(t *testing.T) {
|
|
os.WriteFile(testFile, []byte("foo3"), 0644)
|
|
os.Chtimes(testFile, t3, t3)
|
|
|
|
cmd := exec.Command(tool, "-u", ".")
|
|
if out, err := cmd.Output(); err != nil {
|
|
t.Fatalf("failed with '%s'\n", err)
|
|
} else {
|
|
checkOut(t, string(out), "upd test.txt")
|
|
}
|
|
})
|
|
|
|
// Now update test with the same modified to simulate damage (expect DMG)"
|
|
t.Run("expect-DMG", func(t *testing.T) {
|
|
os.WriteFile(testFile, []byte("foo4"), 0644)
|
|
os.Chtimes(testFile, t3, t3)
|
|
|
|
cmd := exec.Command(tool, "-u", ".")
|
|
if out, err := cmd.Output(); err != nil {
|
|
if cmd.ProcessState.ExitCode() != 1 {
|
|
t.Fatalf("expected to fail with exit code 1 vs %d!", cmd.ProcessState.ExitCode())
|
|
}
|
|
checkOut(t, string(out), "DMG test.txt")
|
|
} else {
|
|
t.Fatal("expected to fail!")
|
|
}
|
|
})
|
|
}
|