mirror of
https://github.com/safedep/vet.git
synced 2025-12-11 01:01:10 -06:00
be81848cc0
feat: Add support for DG JSON serialization test: Add test case for dependency graph structure test: Add test case for dependency graph structure (GetDependents) test: Add test case for dependency graph structure (PathToRoot, JSON) refactor: Use factory to initialize package manifest with dependency graph
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type dgTestNode struct {
|
|
Name string `json:"Name"`
|
|
}
|
|
|
|
func (n *dgTestNode) Id() string {
|
|
return n.Name
|
|
}
|
|
|
|
func dependencyGraphAddTestData(dg *DependencyGraph[*dgTestNode]) {
|
|
dg.AddDependency(&dgTestNode{Name: "a"}, &dgTestNode{Name: "b"})
|
|
dg.AddDependency(&dgTestNode{Name: "a"}, &dgTestNode{Name: "c"})
|
|
dg.AddDependency(&dgTestNode{Name: "b"}, &dgTestNode{Name: "c"})
|
|
dg.AddDependency(&dgTestNode{Name: "c"}, &dgTestNode{Name: "d"})
|
|
}
|
|
|
|
func TestDependencyGraphIsPresent(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
assert.False(t, dg.Present())
|
|
|
|
dg.SetPresent(true)
|
|
assert.True(t, dg.Present())
|
|
}
|
|
|
|
func TestDependencyGraphGetDependencies(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
|
|
assert.Equal(t, []*dgTestNode{{Name: "b"}, {Name: "c"}}, dg.GetDependencies(&dgTestNode{Name: "a"}))
|
|
assert.Equal(t, []*dgTestNode{{Name: "c"}}, dg.GetDependencies(&dgTestNode{Name: "b"}))
|
|
assert.Equal(t, []*dgTestNode{{Name: "d"}}, dg.GetDependencies(&dgTestNode{Name: "c"}))
|
|
assert.Equal(t, []*dgTestNode{}, dg.GetDependencies(&dgTestNode{Name: "d"}))
|
|
}
|
|
|
|
func TestDependencyGraphGetDependents(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
|
|
assert.Equal(t, []*dgTestNode{}, dg.GetDependents(&dgTestNode{Name: "a"}))
|
|
assert.Equal(t, []*dgTestNode{{Name: "a"}}, dg.GetDependents(&dgTestNode{Name: "b"}))
|
|
assert.Equal(t, []*dgTestNode{{Name: "a"}, {Name: "b"}}, dg.GetDependents(&dgTestNode{Name: "c"}))
|
|
assert.Equal(t, []*dgTestNode{{Name: "c"}}, dg.GetDependents(&dgTestNode{Name: "d"}))
|
|
}
|
|
|
|
func TestDependencyGraphGetNodes(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
|
|
nodes := dg.GetNodes()
|
|
assert.Contains(t, nodes, &dgTestNode{Name: "a"})
|
|
assert.Contains(t, nodes, &dgTestNode{Name: "b"})
|
|
assert.Contains(t, nodes, &dgTestNode{Name: "c"})
|
|
assert.Contains(t, nodes, &dgTestNode{Name: "d"})
|
|
}
|
|
|
|
func TestDependencyGraphPathToRoot(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
|
|
assert.Equal(t,
|
|
[]*dgTestNode{
|
|
{Name: "d"},
|
|
{Name: "c"},
|
|
{Name: "a"},
|
|
}, dg.PathToRoot(&dgTestNode{Name: "d"}))
|
|
}
|
|
|
|
func TestDependencyGraphMarshalJSON(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
dg.SetPresent(true)
|
|
|
|
json, err := json.Marshal(dg)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "{\"present\":true,\"nodes\":{\"a\":{\"data\":{\"Name\":\"a\"},\"children\":[{\"Name\":\"b\"},{\"Name\":\"c\"}]},\"b\":{\"data\":{\"Name\":\"b\"},\"children\":[{\"Name\":\"c\"}]},\"c\":{\"data\":{\"Name\":\"c\"},\"children\":[{\"Name\":\"d\"}]},\"d\":{\"data\":{\"Name\":\"d\"},\"children\":[]}}}", string(json))
|
|
}
|
|
|
|
func TestDependencyGraphUnmarshalJSON(t *testing.T) {
|
|
dg := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg)
|
|
|
|
dependencyGraphAddTestData(dg)
|
|
dg.SetPresent(true)
|
|
|
|
data, err := json.Marshal(dg)
|
|
assert.Nil(t, err)
|
|
|
|
dg2 := NewDependencyGraph[*dgTestNode]()
|
|
assert.NotNil(t, dg2)
|
|
|
|
err = json.Unmarshal(data, dg2)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, dg, dg2)
|
|
}
|