package model

import (
	"testing"

	"github.com/QuantumNous/new-api/common"
	"github.com/QuantumNous/new-api/constant"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestGetEnabledGroupSetRequiresEnabledAbilityAndChannel(t *testing.T) {
	resetPricingEndpointTestTables(t)

	require.NoError(t, DB.Create(&[]Channel{
		{Id: 1, Type: constant.ChannelTypeOpenAI, Key: "active-key", Name: "active", Status: common.ChannelStatusEnabled},
		{Id: 2, Type: constant.ChannelTypeOpenAI, Key: "manual-key", Name: "manual", Status: common.ChannelStatusManuallyDisabled},
		{Id: 3, Type: constant.ChannelTypeOpenAI, Key: "auto-key", Name: "auto", Status: common.ChannelStatusAutoDisabled},
		{Id: 4, Type: constant.ChannelTypeOpenAI, Key: "ability-key", Name: "ability", Status: common.ChannelStatusEnabled},
	}).Error)
	require.NoError(t, DB.Create(&[]Ability{
		{Group: "active", Model: "gpt-active", ChannelId: 1, Enabled: true},
		{Group: "manual-disabled", Model: "gpt-manual", ChannelId: 2, Enabled: true},
		{Group: "auto-disabled", Model: "gpt-auto", ChannelId: 3, Enabled: true},
		{Group: "ability-disabled", Model: "gpt-ability", ChannelId: 4, Enabled: false},
	}).Error)

	groups, err := GetEnabledGroupSet()
	require.NoError(t, err)
	assert.Equal(t, map[string]struct{}{"active": {}}, groups)
}