65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package filetree
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestAssignDiffType(t *testing.T) {
|
|
tree := NewFileTree()
|
|
tree.AddPath("/usr", BlankFileChangeInfo("/usr", Changed))
|
|
if tree.Root.Children["usr"].Data.DiffType != Changed {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestMergeDiffTypes(t *testing.T) {
|
|
a := Unchanged
|
|
b := Unchanged
|
|
merged := a.merge(b)
|
|
if merged != Unchanged {
|
|
t.Errorf("Expected Unchaged (0) but got %v", merged)
|
|
}
|
|
a = Changed
|
|
b = Unchanged
|
|
merged = a.merge(b)
|
|
if merged != Changed {
|
|
t.Errorf("Expected Unchaged (0) but got %v", merged)
|
|
}
|
|
}
|
|
|
|
func TestDiffTypeFromChildren(t *testing.T) {
|
|
tree := NewFileTree()
|
|
tree.AddPath("/usr", BlankFileChangeInfo("/usr", Unchanged))
|
|
info1 := BlankFileChangeInfo("/usr/bin", Added)
|
|
tree.AddPath("/usr/bin", info1)
|
|
info2 := BlankFileChangeInfo("/usr/bin2", Removed)
|
|
tree.AddPath("/usr/bin2", info2)
|
|
tree.Root.Children["usr"].deriveDiffType(Unchanged)
|
|
if tree.Root.Children["usr"].Data.DiffType != Changed {
|
|
t.Errorf("Expected Changed but got %v", tree.Root.Children["usr"].Data.DiffType)
|
|
}
|
|
}
|
|
|
|
func AssertDiffType(node *FileNode, expectedDiffType DiffType, t *testing.T) error {
|
|
if node.Data == nil {
|
|
t.Errorf("Expected *FileChangeInfo but got nil at Path %s", node.Path())
|
|
return fmt.Errorf("expected *FileChangeInfo but got nil at Path %s", node.Path())
|
|
}
|
|
if node.Data.DiffType != expectedDiffType {
|
|
t.Errorf("Expecting node at %s to have DiffType %v, but had %v", node.Path(), expectedDiffType, node.Data.DiffType)
|
|
return fmt.Errorf("Assertion failed")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func BlankFileChangeInfo(path string, diffType DiffType) (f *FileChangeInfo) {
|
|
result := FileChangeInfo{
|
|
Path: path,
|
|
Typeflag: 1,
|
|
MD5sum: [16]byte{1, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0},
|
|
DiffType: diffType,
|
|
}
|
|
return &result
|
|
}
|