agent_test.go

package telegraf

import (
	"github.com/stretchr/testify/assert"
	"testing"

	// needing to load the plugins
	_ "github.com/influxdb/telegraf/plugins/all"
	// needing to load the outputs
	// _ "github.com/influxdb/telegraf/outputs/all"
)

func TestAgent_LoadPlugin(t *testing.T) {

	// load a dedicated configuration file
	config, _ := LoadConfig("./testdata/telegraf-agent.toml")
	a, _ := NewAgent(config)

	pluginsEnabled, _ := a.LoadPlugins([]string{"mysql"})
	assert.Equal(t, 1, len(pluginsEnabled))

	pluginsEnabled, _ = a.LoadPlugins([]string{"foo"})
	assert.Equal(t, 0, len(pluginsEnabled))

	pluginsEnabled, _ = a.LoadPlugins([]string{"mysql", "foo"})
	assert.Equal(t, 1, len(pluginsEnabled))

	pluginsEnabled, _ = a.LoadPlugins([]string{"mysql", "redis"})
	assert.Equal(t, 2, len(pluginsEnabled))

	pluginsEnabled, _ = a.LoadPlugins([]string{"mysql", "foo", "redis", "bar"})
	assert.Equal(t, 2, len(pluginsEnabled))
}

// TODO enable these unit tests, currently disabled because of a circular import
// func TestAgent_LoadOutput(t *testing.T) {
// 	// load a dedicated configuration file
// 	config, _ := LoadConfig("./testdata/telegraf-agent.toml")
// 	a, _ := NewAgent(config)

// 	outputsEnabled, _ := a.LoadOutputs([]string{"influxdb"})
// 	assert.Equal(t, 1, len(outputsEnabled))

// 	outputsEnabled, _ = a.LoadOutputs([]string{})
// 	assert.Equal(t, 2, len(outputsEnabled))

// 	outputsEnabled, _ = a.LoadOutputs([]string{"foo"})
// 	assert.Equal(t, 0, len(outputsEnabled))

// 	outputsEnabled, _ = a.LoadOutputs([]string{"influxdb", "foo"})
// 	assert.Equal(t, 1, len(outputsEnabled))

// 	outputsEnabled, _ = a.LoadOutputs([]string{"influxdb", "kafka"})
// 	assert.Equal(t, 2, len(outputsEnabled))

// 	outputsEnabled, _ = a.LoadOutputs([]string{"influxdb", "foo", "kafka", "bar"})
// 	assert.Equal(t, 2, len(outputsEnabled))
// }

/*
func TestAgent_DrivesMetrics(t *testing.T) {
	var (
		plugin  plugins.MockPlugin
	)

	defer plugin.AssertExpectations(t)
	defer metrics.AssertExpectations(t)

	a := &Agent{
		plugins: []plugins.Plugin{&plugin},
		Config:  &Config{},
	}

	plugin.On("Add", "foo", 1.2, nil).Return(nil)
	plugin.On("Add", "bar", 888, nil).Return(nil)

	err := a.crank()
	require.NoError(t, err)
}

func TestAgent_AppliesTags(t *testing.T) {
	var (
		plugin  plugins.MockPlugin
		metrics MockMetrics
	)

	defer plugin.AssertExpectations(t)
	defer metrics.AssertExpectations(t)

	a := &Agent{
		plugins: []plugins.Plugin{&plugin},
		metrics: &metrics,
		Config: &Config{
			Tags: map[string]string{
				"dc": "us-west-1",
			},
		},
	}

	m1 := cypress.Metric()
	m1.Add("name", "foo")
	m1.Add("value", 1.2)

	msgs := []*cypress.Message{m1}

	m2 := cypress.Metric()
	m2.Timestamp = m1.Timestamp
	m2.Add("name", "foo")
	m2.Add("value", 1.2)
	m2.AddTag("dc", "us-west-1")

	plugin.On("Read").Return(msgs, nil)
	metrics.On("Receive", m2).Return(nil)

	err := a.crank()
	require.NoError(t, err)
}
*/