yanic/models/graph_test.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)
}