12a09716fc
* [TASK] [models] graph.json should only save nodes with links * [TASK] [models] graph.json should only save nodes with links and there main mac (with tests)
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/FreifunkBremen/yanic/data"
|
|
)
|
|
|
|
type TestNode struct {
|
|
Nodeinfo *data.NodeInfo `json:"nodeinfo"`
|
|
Neighbours *data.Neighbours `json:"neighbours"`
|
|
}
|
|
|
|
func TestGenerateGraph(t *testing.T) {
|
|
assert := assert.New(t)
|
|
nodes := testGetNodesByFile("node1.json", "node2.json", "node3.json", "node4.json")
|
|
|
|
graph := nodes.BuildGraph()
|
|
assert.NotNil(graph)
|
|
assert.Equal(1, graph.Version, "Wrong Version")
|
|
assert.NotNil(graph.Batadv, "no Batadv")
|
|
assert.Equal(false, graph.Batadv.Directed, "directed batadv")
|
|
assert.Len(graph.Batadv.Links, 3, "wrong Links count")
|
|
assert.Equal(4, testNodesCountWithLinks(graph.Batadv.Links), "wrong unneed nodes in graph")
|
|
assert.Len(graph.Batadv.Nodes, 4, "wrong Nodes count")
|
|
// TODO more tests required
|
|
}
|
|
|
|
func testGetNodesByFile(files ...string) *Nodes {
|
|
|
|
nodes := &Nodes{
|
|
List: make(map[string]*Node),
|
|
}
|
|
|
|
for _, file := range files {
|
|
node := testGetNodeByFile(file)
|
|
nodes.Update(file, &data.ResponseData{
|
|
NodeInfo: node.Nodeinfo,
|
|
Neighbours: node.Neighbours,
|
|
})
|
|
}
|
|
|
|
return nodes
|
|
}
|
|
|
|
func testGetNodeByFile(filename string) *Node {
|
|
testnode := &TestNode{}
|
|
testfile(filename, testnode)
|
|
return &Node{
|
|
Nodeinfo: testnode.Nodeinfo,
|
|
Neighbours: testnode.Neighbours,
|
|
}
|
|
}
|
|
|
|
func testfile(name string, obj interface{}) {
|
|
file, err := ioutil.ReadFile("testdata/" + name)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := json.Unmarshal(file, obj); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func testNodesCountWithLinks(links []*GraphLink) int {
|
|
indexMap := make(map[int]bool)
|
|
for _, l := range links {
|
|
indexMap[l.Source] = true
|
|
indexMap[l.Target] = true
|
|
}
|
|
return len(indexMap)
|
|
}
|