package main

import (
	_ "fmt"
	"strconv"
	"strings"
	"testing"

	common "miflorad/common"

	"github.com/stretchr/testify/assert"
)

func TestPublishGraphite(t *testing.T) {
	tables := []struct {
		metric mifloraMetric
	}{
		{mifloraErrorMetric{peripheralId: "peri", failed: 1}},
		{mifloraDataMetric{
			peripheralId: "peri",
			metaData:     common.VersionBatteryResponse{BatteryLevel: 100, FirmwareVersion: "2.7.0"},
			sensorData:   common.SensorDataResponse{Temperature: 24.2, Brightness: 121, Moisture: 16, Conductivity: 101},
			connectTime:  3.42,
			readoutTime:  0.23,
			rssi:         -77,
		}},
	}

	for _, table := range tables {
		publish := make(chan string, 100)
		publishGraphite(table.metric, publish, "foo.base")
		close(publish)
		switch table.metric.(type) {
		case mifloraErrorMetric:
			for line := range publish {
				parts := strings.Split(line, " ")
				assert.Equal(t, 3, len(parts))
				assert.Equal(t, "foo.base.miflora.peri.failed", parts[0])
				assert.Equal(t, "1", parts[1])
				timestamp, err := strconv.ParseInt(parts[2], 10, 64)
				assert.NoError(t, err)
				assert.True(t, timestamp >= 0)
			}
		case mifloraDataMetric:
			for line := range publish {
				parts := strings.Split(line, " ")
				assert.Equal(t, 3, len(parts))
				assert.Equal(t, 0, strings.Index(parts[0], "foo.base.miflora.peri"))
				assert.True(t, len(parts[1]) > 0)
				timestamp, err := strconv.ParseInt(parts[2], 10, 64)
				assert.NoError(t, err)
				assert.True(t, timestamp >= 0)
			}
		}
	}
}

func TestPublishInflux(t *testing.T) {
	tables := []struct {
		metric mifloraMetric
	}{
		{mifloraErrorMetric{peripheralId: "peri", failed: 1}},
		{mifloraDataMetric{
			peripheralId: "peri",
			metaData:     common.VersionBatteryResponse{BatteryLevel: 100, FirmwareVersion: "2.7.0"},
			sensorData:   common.SensorDataResponse{Temperature: 24.2, Brightness: 121, Moisture: 16, Conductivity: 101},
			connectTime:  3.42,
			readoutTime:  0.23,
			rssi:         -77,
		}},
	}

	for _, table := range tables {
		publish := make(chan string, 100)
		publishInflux(table.metric, publish)
		close(publish)
		switch table.metric.(type) {
		case mifloraErrorMetric:
			line := <-publish
			parts := strings.Split(line, " ")
			assert.Equal(t, 3, len(parts))
			assert.Equal(t, "miflora,id=peri", parts[0])
			assert.Equal(t, "failed=1", parts[1])
			timestamp, err := strconv.ParseInt(parts[2], 10, 64)
			assert.NoError(t, err)
			assert.True(t, timestamp >= 0)
		case mifloraDataMetric:
			line := <-publish
			parts := strings.Split(line, " ")
			assert.Equal(t, 3, len(parts))
			assert.Equal(t, "miflora,id=peri", parts[0])
			assert.True(t, len(parts[1]) > 0)
			timestamp, err := strconv.ParseInt(parts[2], 10, 64)
			assert.NoError(t, err)
			assert.True(t, timestamp >= 0)
		}
	}
}