parent
2ac6c072ee
commit
d0b7a52caa
@ -0,0 +1,112 @@ |
||||
package tree |
||||
|
||||
import ( |
||||
"reflect" |
||||
"testing" |
||||
) |
||||
|
||||
func TestSplit(t *testing.T) { |
||||
type args struct { |
||||
code string |
||||
length int |
||||
} |
||||
tests := []struct { |
||||
name string |
||||
args args |
||||
wantRes []string |
||||
}{ |
||||
{ |
||||
name: "test1", |
||||
args: args{ |
||||
code: "A00001B00001C00001", |
||||
length: 6, |
||||
}, |
||||
wantRes: []string{"A00001B00001C00001", "A00001B00001", "A00001"}, |
||||
}, |
||||
} |
||||
for _, tt := range tests { |
||||
t.Run(tt.name, func(t *testing.T) { |
||||
if gotRes := Split(tt.args.code, tt.args.length); !reflect.DeepEqual(gotRes, tt.wantRes) { |
||||
t.Errorf("Split() = %v, want %v", gotRes, tt.wantRes) |
||||
} |
||||
}) |
||||
} |
||||
} |
||||
|
||||
func TestTree(t *testing.T) { |
||||
type args[T any] struct { |
||||
data []T |
||||
parent string |
||||
length int |
||||
pathFunc func(d T) string |
||||
} |
||||
type testCase[T any] struct { |
||||
name string |
||||
args args[T] |
||||
wantRes []Node[T] |
||||
} |
||||
|
||||
type D struct { |
||||
Code string |
||||
Name string |
||||
} |
||||
|
||||
tests := []testCase[D]{ |
||||
{ |
||||
name: "", |
||||
args: args[D]{ |
||||
data: []D{{Code: "A00001", Name: "A00001"}, {Code: "A00001B00001", Name: "A00001B00001"}, {Code: "A00001B00001C00001", Name: "A00001B00001C00001"}}, |
||||
parent: "", |
||||
length: 6, |
||||
pathFunc: func(d D) string { |
||||
return d.Code |
||||
}, |
||||
}, |
||||
wantRes: []Node[D]{ |
||||
{ |
||||
Data: D{Code: "A00001", Name: "A00001"}, |
||||
Children: []Node[D]{ |
||||
{ |
||||
Data: D{Code: "A00001B00001", Name: "A00001B00001"}, |
||||
Children: []Node[D]{ |
||||
{ |
||||
Data: D{Code: "A00001B00001C00001", Name: "A00001B00001C00001"}, |
||||
Children: nil, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
{ |
||||
name: "", |
||||
args: args[D]{ |
||||
data: []D{{Code: "A00001B00001", Name: "A00001B00001"}, {Code: "A00001B00001C00001", Name: "A00001B00001C00001"}}, |
||||
parent: "A00001", |
||||
length: 6, |
||||
pathFunc: func(d D) string { |
||||
return d.Code |
||||
}, |
||||
}, |
||||
wantRes: []Node[D]{ |
||||
{ |
||||
Data: D{Code: "A00001B00001", Name: "A00001B00001"}, |
||||
Children: []Node[D]{ |
||||
{ |
||||
Data: D{Code: "A00001B00001C00001", Name: "A00001B00001C00001"}, |
||||
Children: nil, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
}, |
||||
} |
||||
for _, tt := range tests { |
||||
t.Run(tt.name, func(t *testing.T) { |
||||
if gotRes := Tree(tt.args.data, tt.args.parent, tt.args.length, tt.args.pathFunc); !reflect.DeepEqual(gotRes, tt.wantRes) { |
||||
t.Errorf("Tree() = %v, want %v", gotRes, tt.wantRes) |
||||
} |
||||
}) |
||||
} |
||||
} |
Loading…
Reference in new issue