package collector

import (
	"fmt"
	"testing"

	"queryorchestration/internal/database/repository"

	"github.com/google/uuid"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestParseDBCollector(t *testing.T) {
	c, err := parseDBCollector(nil)
	require.NoError(t, err)
	assert.Nil(t, c)

	minCleanV := int64(1)
	minTextV := int64(2)
	ogc := Collector{
		ClientID:        "hi",
		MinCleanVersion: minCleanV,
		MinTextVersion:  minTextV,
		Fields: map[string]uuid.UUID{
			"example_key": uuid.New(),
		},
	}
	c, err = parseDBCollector(&repository.Fullactivecollector{
		Clientid:        ogc.ClientID,
		Mincleanversion: ogc.MinCleanVersion,
		Mintextversion:  ogc.MinTextVersion,
		Fields:          []byte(fmt.Sprintf("{\"example_key\":\"%s\"}", ogc.Fields["example_key"])),
	})
	require.NoError(t, err)
	assert.EqualExportedValues(t, ogc, *c)

	ogc.MinCleanVersion = 0
	ogc.MinTextVersion = 0
	c, err = parseDBCollector(&repository.Fullactivecollector{
		Clientid: ogc.ClientID,
		Fields:   []byte(fmt.Sprintf("{\"example_key\":\"%s\"}", ogc.Fields["example_key"])),
	})
	require.NoError(t, err)
	assert.EqualExportedValues(t, ogc, *c)
}