yanic/runtime/stats_test.go

package runtime

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/FreifunkBremen/yanic/data"
)

const (
	TEST_SITE   = "ffhb"
	TEST_DOMAIN = "city"
)

func TestGlobalStats(t *testing.T) {
	stats := NewGlobalStats(createTestNodes(), map[string][]string{TEST_SITE: {TEST_DOMAIN}})

	assert := assert.New(t)
	assert.Len(stats, 2)

	//check GLOBAL_SITE stats
	assert.EqualValues(1, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Gateways)
	assert.EqualValues(3, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Nodes)
	assert.EqualValues(25, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Clients)

	// check models
	assert.Len(stats[GLOBAL_SITE][GLOBAL_DOMAIN].Models, 2)
	assert.EqualValues(2, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Models["TP-Link 841"])
	assert.EqualValues(1, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Models["Xeon Multi-Core"])

	// check firmwares
	assert.Len(stats[GLOBAL_SITE][GLOBAL_DOMAIN].Firmwares, 1)
	assert.EqualValues(1, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Firmwares["2016.1.6+entenhausen1"])

	// check autoupdater
	assert.Len(stats[GLOBAL_SITE][GLOBAL_DOMAIN].Autoupdater, 2)
	assert.EqualValues(1, stats[GLOBAL_SITE][GLOBAL_DOMAIN].Autoupdater["stable"])

	// check TEST_SITE stats
	assert.EqualValues(1, stats[TEST_SITE][GLOBAL_DOMAIN].Gateways)
	assert.EqualValues(2, stats[TEST_SITE][GLOBAL_DOMAIN].Nodes)
	assert.EqualValues(23, stats[TEST_SITE][GLOBAL_DOMAIN].Clients)

	// check models
	assert.Len(stats[TEST_SITE][GLOBAL_DOMAIN].Models, 2)
	assert.EqualValues(1, stats[TEST_SITE][GLOBAL_DOMAIN].Models["TP-Link 841"])
	assert.EqualValues(1, stats[TEST_SITE][GLOBAL_DOMAIN].Models["Xeon Multi-Core"])

	// check firmwares
	assert.Len(stats[TEST_SITE][GLOBAL_DOMAIN].Firmwares, 1)
	assert.EqualValues(1, stats[TEST_SITE][GLOBAL_DOMAIN].Firmwares["2016.1.6+entenhausen1"])

	// check autoupdater
	assert.Len(stats[TEST_SITE][GLOBAL_DOMAIN].Autoupdater, 1)
	assert.EqualValues(0, stats[TEST_SITE][GLOBAL_DOMAIN].Autoupdater["stable"])

	// check TEST_DOMAIN stats
	assert.EqualValues(1, stats[TEST_SITE][TEST_DOMAIN].Gateways)
	assert.EqualValues(1, stats[TEST_SITE][TEST_DOMAIN].Nodes)
	assert.EqualValues(0, stats[TEST_SITE][TEST_DOMAIN].Clients)

	// check models
	assert.Len(stats[TEST_SITE][TEST_DOMAIN].Models, 1)
	assert.EqualValues(0, stats[TEST_SITE][TEST_DOMAIN].Models["TP-Link 841"])
	assert.EqualValues(1, stats[TEST_SITE][TEST_DOMAIN].Models["Xeon Multi-Core"])

	// check firmwares
	assert.Len(stats[TEST_SITE][TEST_DOMAIN].Firmwares, 0)
	assert.EqualValues(0, stats[TEST_SITE][TEST_DOMAIN].Firmwares["2016.1.6+entenhausen1"])

	// check autoupdater
	assert.Len(stats[TEST_SITE][TEST_DOMAIN].Autoupdater, 1)
	assert.EqualValues(0, stats[TEST_SITE][TEST_DOMAIN].Autoupdater["stable"])
}

func createTestNodes() *Nodes {
	nodes := NewNodes(&NodesConfig{})

	nodeData := &Node{
		Online: true,
		Statistics: &data.Statistics{
			Clients: data.Clients{
				Total: 23,
			},
		},
		Nodeinfo: &data.Nodeinfo{
			NodeID: "abcdef012345",
			Hardware: data.Hardware{
				Model: "TP-Link 841",
			},
			System: data.System{
				SiteCode: TEST_SITE,
			},
		},
	}
	nodeData.Nodeinfo.Software.Firmware = &struct {
		Base    string `json:"base,omitempty"`
		Release string `json:"release,omitempty"`
	}{
		Release: "2016.1.6+entenhausen1",
	}
	nodes.AddNode(nodeData)

	nodes.AddNode(&Node{
		Online: true,
		Statistics: &data.Statistics{
			Clients: data.Clients{
				Total: 2,
			},
		},
		Nodeinfo: &data.Nodeinfo{
			NodeID: "112233445566",
			Hardware: data.Hardware{
				Model: "TP-Link 841",
			},
			Software: data.Software{
				Autoupdater: &struct {
					Enabled bool   `json:"enabled,omitempty"`
					Branch  string `json:"branch,omitempty"`
				}{
					Enabled: true,
					Branch:  "stable",
				},
			},
		},
	})

	nodes.AddNode(&Node{
		Online: true,
		Nodeinfo: &data.Nodeinfo{
			NodeID: "0xdeadbeef0x",
			VPN:    true,
			Hardware: data.Hardware{
				Model: "Xeon Multi-Core",
			},
			System: data.System{
				SiteCode:   TEST_SITE,
				DomainCode: TEST_DOMAIN,
			},
		},
	})

	return nodes
}