67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/FreifunkBremen/respond-collector/data"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
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")
|
|
|
|
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.Equal(3, len(graph.Batadv.Nodes), "wrong Nodes count")
|
|
assert.Equal(2, len(graph.Batadv.Links), "wrong Links 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)
|
|
}
|
|
}
|