package tivan

import (
	"testing"

	"github.com/influxdb/tivan/plugins"
	"github.com/stretchr/testify/require"
	"github.com/vektra/cypress"
	"github.com/vektra/neko"
)

func TestAgent(t *testing.T) {
	n := neko.Start(t)

	var (
		plugin  plugins.MockPlugin
		metrics MockMetrics
	)

	n.CheckMock(&plugin.Mock)
	n.CheckMock(&metrics.Mock)

	n.It("drives the plugins and sends them to the metrics", func() {
		a := &Agent{
			plugins: []plugins.Plugin{&plugin},
			metrics: &metrics,
			Config:  &Config{},
		}

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

		m2 := cypress.Metric()
		m2.Add("name", "bar")
		m2.Add("value", 888)

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

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

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

	n.It("applies tags as the messages pass through", func() {
		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)
	})

	n.Meow()
}