package stomp

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
	"github.com/testcontainers/testcontainers-go/wait"

	"github.com/influxdata/telegraf/config"
	"github.com/influxdata/telegraf/plugins/serializers/json"
	"github.com/influxdata/telegraf/testutil"
)

func TestConnectAndWrite(t *testing.T) {
	if testing.Short() {
		t.Skip("Skipping integration test in short mode")
	}

	servicePort := "61613"
	container := testutil.Container{
		Image:        "rmohr/activemq",
		ExposedPorts: []string{servicePort},
		WaitingFor:   wait.ForListeningPort(servicePort),
	}
	err := container.Start()
	require.NoError(t, err, "failed to start container")
	defer container.Terminate()
	var url = fmt.Sprintf("%s:%s", container.Address, container.Ports[servicePort])

	s := &json.Serializer{
		TimestampUnits:  config.Duration(10 * time.Second),
		TimestampFormat: "yyy-dd-mmThh:mm:ss",
	}
	require.NoError(t, s.Init())

	st := &STOMP{
		Host:          url,
		QueueName:     "test_queue",
		HeartBeatSend: 0,
		HeartBeatRec:  0,
		Log:           testutil.Logger{},
		serialize:     s,
	}
	require.NoError(t, st.Connect())

	require.NoError(t, st.Write(testutil.MockMetrics()))
}