parse_config_test.go

package main

import (
	"bytes"
	"encoding/base64"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"gopkg.in/yaml.v2"
)

var testNodes = GlobalConf {
	EndPoint: "opc.tcp://localhost:4096",
	Nodes: []NodeConfig {
		{
			NodeName:   "foo",
			MetricName: "bar",
		},{
			NodeName:   "baz",
			MetricName: "bak",
			ExtractBit: 4,
		},
	},
}

func TestReadNodeFile(t *testing.T) {
	data, _ := yaml.Marshal(testNodes)
	results, endpoint, err := parseConfigYAML(bytes.NewReader(data))
	assert.NoError(t, err)
	assert.Equal(t, testNodes.EndPoint, endpoint)
	assert.Equal(t, len(testNodes.Nodes), len(results))
	assert.IsType(t, NodeConfig{}, results[0])
	assert.Equal(t, testNodes.Nodes[0].NodeName, results[0].NodeName)
	assert.Equal(t, testNodes.Nodes[0].MetricName, results[0].MetricName)

	assert.Nil(t, results[0].ExtractBit)
	assert.Equal(t, 4, results[1].ExtractBit)

	results, endpoint, err = parseConfigYAML(strings.NewReader("foooob not valid json here"))
	assert.Error(t, err)
	assert.Empty(t, results)
}

func TestB64Config(t *testing.T) {
	data, _ := yaml.Marshal(testNodes)
	encodedData := base64.StdEncoding.EncodeToString(data)
	results, endpoint, err := readConfigBase64(&encodedData)
	assert.NoError(t, err)
	assert.Equal(t, testNodes.EndPoint, endpoint)
	assert.Equal(t, len(testNodes.Nodes), len(results))
	assert.IsType(t, NodeConfig{}, results[0])
	assert.Equal(t, testNodes.Nodes[0].NodeName, results[0].NodeName)
	assert.Equal(t, testNodes.Nodes[0].MetricName, results[0].MetricName)
}

func TestJsonConfig(t *testing.T) {
	// Luckily JSON is also YAML
	json := `{"endpoint":"opc.tcp://localhost:4096", "nodes":[{"metricName": "foo", "nodeName": "whatever", "extractBit": 1}]}`
	results, _, err := parseConfigYAML(strings.NewReader((json)))
	assert.NoError(t, err)
	assert.Equal(t, 1, len(results))
	assert.Equal(t, "foo", results[0].MetricName)
	assert.Equal(t, "whatever", results[0].NodeName)
	assert.Equal(t, 1, results[0].ExtractBit)
}