query-orchestration/internal/query/result/processor/service_test.go

package resultprocessor_test

import (
	"testing"

	resultprocessor "queryorchestration/internal/query/result/processor"

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

func TestCreateGetConfig(t *testing.T) {
	entity := resultprocessor.Create{}

	assert.Nil(t, entity.GetConfig())

	cfg := "example_config"
	entity.Config = &cfg
	assert.NotNil(t, entity.GetConfig())
	assert.Equal(t, cfg, *entity.GetConfig())
}

func TestCreateSetConfig(t *testing.T) {
	entity := resultprocessor.Create{}

	assert.Nil(t, entity.Config)

	cfg := "example_config"
	entity.SetConfig(&cfg)
	assert.Equal(t, cfg, *entity.Config)
}

func TestUpdateGetConfig(t *testing.T) {
	entity := resultprocessor.Update{}

	assert.Nil(t, entity.GetConfig())

	cfg := "example_config"
	entity.Config = &cfg
	assert.NotNil(t, entity.GetConfig())
	assert.Equal(t, cfg, *entity.GetConfig())
}

func TestUpdateSetConfig(t *testing.T) {
	entity := resultprocessor.Update{}

	assert.Nil(t, entity.Config)

	cfg := "example_config"
	entity.SetConfig(&cfg)
	assert.Equal(t, cfg, *entity.Config)
}

func TestCreateGetRequiredQueryIDs(t *testing.T) {
	entity := resultprocessor.Create{}

	assert.Nil(t, entity.GetRequiredQueryIDs())

	ids := []uuid.UUID{uuid.New()}
	entity.RequiredQueryIDs = &ids
	assert.NotNil(t, entity.GetRequiredQueryIDs())
	assert.Equal(t, ids, *entity.GetRequiredQueryIDs())
}

func TestCreateSetRequiredQueryIDs(t *testing.T) {
	entity := resultprocessor.Create{}

	assert.Nil(t, entity.RequiredQueryIDs)

	ids := []uuid.UUID{uuid.New()}
	entity.SetRequiredQueryIDs(&ids)
	assert.Equal(t, ids, *entity.RequiredQueryIDs)
}

func TestUpdateGetRequiredQueryIDs(t *testing.T) {
	entity := resultprocessor.Update{}

	assert.Nil(t, entity.GetRequiredQueryIDs())

	ids := []uuid.UUID{uuid.New()}
	entity.RequiredQueryIDs = &ids
	assert.NotNil(t, entity.GetRequiredQueryIDs())
	assert.Equal(t, ids, *entity.GetRequiredQueryIDs())
}

func TestUpdateSetRequiredQueryIDs(t *testing.T) {
	entity := resultprocessor.Update{}

	assert.Nil(t, entity.RequiredQueryIDs)

	ids := []uuid.UUID{uuid.New()}
	entity.SetRequiredQueryIDs(&ids)
	assert.Equal(t, ids, *entity.RequiredQueryIDs)
}